o uhn2@sddlmZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z ddl mZddlmZdd lmZmZmZmZmZmZmZmZmZmZer^dd lZdd lmZdd l m!Z!Gd ddZ"d S)) annotations)cached_property) TYPE_CHECKINGcast)format_iso8824_dateparse_iso8824_date)CatalogAttributes)"FileSpecificationDictionaryEntries)PageAttributes) PdfReadError) ArrayObjectByteStringObjectDecodedStreamObjectDictionaryObject NameObject NullObject NumberObject StreamObjectTextStringObjectis_null_or_noneN) Generator) PdfWriterc@seZdZdZdJdd ZedKd d ZedLddZedMddZ e j dNddZ edMddZ e j dNddZ edKddZ e j dOddZ edPd d!Z edQd"d#ZedQd$d%ZedMd&d'Zej dRd)d'ZedSd+d,Zej dTd-d,ZedUd/d0Zej dVd2d0ZedWd4d5Zej dXd6d5ZedWd7d8Zej dXd9d8ZedYd;d<Zej dZd>d<ZdKd?d@Zed[dDdEZed\dGdHZdIS)] EmbeddedFilez Container holding the information on an embedded file. Attributes are evaluated lazily if possible. Further information on embedded files can be found in section 7.11 of the PDF 2.0 specification. namestr pdf_objectrreturnNonecCs||_||_dS)z Args: name: The (primary) name as provided in the name tree. pdf_object: The corresponding PDF object to allow retrieving further data. N)_namer)selfrrr J/var/www/html/bot/env/lib/python3.10/site-packages/pypdf/generic/_files.py__init__'s zEmbeddedFile.__init__cCs|jS)zEThe (primary) name of the embedded file as provided in the name tree.)rrr r r!r0szEmbeddedFile.namewriterrcontent str | bytesc Cs>t|tr |d}t}|||ttjtdit }|td| |iddl m }t }|ttjtdtt j||tt j|itj|jvr`| t |jttj<tt |jtj}d|vrt ttjti} | | |td<ntt |d} tt| tj} | |||g|||dS) aO Create a new embedded file and add it to the PdfWriter. Args: writer: The PdfWriter instance to add the embedded file to. name: The filename to display. content: The data in the file. Returns: EmbeddedFile instance for the newly created embedded file. latin-1z /EmbeddedFilez/Fr)create_string_objectz /Filespec/EmbeddedFilesrr) isinstancerencoderset_dataupdaterPGTYPEr _add_object pypdf.genericr(r FEFCANAMES _root_objectrr extend) clsr$rr% file_entryef_entryr(filespec names_dictembedded_files_names_dictionary names_arrayr r r! _create_new5s6       zEmbeddedFile._create_new str | NonecCsDtjtjfD]}||jvr|j|}t|stt|SqdS)z3Retrieve the alternative name (file specification).N)r UFr3r get_objectrrr)rkeyvaluer r r!alternative_namens zEmbeddedFile.alternative_namerETextStringObject | NonecCs|dur&tj|jvrt|jttj<tj|jvr$t|jttj<dSdStj|jvr4||jttj<tj|jvrD||jttj<dSdS)z.Set the alternative name (file specification).N)r rBrrrr3rrEr r r!rFzs    cCs|jtj}t|r dS|S)zRetrieve the description.N)rgetr DESCrrHr r r! descriptionszEmbeddedFile.descriptioncCs2|durt|jttj<dS||jttj<dS)zSet the description.N)rrrr rJrHr r r!rKscCs|jddS)zJRetrieve the relationship of the referring document to this embedded file./AFRelationshipz /Unspecified)rrIr#r r r!associated_file_relationshipsz)EmbeddedFile.associated_file_relationshiprcCs||jtd<dS)zESet the relationship of the referring document to this embedded file.rLN)rrrHr r r!rMsrcCshd|jvr td|jtt|jd}tjtjfD]}||vr,tt||Sqtd|)z)Retrieve the actual embedded file stream.z/EFz/EF entry not found: z'No /(U)F key found in file dictionary: ) rr rrr rBr3rrC)refrDr r r!_embedded_files zEmbeddedFile._embedded_filecCs|jdtS)z&Retrieve the file-specific parameters./Params)rOrIrrCr#r r r!_paramsszEmbeddedFile._paramscCs*|j}d|vrt|td<tt|dS)z3Ensure the /Params dictionary exists and return it.rP)rOrrr)r embedded_filer r r!_ensure_paramsszEmbeddedFile._ensure_paramscC|jd}t|r dS|S)zERetrieve the subtype. This is a MIME media type, prefixed by a slash./SubtypeN)rOrIrrHr r r!subtype zEmbeddedFile.subtypeNameObject | NonecC0|j}|durt|td<dS||td<dS)zGSet the subtype. This should be a MIME media type, prefixed by a slash.NrU)rOrr)rrErRr r r!rVbytescCs |jS)z!Retrieve the actual file content.)rOget_datar#r r r!r%s zEmbeddedFile.contentcCs$t|tr |d}|j|dS)zSet the file content.r'N)r+rr,rOr-rHr r r!r%s   int | NonecCrT)z4Retrieve the size of the uncompressed file in bytes./SizeNrQrIrrHr r r!sizerWzEmbeddedFile.sizeNumberObject | NonecCrY)z/Set the size of the uncompressed file in bytes.Nr^rSrrrrEparamsr r r!r`rZdatetime.datetime | NonecCt|jdS)z$Retrieve the file creation datetime. /CreationDaterrQrIr#r r r! creation_datezEmbeddedFile.creation_datecC<|j}|durt|td<dSt|}t||td<dS)zSet the file creation datetime.NrgrSrrrrrrErddate_strr r r!ri cCrf)z4Retrieve the datetime of the last file modification./ModDaterhr#r r r!modification_daterjzEmbeddedFile.modification_datecCrk)z/Set the datetime of the last file modification.Nrprlrmr r r!rqro bytes | NonecCrT)z5Retrieve the MD5 checksum of the (uncompressed) file. /CheckSumNr_rHr r r!checksumrWzEmbeddedFile.checksumByteStringObject | NonecCrY)z0Set the MD5 checksum of the (uncompressed) file.Nrsrbrcr r r!rtrZcCsd|jjd|jdS)N) __class____name__rr#r r r!__repr__szEmbeddedFile.__repr__namesr Generator[EmbeddedFile]ccsHt|D]\}}t|ts!|}||d}t||dVqdS)z Convert the given name tree into class instances. Args: names: The name tree to load the data from. Returns: Iterable of class instances for the files found. r*N) enumerater+rrCr)r9r{irfile_dictionary direct_namer r r!_load_from_namess zEmbeddedFile._load_from_namescatalogccsztttt|dd}Wn tyYdSwd|vr?tt|dD]}|}d|vr>|tt|dEdHq'd|vrR|tt|dEdHdSdS)aI Load the embedded files for the given document catalog. This method and its signature are considered internal API and thus not exposed publicly for now. Args: catalog: The document catalog to load from. Returns: Iterable of class instances for the files found. z/Namesr)Nz/Kids)rrKeyErrorr rCr)r9r containerkidr r r!_load-s$  zEmbeddedFile._loadN)rrrrrr)rr)r$rrrr%r&rr)rrA)rErGrr)rErrr)rr)rr)rErXrr)rr[)rEr&rr)rr])rErarr)rre)rErerr)rrr)rErurr)r{r rr|)rrrr|)ry __module__ __qualname____doc__r"propertyr classmethodr@rFsetterrKrMrOrQrrSrVr%r`rirqrtrzrrr r r r!rsl   8                       r)# __future__r functoolsrtypingrr pypdf._utilsrrpypdf.constantsrr5r r r/ pypdf.errorsr r2r r rrrrrrrrdatetimecollections.abcr pypdf._writerrrr r r r!s     0