o
Df!+ @ s d dl Z d dlZd dlZd dlmZ ddlmZ ddlmZ d dl Z d dl
Z
ze e W n
e
y9 eZ eZ Y nw G dd deZdS ) N) OrderedDict )Source)
Configurationc @ s e Zd ZdZG dd deZ d 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 F/home/ubuntu/webapp/venv/lib/python3.10/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 u r
t n|| _z
| jjd| _W n ty( | jj| _Y nw t | _| j r;| j
| | | jj | _ |d urJ| j
| |d u rPi n|| _
|| _|| _| | _|| _g | _d S )Nutf-8)r sourcer
configuration wkhtmltopdfdecodeAttributeErrorr optionsisStringupdate_find_options_in_metaenvirontoccover cover_firstverbosecss stylesheets)
r
url_or_filetype_r r r r# r r! r" r r r r
) s*
zPDFKit.__init__c c sn | |D ].\}}|V t|ttfr1t|dkr"|d r"|d s&J d|d V |d V q|V qdS )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
isinstancelisttuplelen)r
optsoptkeyoptvalr r r _genargsD s $
zPDFKit._genargsc c s | j r
| | j | jV | js| jddi | | jD ]}|r&|V q | jr4| jr4dV | jV | j rHdV | | j D ]}|rG|V q@| jrU| jsUdV | jV | j
s_| j
rcdV nt
| j
j
trq| j
V n
| j
j
D ]}|V qu|r|V dS dV dS )z0
Generator of all command parts
--quiet r r -N)r# _prepend_cssr r" r r r0 r r! r r r isFileObjr)
basestringto_s)r
pathargpartsr r r _commandT s>
zPDFKit._commandc C s t | |S r )r* r; )r
r8 r r r command s zPDFKit.commandc C sx | dkrd S | }t|dkr | d dkr d S d|v r&td| d|v r0td| |p3d }td
| |) 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_codestderr stderr_lines error_msgr r r handle_error s zPDFKit.handle_errorc
C s | |}tjdkr*t }| jtjO _tj|_tj |tj
tj
tj
| j |d}ntj |tj
tj
tj
| j d}| j
sE| j rN| jrN| j d}n| j r]| j j d}nd }|j|d\}}|pl|pld}|jddd}|j}| || d |vrtj| |s|S z/tj|dd
} | d }
|
d krt d
d | W d W dS 1 sw Y W d S t t fy } z t dd || f d } ~ ww )Nwin32)stdinstdoutrE env startupinfo)rJ rK rE rL r )input replace)errorsr1 encoding r2 zBCommand failed: %s
Check whhtmltopdf output without 'quiet' option TzFCommand failed: %s
Check whhtmltopdf output without 'quiet' option
%s ) r< sysplatform
subprocess STARTUPINFOdwFlagsSTARTF_USESHOWWINDOWSW_HIDE wShowWindowPopenPIPEr r r isFiler# r7 encoder5 read communicater
returncoderH rK writecodecsopenrB joinOSError) r
r8 argsrM resultrN rK rE rD ftexter r r to_pdf sb
( z
PDFKit.to_pdfc c s t | D ];\}}d|vrd| | }n| |}t|t tfr.|D ]}||fV q%qt|tr5dn|}||r>t|n|fV qdS )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--%sr2 N)r* items_normalize_argr) r+ boolunicode)r
r keyvaluenormalized_keyr/ normalized_valuer r r r( s
zPDFKit._normalize_optionsc C s | S r )lower)r
argr r r rq zPDFKit._normalize_argc C s d| S )Nzr )r
stylesheetr r r _style_tag_for rz zPDFKit._style_tag_forc C sJ | j s t| j j tr| dt|ts|g}g }|D ] }tj|dd}|| W d n1 s8w Y q d |}| j
rwtj| j dd }| }W d n1 saw Y t |
d| |d d| _ d S | j rd| j v r| j
d| |d | j _ d S | || j | j _ d S d S )Nz6CSS files can be added only to a single file or stringzUTF-8rR
zstring)r isUrlr) r* r rf rg appendrb rh r` r7 r rP r| r )r
r8 css_dataprl inpr r r r4 s4
zPDFKit._prepend_cssc C sz t |tjs |jjdkr| }i }td|D ]"}td| j j
|r:td| j j
|d }td|d ||< q|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 rb refindallsearchr meta_tag_prefix)r
contentfoundxnamer r r r s z PDFKit._find_options_in_meta)NNNNNFFr )r r r r Exceptionr r
r0 r; r< staticmethodrH ro r( rq r| r4 r r r r r r s"
0
A r )r rX rV collectionsr r r r r r rf r6 rs NameErrorstrobjectr r r r r s