3
g!+ @ s d dl Z d dlZd dlZd dlmZ ddlmZ ddlmZ d dl Z d dl
Z
y 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 )N)msg)selfr r K/var/www/tester-filtro-web/env/lib/python3.6/site-packages/pdfkit/pdfkit.py__init__# s z#PDFKit.ImproperSourceError.__init__c C s | j S )N)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|| _y| jjjd| _W n tk
rR | jj| _Y nX t | _| jj rx| jj
| j | | jj | _ |d k r| jj
| |d kri n|| _
|| _|| _| | _|| _g | _d S )Nzutf-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 sp xj| j |D ]\\}}|V t|ttfrbt|dkrD|d rD|d sLtd|d V |d V q |V q W 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| j | jV | js0| jjddi x | j| jD ]}|r>|V q>W | jrj| jrjdV | jV | j rdV x | j| j D ]}|r|V qW | jr| j rdV | jV | j
j s| j
j rdV n4t
| j
j
tr| j
j V nx| j
j
D ]
}|V qW |r|V ndV dS )z0
Generator of all command parts
z--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 | j|S )N)r% r4 )r r2 r r r
command s zPDFKit.commandc C s| | dkr d S |j }t|dkr8|j d j dkr8d S d|krLtd| d|kr`td| |pfd }td
j| |d S ) Nr r 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 | j |}tjdkrTtj }| jtjO _tj|_tj |tj
tj
tj
| j |d}n tj |tj
tj
tj
| j d}| j j
s| j j r| jr| j j jd}n"| j j r| j j j jd}nd }|j|d\}}|p|pd}|jddd}|j}| j|| d |krtjj| |s|S yDtj|dd
,} | jd }
|
d krRt d
dj | dS Q R X W n> t t fk
r } z t ddj || f W Y d d } ~ X nX d S )Nwin32)stdinstdoutr> env startupinfo)rB rC r> rD zutf-8)input replace)errorsz--quiet)encoding r, zBCommand failed: %s
Check whhtmltopdf output without 'quiet' option TzFCommand failed: %s
Check whhtmltopdf output without 'quiet' option
%s ) r5 sysplatform
subprocessZ STARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWZSW_HIDEZ wShowWindowPopenPIPEr r r isFiler r1 encoder/ read communicater
returncoder@ rC writecodecsopenr; joinOSError) r r2 argsrE resultrF rC r> r= ftexter r r
to_pdf sT
z
PDFKit.to_pdfc c s xt |j D ]z\}}d|kr.d| j| }n
| j|}t|t tfr`x@|D ]}||fV qLW qt|trndn|}||rt|n|fV qW 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--%sr, N)r% items_normalize_argr$ r& boolunicode)r r keyvalueZnormalized_keyr* Znormalized_valuer r r
r# s
zPDFKit._normalize_optionsc C s |j S )N)lower)r argr r r
rc s zPDFKit._normalize_argc C s d| S )Nzr )r Z
stylesheetr r r
_style_tag_for s zPDFKit._style_tag_forc C s" | j j st| j j tr"| jdt|ts2|g}g }x4|D ],}tj|dd}|j|j W d Q R X qstring)r ZisUrlr$ r% r rX rY appendrT rZ rR r1 r rH rj r )r r2 Zcss_datapr^ inpr r r
r. s(
zPDFKit._prepend_cssc C s~ t |tjs|jjdkr |j }i }xTtjd|D ]D}tjd| j j
|r2tjd| j j
|d }tjd|d ||< q2W |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
rT 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+ r4 r5 staticmethodr@ ra r# rc rj r. r r r r r
r s
0
A r )rt rO rM collectionsr r r r r rq rX r0 re NameErrorstrobjectr r r r r
s