o e @sdZddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZGd d d ZGd ddZdddZdS)zCDefines context objects that are passed to child processes/threads.) annotationsN) Namespace)copy)Path)Iterator) PluginManager)PdfInfo)PageInfoc@sPeZdZUdZded<ded<ded<ded <dd d ZdddZdddZdS) PdfContextz7Holds the context for a particular run of the pipeline.roptionsroriginrpdfinforplugin_manager work_foldercCs"||_||_||_||_||_dSN)r rr r r)selfr rr r rrDD:\Projects\ConvertPro\env\Lib\site-packages\ocrmypdf/_jobcontext.py__init__s  zPdfContext.__init__namestrreturncCs |j|S)zGenerate a ``Path`` for an intermediate file involved in processing. The path will be in a temporary folder that is common for all processing of this particular PDF. )rrrrrrget_path,s zPdfContext.get_pathIterator[PageContext]ccs*t|j}t|D]}t||Vq dS)z%Get all ``PageContext`` for this PDF.N)lenr range PageContext)rZnpagesnrrrget_page_contexts4s   zPdfContext.get_page_contextsN)r rrrr rr rrrrr)rr)__name__ __module__ __qualname____doc____annotations__rrrrrrrr s   r c@sVeZdZUdZded<ded<ded<ded <d ed <dddZdddZddZdS)rzHolds our context for a page. Must be pickable, so stores only intrinsic/simple data elements or those capable of their serializing themselves via ``__getstate__``. rr rr intpagenor pageinforr pdf_contextr cCs6|j|_|j|_|j|_||_|j||_|j|_dSr)rr r r'r r(r)rr)r'rrrrHs   zPageContext.__init__rrrcCs|j|jddd|S)zGenerate a ``Path`` for a file that is part of processing this page. The path will be based in a common temporary folder and have a prefix based on the page number. Z06d_)rr'rrrrrPszPageContext.get_pathcCs`|j}t|j|d<t|djtttjfsd|d_t|dj tttjfs.d|d_ |S)Nr stream) __dict__rr isinstanceZ input_filerbytesosPathLikeZ output_file)rstaterrr __getstate__Xs   zPageContext.__getstate__N)r)r r )r!r"r#r$r%rrr3rrrrr;s    rrrr rcCs0|jrtd|tjddStj|dddS)Nz%Temporary working files retained at: )fileT) ignore_errors)Zkeep_temporary_filesprintsysstderrshutilrmtree)rr rrrcleanup_working_filescsr;)rrr r)r$ __future__rr0r9r7argparserrpathlibrtypingrZpluggyrZocrmypdf.pdfinforZocrmypdf.pdfinfo.infor r rr;rrrrs        %(