o
i!+ @ 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 WC:\Users\psant\Downloads\Anu Raj Backend\school\venv\Lib\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_firstverbosecssZ stylesheets)
r
Z 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
optsZoptkeyoptvalr 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 r- r r! r r r isFileObjr'
basestringto_s)r
pathZargpartsr r r _commandT s>
zPDFKit._commandc C s t | |S r )r( r7 )r
r5 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_codestderrZ 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)stdinstdoutrA env startupinfo)rE rF rA rG r )input replace)errorsr. encoding r/ zBCommand failed: %s
Check whhtmltopdf output without 'quiet' option TzFCommand failed: %s
Check whhtmltopdf output without 'quiet' option
%s ) r8 sysplatform
subprocess STARTUPINFOdwFlagsSTARTF_USESHOWWINDOWSW_HIDE wShowWindowPopenPIPEr r r isFiler# r4 encoder2 read communicater
returncoderC rF writecodecsopenr> joinOSError) r
r5 argsrH resultrI rF rA r@ ftexter r r to_pdf sb
( z
PDFKit.to_pdfc c s t | D ]:\}}d|vrd| | }n| |}t|t tfr-|D ]}||fV q%t|tr4dn|}||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--%sr/ 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 rl zPDFKit._normalize_argc C s d| S )Nzr )r
Z
stylesheetr r r _style_tag_for rs zPDFKit._style_tag_forc C sJ | j s t| j j tr| dt|ts|g}g }|D ] }tj|dd}|| W d q 1 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-8rM
zstring)r ZisUrlr' r( r ra rb appendr] rc r[ r4 r rK rt r )r
r5 Zcss_dataprg inpr r r r1 s2
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=["']([^"']*)N) r' ioIOBase __class__r r] refindallsearchr Zmeta_tag_prefix)r
contentfoundxnamer r r r s"
z PDFKit._find_options_in_meta)NNNNNFFr )r r r r Exceptionr r
r- r7 r8 staticmethodrC rj r& rl rt r1 r r r r r r s"
0
A r )r~ rS rQ collectionsr r r r r r{ ra r3 rn NameErrorstrobjectr r r r r s