B S]2 @sddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZmZmZmZmZdd lmZdd lmZmZdd lmZdddddddddg Zee efZ!ee ej"fZ#edddZ$e dddZ%eeej&dddZ'dd Z(e!e#ed!d"dZ)d1e!e#e e ed%d&dZ*e!e#e+d!d'dZ,d2e!e#e e e d%d(dZ-ee!e#eed!d)dZ.e!e e/d*d+dZ0e!ee dd,dZ1Gd-d.d.ej&Z2d/d0Z3dS)3N)abc)contextmanagersuppress) import_module)ResourceLoader)BytesIO TextIOWrapper)Path) ModuleType)IterableIteratorOptionalSetUnion)cast)BinaryIOTextIO)ZipImportErrorPackageResourcecontents is_resource open_binary open_textpath read_binary read_text)returncCs\t|dr0|jjdkr*td|jjqX|Sn(t|}|jjdkrTtd|n|SdS)zTake a package name or module object and return the module. If a name, the module is imported. If the passed or imported module object is not a package, raise an exception. __spec__Nz{!r} is not a package)hasattrrsubmodule_search_locations TypeErrorformatnamer)packagemoduler'$lib/python3.7/importlib/resources.py _get_package"s   r)cCs,tj|\}}|r$td|n|SdS)zNormalize a path by ensuring it is a string. If the resulting string contains path separators, an exception is raised. z{!r} must be only a file nameN)osrsplit ValueErrorr#)rparent file_namer'r'r(_normalize_path6sr/)r%rcCs,|j}t|jdr(ttj|j|jSdS)Nget_resource_reader)rr loaderr resources_abcResourceReaderr0r$)r%specr'r'r(_get_resource_readerBs  r5cCs&|jjdks|jjs"td|dS)NzPackage has no location )rorigin has_locationFileNotFoundError)r%r'r'r(_check_locationPsr9)r%resourcerc Cst|}t|}t|}|dk r*||St|tj|jj }tj |}tj ||}y t |ddSt k rtt|jj}d}t|jjdrtt ||}WdQRX|dkr|jj}d||} t| nt|SYnXdS)zDReturn a file-like object opened for binary reading of the resource.Nrb)modeget_dataz{!r} resource not found in {!r})r/r)r5 open_resourcer9r*rabspathrr6dirnamejoinopenOSErrorrrr1r rr=r$r#r8r) r%r:readerabsolute_package_path package_path full_pathr1data package_namemessager'r'r(rUs.     utf-8strict)r%r:encodingerrorsrc Cst|}t|}t|}|dk r2t||||St|tj|j j }tj |}tj ||}yt |d||dStk rtt|j j}d} t|j jdrtt||} WdQRX| dkr|j j} d|| } t| ntt| ||SYnXdS)zBReturn a file-like object opened for text reading of the resource.Nr)r<rMrNr=z{!r} resource not found in {!r})r/r)r5r r>r9r*rr?rr6r@rArBrCrrr1r rr=r$r#r8r) r%r:rMrNrDrErFrGr1rHrIrJr'r'r(rts.   c Cs.t|}t|}t|| }|SQRXdS)z+Return the binary contents of the resource.N)r/r)rread)r%r:fpr'r'r(rs c Cs2t|}t|}t|||| }|SQRXdS)zReturn the decoded string of the resource. The decoding-related arguments have the same semantics as those of bytes.decode(). N)r/r)rrP)r%r:rMrNrQr'r'r(rs c cst|}t|}t|}|dk rLyt||VdStk rHYqTXnt|t|jjj }||}| rz|Vnvt ||}| }WdQRXt \}}z$t||t|t|VWdyt|Wntk rYnXXdS)akA context manager providing a file path object to the resource. If the resource does not already exist on its own on the file system, a temporary file will be created. If the file was created, the file will be deleted upon exiting the context manager (no exception is raised if the file was deleted prior to the context manager exiting). N)r/r)r5r resource_pathr8r9rr6r-existsrrPtempfileZmkstempr*writecloseremove) r%r:rDpackage_directoryZ file_pathrQrHfdZraw_pathr'r'r(rs2     )r%r$rc Cszt|}t|t|}|dk r*||Sytt|}Wnttfk rRdSX||kr`dSt|j j j |}| S)zYTrue if 'name' is a resource inside 'package'. Directories are *not* resources. NF) r)r/r5rsetrNotADirectoryErrorr8r rr6r-is_file)r%r$rDZpackage_contentsrr'r'r(rs cCsTt|}t|}|dk r |S|jjdks4|jjs8dSt|jjj}t |SdS)zReturn an iterable of entries in 'package'. Note that not all entries are resources. Specifically, directories are not considered resources. Use `is_resource()` on each entry returned here to check if it is a resource or not. Nr') r)r5rrr6r7r r-r*listdir)r%rDrXr'r'r(rsc@s8eZdZdZddZddZddZdd Zd d Zd S) _ZipImportResourceReaderzPrivate class used to support ZipImport.get_resource_reader(). This class is allowed to reference all the innards and private parts of the zipimporter. cCs||_||_dS)N) zipimporterfullname)selfr_r`r'r'r(__init__sz!_ZipImportResourceReader.__init__cCsN|jdd}|d|}yt|j|Stk rHt|YnXdS)N./)r`replacerr_r=rCr8)rar:fullname_as_pathrr'r'r(r>s z&_ZipImportResourceReader.open_resourcecCstdS)N)r8)rar:r'r'r(rRsz&_ZipImportResourceReader.resource_pathcCsF|jdd}|d|}y|j|Wntk r@dSXdS)NrcrdFT)r`rer_r=rC)rar$rfrr'r'r(r%sz$_ZipImportResourceReader.is_resourcec cst|j|j}||jj}|jdks.t|j}t }xp|jj D]d}yt||}Wnt k rpwDYnX|jj}t |dkr|jVqD||krD| ||VqDWdS)Nz __init__.pyr)r r_ get_filenamer`Z relative_toarchiver$AssertionErrorr-rZ_filesr,lenadd)raZ fullname_pathZ relative_pathrFZ subdirs_seenfilenameZrelative parent_namer'r'r(r0s    z!_ZipImportResourceReader.contentsN) __name__ __module__ __qualname____doc__rbr>rRrrr'r'r'r(r^ s  r^cCs2y||sdSWntk r&dSXt||S)N) is_packagerr^)r_r`r'r'r(_zipimport_get_resource_readerQs  rt)rKrL)rKrL)4r*rTrr2 contextlibrr importlibrZ importlib.abcriorr Zpathlibr typesr typingr r rrrrZ typing.iorr zipimportr__all__strrPathLikerr)r/r3r5r9rrbytesrrrboolrrr^rtr'r'r'r(sN          !  ,E