U
7h!+ @ s d dl Z d dlZd dlZd dlmZ ddlmZ ddlmZ d dl Z d dl
Z
z e e W n e
k
rt eZ eZ Y nX G dd deZdS ) N) OrderedDict )Source)
Configurationc @ s e Zd ZdZG dd deZd ddZdd Zd d
d Zd d d
Z e
dd Z d ddZ dd Z
dd Zdd Zdd Zdd ZdS ) PDFKita,
Main class that does all generation routine.
:param url_or_file: str - either a URL, a path to a file or a string containing HTML
to convert
:param type_: str - either 'url', 'file' or 'string'
:param options: dict (optional) with wkhtmltopdf options, with or w/o '--'
:param toc: dict (optional) - toc-specific wkhtmltopdf options, with or w/o '--'
:param cover: str (optional) - url/filename with a cover html page
:param configuration: (optional) instance of pdfkit.configuration.Configuration()
c @ s e Zd ZdZdd Zdd ZdS )zPDFKit.ImproperSourceErrorz!Wrong source type for stylesheetsc C s
|| _ d S Nmsg)selfr r G/var/www/formularioweb/env/lib/python3.8/site-packages/pdfkit/pdfkit.py__init__# s z#PDFKit.ImproperSourceError.__init__c C s | j S r r )r
r r r __str__& s z"PDFKit.ImproperSourceError.__str__N)__name__
__module__ __qualname____doc__r
r r r r r ImproperSourceError s r NFc
C s t ||| _|d krt n|| _z| jjd| _W n tk
rR | jj| _Y nX t | _| j rx| j
| | | jj | _ |d k r| j
| |d kri n|| _
|| _|| _| | _|| _g | _d S )Nutf-8)r sourcer
configuration wkhtmltopdfdecodeAttributeErrorr optionsisStringupdate_find_options_in_metaenvirontoccover cover_firstverbosecssZ stylesheets)
r
Z url_or_filetype_r r r r# r r! r" r r r r
) s(
zPDFKit.__init__c c sl | |D ]\\}}|V t|ttfr`t|dkrB|d rB|d sJtd|d V |d V q
|V q
dS )z
Generator of args parts based on options specification.
Note: Empty parts will be filtered out at _command generator
r r zFOption value can only be either a string or a (tuple, list) of 2 itemsN)_normalize_options
isinstancelisttuplelenAssertionError)r
optsZoptkeyoptvalr r r _genargsD s $
zPDFKit._genargsc c s | j r| | j | jV | js0| jddi | | jD ]}|r<|V q<| jrf| jrfdV | jV | j rdV | | j D ]}|r~|V q~| jr| jsdV | jV | j
s| j
rdV n0t
| j
j
tr| j
V n| j
j
D ]
}|V q|r|V ndV dS )z0
Generator of all command parts
--quiet r r -N)r# _prepend_cssr r" r r r. r r! r r r isFileObjr'
basestringto_s)r
pathZargpartsr r r _commandT s8 zPDFKit._commandc C s t | |S r )r( r8 )r
r6 r r r command s zPDFKit.commandc C s| | dkr d S | }t|dkr8| d dkr8d S d|krLtd| d|kr`td| |pfd }td
| |d S ) Nr r Donezcannot connect to X serverz%s
You will need to run wkhtmltopdf within a "virtual" X server.
Go to the link below for more information
https://github.com/JazzCore/python-pdfkit/wiki/Using-wkhtmltopdf-without-X-serverErrorz wkhtmltopdf reported an error:
z
Unknown Errorz5wkhtmltopdf exited with non-zero code {0}. error:
{1})
splitlinesr* stripIOErrorformat) exit_codestderrZ stderr_lines error_msgr r r handle_error s zPDFKit.handle_errorc
C s | |}tjdkrTt }| jtjO _tj|_tj |tj
tj
tj
| j |d}n tj |tj
tj
tj
| j d}| j
s| j r| jr| j d}n"| j r| j j d}nd }|j|d\}}|p|pd}|jddd}|j}| || d |krtj| |s|S zPtj|dd
8} | d }
|
d krRt d
d | W 5 Q R W dS Q R X W n> t t fk
r } zt dd || f W 5 d } ~ X Y nX d S )Nwin32)stdinstdoutrB env startupinfo)rF rG rB rH r )input replace)errorsr/ encoding r0 zBCommand failed: %s
Check whhtmltopdf output without 'quiet' option TzFCommand failed: %s
Check whhtmltopdf output without 'quiet' option
%s ) r9 sysplatform
subprocessZ STARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWZSW_HIDEZ wShowWindowPopenPIPEr r r isFiler# r5 encoder3 read communicater
returncoderD rG writecodecsopenr? joinOSError) r
r6 argsrI resultrJ rG rB rA ftexter r r to_pdf s\
z
PDFKit.to_pdfc c s t | D ]v\}}d|kr,d| | }n
| |}t|t tfrZ|D ]}||fV qHq t|trhdn|}||rzt|n|fV q dS )ah Generator of 2-tuples (option-key, option-value).
When options spec is a list, generate a 2-tuples per list item.
:param options: dict {option name: value}
returns:
iterator (option-key, option-value)
- option names lower cased and prepended with
'--' if necessary. Non-empty values cast to str
z--z--%sr0 N)r( items_normalize_argr' r) boolunicode)r
r keyvalueZnormalized_keyr- Znormalized_valuer r r r& s
zPDFKit._normalize_optionsc C s | S r )lower)r
argr r r rh s zPDFKit._normalize_argc C s d| S )Nzr )r
stylesheetr r r _style_tag_for s zPDFKit._style_tag_forc
C s | j st| j j tr"| dt|ts2|g}g }|D ],}tj|dd}|| W 5 Q R X q:d |}| j
rtj| j dd}| }W 5 Q R X t |
d| |d d| _ nV| j rd| j kr| j
d| |d | j _ n| || j | j _ d S )Nz6CSS files can be added only to a single file or stringzUTF-8rN
zstring)r ZisUrlr' r( r r] r^ appendrY r_ rW r5 r rL rp r )r
r6 Zcss_dataprc Zinpr r r r2 s.
zPDFKit._prepend_cssc C sz t |tjs|jjdkr | }i }td|D ]D}td| j j
|r0td| j j
|d }td|d ||< q0|S )zReads 'content' and extracts options encoded in HTML meta tags
:param content: str or file-like object - contains HTML to parse
returns:
dict: {config option: value}
StreamReaderWriterz ]*>z name=["']%szname=["']%s([^"']*)r zcontent=["']([^"']*)) r' ioIOBase __class__r rY refindallsearchr Zmeta_tag_prefix)r
contentfoundxnamer r r r s
z PDFKit._find_options_in_meta)NNNNNFF)N)N)N)r r r r Exceptionr r
r. r8 r9 staticmethodrD rf r& rh rp r2 r r r r r r s&
0
A r )ry rT rR collectionsr r r r r rv r] r4 rj NameErrorstrobjectr r r r r s