U giD @sNdZddlZddlmZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZdd lmZdd lmZdd l mZmZd d ddddgZeddGdddZGdd d e jZeddGdddZeddGdddZejeddd Zdddddddddeeeeee jee j eeeeeej!dddZ"dS)zO Utility classes for handling embedded files in PDFs. .. versionadded:: 0.7.0 N) dataclass)datetime)ListOptional)x509)cryptgenericmiscwriter)RecipientEncryptionPolicy) get_courier)pdf_name pdf_string embed_fileEmbeddedFileObjectEmbeddedFileParamsFileSpecRelatedFileSpecwrap_encrypted_payloadT)frozenc@sFeZdZUdZeed<dZeed<dZee ed<dZ ee ed<dS)rT embed_sizeembed_checksumN creation_datemodification_date) __name__ __module__ __qualname__rbool__annotations__rrrrrr r ;/tmp/pip-unpacked-wheel-owvgwkas/pyhanko/pdf_utils/embed.pyrs  csfeZdZed ejeeeee ddddZ d ejeeee dfdd Z d fd d Z Z S)rTN) pdf_writerdataparams mime_typereturncCs t||||d}|r||S)a Construct an embedded file object from file data. This is a very thin wrapper around the constructor, with a slightly less intimidating API. .. note:: This method will not register the embedded file into the document's embedded file namespace, see :func:`.embed_file`. :param pdf_writer: PDF writer to use. :param data: File contents, as a :class:`bytes` object. :param compress: Whether to compress the embedded file's contents. :param params: Optional embedded file parameters. :param mime_type: Optional MIME type string. :return: An embedded file object. )r" stream_datar$r%)rcompress)clsr"r#r(r$r%resultr r r!from_file_data?s!z!EmbeddedFileObject.from_file_data)r"r$r%csTtj||||jdtd|d<|dk r>td||d<|||_||_dS)N) dict_datar' encoded_datahandlerz /EmbeddedFile/Type/z/Subtype)super__init__security_handlerr r add_object ef_stream_refr$)selfr"r,r'r-r$r% __class__r r!r2ks  zEmbeddedFileObject.__init__c s|dk r:|js:|j}|j}|j}|dk r:||kr:|||j}|dk rt|d<}|jrrt t |j |d<|j rt |j } t| |d<|jdk rt|j|d<|jdk rt|j|d<tj|||ddS)Nz/Paramsz/Sizez /CheckSumz /CreationDatez/ModDate)r. container_ref)Z_has_crypt_filtercrypt_filter_configZembedded_file_filter_namestream_filter_nameZadd_crypt_filterr$r DictionaryObjectrZ NumberObjectlenr#rhashlibmd5digestZByteStringObjectrZpdf_daterr1write_to_stream) r6streamr.r9ZcfcZef_filter_namer;r$Z param_dictZchecksumr7r r!rAs<   z"EmbeddedFileObject.write_to_stream)TNN)NNNNN)NN)rrr classmethodr BasePdfFileWriterbytesrrstrr+r2rA __classcell__r r r7r!r>s,.c@s8eZdZUdZeed<eed<eeddddZ dS)rz= Dataclass modelling a RelatedFile construct in PDF. name embedded_datalstcsfdd}t|S)Nc3s&D]}t|jV|jjVqdS)N)r rrHrIr5)ZrfsrJr r!_gensz/RelatedFileSpec.fmt_related_files.._gen)r ArrayObject)r)rKrLr rJr!fmt_related_filess z!RelatedFileSpec.fmt_related_filesN) rrr__doc__rFrrrCrrNr r r r!rs  c@seZdZUdZeed<dZeeed<dZee ed<dZ eeed<dZ ee j ed<dZeeeed<dZeeeed <e jd d d ZdS) rzD Dataclass modelling an embedded file description in a PDF. file_spec_stringN file_namerI descriptionaf_relationshipf_related_filesuf_related_files)r&cCsttdtdtdt|ji}|jdk r Represent the file spec as a PDF dictionary. r/z /Filespecz/FNz/UFz/EFz/Descz/AFRelationshipz/RF)r r<rrrPrQrIr5rRTextStringObjectrSrTrUrrN)r6r*Zef_dictZ f_relatedZ uf_relatedrfr r r! as_pdf_objects<        zFileSpec.as_pdf_object)rrrrOrFrrQrrIrrRrSr Z NameObjectrTrrrUr<rXr r r r!rs   )r"specc Cs|j}|dkrtd|}|j}z |d}Wn2tk rdt}|||d<| YnXz |d}Wn4tk rt}|||d<| |YnXd|krt dz |d}Wn4tk rt }|||d<| |YnX| t|j||}| || ||jdk r|jddz |d } Wn4tk rt } || |d <| YnX| |n |jd ddS) z Embed a file in the document-wide embedded file registry of a PDF writer. :param pdf_writer: PDF writer to house the embedded file. :param spec: File spec describing the embedded file. :return: Nz/File spec does not have an embedded file streamz/Namesz/EmbeddedFilesz/Kidsz,Only flat name trees are supported right now)r)versionz/AF)r)rIr Z PdfWriteErrorrXrootKeyErrorr r<r4Z update_rootZupdate_containerNotImplementedErrorrMappendrrPrSZensure_output_version) r"rYZ ef_streamZspec_objr]Z names_dictZ ef_name_treeZ ef_name_arrZ spec_obj_refZ root_af_arrr r r!r"sV           zattachment.pdfzWrapped document)passwordcertsr3rPr$rQrRinclude_explanation_page)plaintext_payloadrarbr3rPr$rQr&c Cst} |dkr|dk|dkkr(td|dkr|dk sZ dataclassesrrtypingrrZ asn1cryptorrr r r Z crypt.pubkeyr Z font.basicr rr__all__rrorrrrDrrErFZ CertificateZSecurityHandlerrlrr r r r!sT       g]H