3 Wz)@s|dZddlmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"ej#ddkre j$d d n e j$d d ej%d Z&ej%dZ'ddZ(ddZ)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/dd Z0d!d"Z1ed#d$Z2d%d&Z3Gd'd(d(e4Z5e j6d)d*Z7dS)+z weasyprint.utils ---------------- Various utility functions and classes. :copyright: Copyright 2011-2014 Simon Sapin and contributors, see AUTHORS. :license: BSD, see LICENSE for details. )divisionunicode_literalsN)VERSION_STRING)LOGGER)urljoinurlsplitquoteunquoteunquote_to_bytesurlopenurllib_get_content_typeurllib_get_charseturllib_get_filenameRequest parse_email pathname2urlunicode base64_decodeStreamingGzipFileFILESYSTEM_ENCODINGz image/svg+xmlz.svgs image/svg+xmls.svgz^([a-zA-Z][a-zA-Z0-9.+-]+):s^([a-zA-Z][a-zA-Z0-9.+-]+):cCs2|jdr|S|j|jdr tnd}t|ddS)zmTurn an IRI that can contain any Unicode character into an ASCII-only URI that conforms to RFC 3986. zdata:zfile:zutf-8s/:?#[]@!$&'()*+,;=~%)safe) startswithencoderr )urlrO/var/www/html/enquirykeeper_venv/lib/python3.6/site-packages/weasyprint/urls.py iri_to_uri2s  rcCsJtjj|}tjj|r$|tjj7}t|}|jdr>d|Sd|SdS)zReturn file URL of `path`z///zfile:zfile://N)ospathabspathisdirseprr)r rrrpath2urlEs    r$cCstt|trtntj|S)N)bool isinstancerUNICODE_SCHEME_REBYTES_SCHEME_REmatch)rrrrurl_is_absoluteUsr*cCs |jjjS)uReturn the URL associated with a lxml document. This is the same as the HtmlElement.base_url property, but dont’t want to require HtmlElement. )Z getroottreeZdocinfoURL)elementrrrelement_base_url[sr-cCs4|j|dj}|r0tt||d|j|||jSdS)zGet the URI corresponding to the ``attr_name`` attribute. Return ``None`` if: * the attribute is empty or missing or, * the value is a relative URI but the document has no base URI. Otherwise, return an absolute URI. z<%s %s="%s"> at line %sN)getstripurl_joinr-tagZ sourceline)r, attr_namevaluerrrget_url_attributees  r5cGs>t|rt|S|r"tt||Stjd|f|dSdS)zgLike urllib.urljoin, but issue a warning and return None if base_url is required but missing. z+Relative URI reference without a base URI: N)r*rrrwarning)Zbase_urlrcontextargsrrrr1ws r1cCs|j|dj}|jdr:t|dkr:dt|ddfSt||}|rt|}|rt|}|ddt|ddkrdt|jfSd|fSdS) zZReturn ('external', absolute_uri) or ('internal', unquoted_fragment_id) or None. r.#rZinternalNZexternalr:) r/r0rlenr r5r-rfragment)r,r3Z attr_valueZuriZ document_urlparsedrrrget_link_attributes r>cCst|r |St|S)zGet a ``scheme://path`` URL from ``string``. If ``string`` looks like an URL, return it unchanged. Otherwise assume a filename and convert it to a ``file://`` URL. )r*r$)stringrrr ensure_urlsr@cCs(dt|d}|r |d|7}t|S)aiDecode base64, padding being optional. "From a theoretical point of view, the padding character is not needed, since the number of missing bytes can be calculated from the number of Base64 digits." https://en.wikipedia.org/wiki/Base64#Padding :param data: Base64 data as an ASCII byte string :returns: The decoded byte string. =)r;r)datamissing_paddingrrrsafe_base64_decodes  rEc Csy|jdd\}}Wntk r0tdYnX|dd}|r|jd}|dkrd||dkr|d|}||dd}n|}d }td |}|j}|j}n d }d }d }t|}|d krt|}t ||||dS)uDecode URLs with the 'data' scheme. urllib can handle them in Python 2, but that is broken in Python 3. Inspired from Python 2.7.2’s urllib.py. ,rz bad data URLN;r=r.zContent-type: z text/plainzUS-ASCIIbase64)r? mime_typeencodingredirected_url) split ValueErrorIOErrorrfindrget_content_typeget_content_charsetr rEdict) rheaderrCZsemi content_typerLmessagerKcharsetrrr open_data_urls.      rYz gzip, deflate)z User-AgentzAccept-Encodingc Cs|jjdrt|Stj|r t|}tt|td}t |j t |t |t |d}|jjd}|dkrtdkrtjtj|jdj|d<|jnt|d|d <nX|d kr|j}ytj||d<Wn(tjk rtj|d |d<YnXn||d <|Std |dS)uFFetch an external resource such as an image or stylesheet. Another callable with the same signature can be given as the :obj:`url_fetcher` argument to :class:`HTML` or :class:`CSS`. (See :ref:`url-fetchers`.) :type url: Unicode string :param url: The URL of the resource to fetch :raises: any exception to indicate failure. Failures are logged as warnings, with the string representation of the exception in the message. :returns: In case of success, a dict with the following keys: * One of ``string`` (a byte string) or ``file_obj`` (a file-like object) * Optionally: ``mime_type``, a MIME type extracted eg. from a *Content-Type* header. If not provided, the type is guessed from the file extension in the URL. * Optionally: ``encoding``, a character encoding extracted eg. from a *charset* parameter in a *Content-Type* header * Optionally: ``redirected_url``, the actual URL of the resource in case there were eg. HTTP redirects. * Optionally: ``filename``, the filename of the resource. Usually derived from the *filename* parameter in a *Content-Disposition* header If a ``file_obj`` key is given, it is the caller’s responsability to call ``file_obj.close()``. zdata:)headers)rMrKrLfilenamezContent-EncodinggzipN)fileobjr?file_objdeflatezNot an absolute URI: %ri)lowerrrYr'r)rr r HTTP_HEADERSrTgeturlr rrinfor/rr\ZGzipFileioBytesIOreadclosezlib decompresserrorrO)rresponseresultZcontent_encodingrCrrrdefault_url_fetchers2    rnc@seZdZdZdS)URLFetchingErrorz)Some error happened when fetching an URL.N)__name__ __module__ __qualname____doc__rrrrro$sroc"csy ||}WnLtk rX}z0t|j}t|}t|rBd||fn|WYdd}~XnX|jd||jddd|krz |VWdy|djWn&tk rtjd|t j YnXXn|VdS)z9Call an url_fetcher, fill in optional data, and clean up.z%s: %sNrMrKr^z$Error when closing stream for %s: %s) Exceptiontyperpstrro setdefaultrhrr6 traceback format_exc)Z url_fetcherrrmexcnamer4rrrfetch(s"  *   r|)8rs __future__rrreresysos.pathrZ mimetypes contextlibr\rirxr.rloggerrcompatrrr r r r r rrrrrrrrr version_infoZadd_typecompiler'r(rr$r*r-r5r1r>r@rErYrbrnrProcontextmanagerr|rrrr sB  H      *>