dZddlZddlZddlZddlZddlZddlmZmZej ej zZ e e Z eeZejjr ejrdZndZdZGddZGd d ZGd d Zd ZdZdS)z8 PEP-302 and PEP-451 importers for frozen applications. N)ArchiveReadErrorZlibArchiveReaderctj||ztjddS)N )sysstderrwrite)msgas (PyInstaller\loader\pyimod02_importers.pytracer &s8 q!!! cdS)N)r r s r r r +s rcddlm}tj|j}||}tjdd}|||dS)a Decode bytes representing source code and return the string. Universal newline support is used in the decoding. Based on CPython's implementation of the same functionality: https://github.com/python/cpython/blob/3.9/Lib/importlib/_bootstrap_external.py#L679-L688 r)detect_encodingNT)decoder translate)tokenizerioBytesIOreadlineIncrementalNewlineDecoderdecode) source_bytesrsource_bytes_readlineencodingnewline_decoders r _decode_sourcer/st)(((((J|44=455H244PPPO  ! !,"5"5hqk"B"B C CCrceZdZdZdZdS)PyiFrozenImporterStatea An object encapsulating extra information for PyiFrozenImporter, to be stored in `ModuleSpec.loader_state`. Having a custom type allows us to verify that module spec indeed contains the original loader state data, as set by `PyiFrozenImporter.find_spec`. c||_dS)N)pyz_entry_name)self entry_names r __init__zPyiFrozenImporterState.__init__Cs(rN)__name__ __module__ __qualname____doc__r&rrr r!r!=s- )))))rr!creZdZdZdZedZdZdZdZ dZ dZ d Z d Z dd Zd ZdZdZd S)PyiFrozenImportera8 Load bytecode of Python modules from the executable created by PyInstaller. Python bytecode is zipped and appended to the executable. NOTE: PYZ format cannot be replaced by zipimport module. The problem is that we have no control over zipimport; for instance, it does not work if the zip file is embedded into a PKG that is appended to an executable, like we create in one-file mode. This used to be PEP-302 finder and loader class for the ``sys.meta_path`` hook. A PEP-302 finder requires method find_module() to return loader class with method load_module(). However, both of these methods were deprecated in python 3.4 by PEP-451 (see below). Therefore, this class now provides only optional extensions to the PEP-302 importer protocol. This is also a PEP-451 finder and loader class for the ModuleSpec type import system. A PEP-451 finder requires method find_spec(), a PEP-451 loader requires methods exec_module(), load_module() and (optionally) create_module(). All these methods are implemented in this one class. ctjD]d} t|d|_t d|tj|n*#t $rYVt$rYawxYwtdt|jj |_ tj |_d|_dS)zY Load, unzip and initialize the Zip archive bundled with the executable. T) check_pymagicz$# PyInstaller: PyiFrozenImporter(%s)zCannot load frozen modules.N)rpathr _pyz_archiver removeIOErrorr ImportErrorsettockeys_threadRLock_lock _toc_tree)r$ pyz_filepaths r r&zPyiFrozenImporter.__init__\s H = =L $5lRV$W$W$W!!>> " " " " " " " " " " " " " " " " " "s'<AAc,t}|jD]}}|d}|}|j|r|D]}||i}O|ddD]}||i}d||d<~|S)N.)dictr5splitr0 is_package setdefault)r$treer%name_componentscurrentname_components r r=z(PyiFrozenImporter._build_pyz_prefix_treesvv( 2 2J(..s33OG ++J77 2&5EEN%00DDGGE'6crc&:EEN%00DDGG/1+,, rc||jvr@ |j|S#t$r}t d||d}~wwxYwt d|)N'PyiFrozenImporter cannot handle module )r5r0is_pep420_namespace_package Exceptionr3)r$fullnamees r _is_pep420_namespace_packagez.PyiFrozenImporter._is_pep420_namespace_packages tx  V a(DDXNNN a a a!"XH"X"XYY_`` aTTTUU U% AAAc||jvr@ |j|S#t$r}t d||d}~wwxYwt d|)NrL)r5r0rErNr3)r$rOrPs r rEzPyiFrozenImporter.is_packages tx  V a(33H=== a a a!"XH"X"XYY_`` aTTTUU UrRc |dkrtjdjS|j|S#t $r}t d||d}~wwxYw)z| Get the code object associated with the module. ImportError should be raised if module not found. __main__rLN)rmodules _pyi_main_cor0extractrNr3)r$rOrPs r get_codezPyiFrozenImporter.get_codes} ]:% <{:.;;$,,X66 6 ] ] ]TTTUU[\ \ ]s99 AAAc||jvr||r|dz }tjt |dtjdz} t|d5}| }dddn #1swxYwYt|S#t$rYnwxYwdStd|z)z Method should return the source code for the module as a string. But frozen modules does not contain source code. Return None, unless the corresponding source file was explicitly collected to the filesystem. .__init__r@z.pyrbNzNo module named ) r5rEosr/join SYS_PREFIXreplacesepopenreadrFileNotFoundErrorr3)r$rOfilenamefprs r get_sourcezPyiFrozenImporter.get_sources tx  =x(( (K'w||J0@0@bf0M0MPU0UVVH (D))-R#%7799L---------------%l333$    408;<< z-PyiFrozenImporter.find_spec..=s4>>$#=#=r)rEorigin loader_state)rsplit startswithr_ SYS_PREFIXLENrDr]raappendr^r5r rQimportlib._bootstrap_externalro_frozen_importlib ModuleSpecr/dirnamerlsubmodule_search_locationsrEr! has_location) r$rOr/targetr%modnameppartsrospecrsis_pkgs ` r rqzPyiFrozenImporter.find_specsb$  "ooc**2.G " "||J//mnn%Qx&!!""IE W%%% XXe__ )UWackmnoooE"  ?48# ?% 3X>>>   )8 4 4 44  , ,Z 8 8  D D D D D D%/$4PPPD /=n!2!2:!>!>??@>=== //D +K"":..,, +  / ;;    !  _/1wt?P?PQ[?\?\/]/].^D + rcdS)a PEP-451 loader.create_module() method for the ``sys.meta_path`` hook. Loaders may also implement create_module() that will return a new module to exec. It may return None to indicate that the default module creation code should be used. One use case, though atypical, for create_module() is to provide a module that is a subclass of the builtin module type. Most loaders will not need to implement create_module(). create_module() should properly handle the case where it is called more than once for the same spec/module. This may include returning None or raising ImportError. Nr)r$rs r create_modulezPyiFrozenImporter.create_module[s trc|j}t|jtr |jj}nt|jt r|jtsJ|jtd tj d}| dr |dd}| dr |dd}n$tdt|j||}|td|jd t%|d sJ|j*tj|jg|_t1||jdS) a PEP-451 loader.exec_module() method for the ``sys.meta_path`` hook. Loaders will have a new method, exec_module(). Its only job is to "exec" the module and consequently populate the module's namespace. It is not responsible for creating or preparing the module object, nor for any cleanup afterward. It has no return value. exec_module() will be used during both loading and reloading. exec_module() should properly handle the case where it is called more than once. For some kinds of modules this may mean raising ImportError every time after the first time the method is called. This is particularly relevant for reloading, where some kinds of modules do not support in-place reloading. Nr@rkr[iz:Module's spec contains loader_state of incompatible type: z Failed to retrieve bytecode for !__file__)__spec__ isinstancertr!r#rCrsrvr_rwr`r]raendswith RuntimeErrortyperYrrhasattrr}r/r|r__path__exec__dict__)r$moduler module_namebytecodes r exec_modulezPyiFrozenImporter.exec_modulels d')? @ @ w +:KK )4 0 0 w;))*55 5 5 5+mnn5==bfcJJK##F++ /)#2#. ##K00 /)#2#. u\`aear\s\suuvv v==--  RP$)PPPQQ Qvz*****  * A "wv??@FO Xv'''''rc"t||S)zG Return importlib.resource-compatible resource reader. )PyiFrozenResourceReader)r$rOs r get_resource_readerz%PyiFrozenImporter.get_resource_readers'tX666r)NN)r'r(r)r*r&propertyr>r=rQrErYrgrirlrqrrrrrr r,r,Hs&"""H""X""VVVVVV]]]$===0   ccccJ"7(7(7(r77777rr,c6eZdZdZdZdZdZdZdZdZ dS) ra Resource reader for importlib.resources / importlib_resources support. Supports only on-disk resources, which should cover the typical use cases, i.e., the access to data files; PyInstaller collects data files onto filesystem, and as of v6.0.0, the embedded PYZ archive is guaranteed to contain only .pyc modules. When listing resources, source .py files will not be listed as they are not collected by default. Similarly, sub-directories that contained only .py files are not reconstructed on filesystem, so they will not be listed, either. If access to .py files is required for whatever reason, they need to be explicitly collected as data files anyway, which will place them on filesystem and make them appear as resources. For on-disk resources, we *must* return path compatible with pathlib.Path() in order to avoid copy to a temporary file, which might break under some circumstances, e.g., metpy with importlib_resources back-port, due to: https://github.com/Unidata/MetPy/blob/a3424de66a44bf3a92b0dcacf4dff82ad7b86712/src/metpy/plots/wx_symbols.py#L24-L25 (importlib_resources tries to use 'fonts/wx_symbols.ttf' as a temporary filename suffix, which fails as it contains a separator). Furthermore, some packages expect files() to return either pathlib.Path or zipfile.Path, e.g., https://github.com/tensorflow/datasets/blob/master/tensorflow_datasets/core/utils/resource_utils.py#L81-L97 This makes implementation of mixed support for on-disk and embedded resources using importlib.abc.Traversable protocol rather difficult. So in order to maximize compatibility with unfrozen behavior, the below implementation is basically equivalent of importlib.readers.FileReader from python 3.10: https://github.com/python/cpython/blob/839d7893943782ee803536a47f1d4de160314f85/Lib/importlib/readers.py#L11 and its underlying classes, importlib.abc.TraversableResources and importlib.abc.ResourceReader: https://github.com/python/cpython/blob/839d7893943782ee803536a47f1d4de160314f85/Lib/importlib/abc.py#L422 https://github.com/python/cpython/blob/839d7893943782ee803536a47f1d4de160314f85/Lib/importlib/abc.py#L312 cddlm}||_|tjj|d|_dS)Nr)Pathr@)pathlibrimporterr_MEIPASSjoinpathrDr/)r$rrrrs r r&z PyiFrozenResourceReader.__init__sF        /DD&&/CA rcv||dS)Nr\)filesrrb)r$resources r open_resourcez%PyiFrozenResourceReader.open_resources,zz||$$X..33D999rcPt|j|S)N)strr/r)r$rs r resource_pathz%PyiFrozenResourceReader.resource_paths 49%%h//000rct||S)N)rris_file)r$r/s r is_resourcez#PyiFrozenResourceReader.is_resources*zz||$$T**22444rcbd|DS)Nc3$K|] }|jV dS)N)rr).0items r z z3PyiFrozenResourceReader.contents..s$==d ======r)riterdir)r$s r contentsz PyiFrozenResourceReader.contentss*==djjll&:&:&<&<====rc|jS)N)r/)r$s r rzPyiFrozenResourceReader.filess yrN) r'r(r)r*r&rrrrrrrr rrs{<BBB :::111555>>>rrct}tj|tjD]>}t |dr,|jdkr!tj|n?g}ttjD]O}t|dddkr8tj|||vr||Ptj t| |tj d_ n#t$rYnwxYwtj dkrtdSdS)a Install PyiFrozenImporter class into the import machinery. This function installs the PyiFrozenImporter class into the import machinery of the running process. The importer is added to sys.meta_path. It could be added to sys.path_hooks, but sys.meta_path is processed by Python before looking at sys.path! The order of processing import hooks in sys.meta_path: 1. built-in modules 2. modules from the bundled ZIP archive 3. C extension modules 4. Modules from sys.path r'WindowsRegistryFinderN PathFinderrU) )r,r meta_pathrxrr'r1reversedgetattrextendrV __loader__rN version_info_fixup_frozen_stdlib)rr path_finderss r installrsi !""HM"""   4 $ $ :Q)Q  M  & & & E L''** 4T * *l : * M  & & &<' *##D)))M,//000  -5 J**       7"sD)) D65D6c0ddl}tjs( tjt_n#t$rYnwxYwtjD]\}}||s||}|j j }|j }|r|dz }tj jtjg|dRdz}t!|ds ||_n#t$rYnwxYw|j |dkr||_dS)Nrr[r@rkrzimportlib._bootstrap)_impr _stdlib_dirrAttributeErrorrVitems is_frozenis_frozen_packagerrtorignamer]r/r^rDrrre)rrrrrt orig_nameres r rrsXKKK ? !lCOO    D  #{0022-- V~~k**  '' 443  )  %  $I7< Dys/C/CDDDvMvz**  "*!       -Y:P-P -$,L !7--s) 66)C11 C>=C>)r*rr]rrzr7pyimod01_archiverrrrar_lenrwrimp_new_moduleflagsverboserr rr!r,rrrrrr rsy @@@@@@@@ \BF " J c9       D D D))))))))a7a7a7a7a7a7a7a7H 22222222j000n(-(-(-(-(-r