¯wg<'dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z dZ dZdZej!d Zd Zgd Zd Zd ZeGddZGddZddZedk(reyy)a4 Freeze Python packages. Freezing makes it possible to ship arbitrary Python modules as part of a C++ library. The Python source of the module is compiled to bytecode and written to `.c` files, to be imported by Python's built-in FrozenImporter. In a normal Python installation, FrozenImporter is only used to bootstrap the initialization of the import machinery. Python's importers are defined in Python (see `_bootstrap.py` and `_bootstrap_external.py`) but need to be retrieved before any importers are available. Freezing the module bytecode resolves this circular dependency. This script will freeze the Python standard library. It produces two things: - Bytecode files: A set of `.c` that define C variables containing Python bytecode. - Main file: A `main.c` file listing all of these modules in the right form to be consumed by FrozenImporter. The library that wishes to these modules make them available to the local Python instance by extending `PyImport_FrozenModules` appropriately (see https://docs.python.org/3/c-api/import.html#c.PyImport_FrozenModules). N) dataclass)Path)Listzz#include z // Compiled standard library modules. These should be appended to the existing // `PyImport_FrozenModules` that ships with CPython. struct _frozen {}[] = {{ _PyImport_FrozenModulesz {0, 0, 0} /* sentinel */ }; ) dbmcursestkinterr testtests idle_testz__phello__.foo.pyz _bootstrap.pyz_bootstrap_external.pycBtjfd}|S)Ncx|dxjdz c_|i|}|dxjdzc_|S)Nr)indent)argskwargsretfns X/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/torch/utils/_freeze.pywrapperzindent_msg..wrapperPs; Q!$!&! Q! ) functoolswraps)rrs` r indent_msgrOs%__R Nrc6eZdZUeed<eed<eed<eed<y) FrozenModule module_namec_namesizebytecodeN)__name__ __module__ __qualname__str__annotations__intbytesrrrrZs K IOrrceZdZdefdZdedefdZdZdZ de fd Z ded efd Z e ded efd Zd ed edeefdZdedej$fdZe ded efdZy)Freezerverbosec.g|_d|_||_y)Nr)frozen_modulesrr,)selfr,s r__init__zFreezer.__init__gs24 $ rpathcodec|jsyt|jD]}tddt|d|y)Nz )end )r,rangerprint)r/r1r2is rmsgz Freezer.msglsC||  t{{# "A &b ! " avrc vttDcgc]}d|d }}|Dcgc],}ttjj ||d.}}t j|}|jD]}|j|t||D]}|jycc}wcc}w)z~ Write the `.c` files containing the frozen bytecode. Shared frozen modules evenly across the files. bytecode_z.cwN) r7NUM_BYTECODE_FILESopenosr1join itertoolscycler. write_frozennextclose) r/ install_rootr9bytecode_file_namesnamebytecode_filesitmfs rwrite_bytecodezFreezer.write_bytecodexs ;@@R:STQ1#R0TTDW <@DlD13 7  __^ ,$$ +A   ab * +  A GGI U s B11B6c jttjj|dd5}|j t |j D]!}|j d|jd#|j tj||j D];}|j d|jd|jd|jd=|j t|r*|j t|j td d d y #1swYy xYw) zNWrite the `main.c` file containing a table enumerating all the frozen modules.zmain.cr=zextern unsigned char z[]; z {"z", z, z}, N)r?r@r1rAwrite MAIN_INCLUDESr.rMAIN_PREFIX_TEMPLATEformatrr MAIN_SUFFIX FAKE_PREFIX)r/rGoss symbol_nameoutfprLs r write_mainzFreezer.write_mains "'',,|X6 < ) KK &(( E 3AHH:UCD E KK,33K@ A(( Q eAMM?#ahhZr!&&OP Q KK $ K( K( ) ) )s C5D))D2rLc8|jd|jdtdt|jdD]G}|jdt |j||dzD]}|jd|zI|jdy) zz*Freezer.compile_package..sUUUZZ=8UsSP)r`rI DENY_LISTr:anyiterdirrc)r/r1r^is_package_dirris rrazFreezer.compile_packages{{}} 99 ! HHT3  UdllnUU HHT3   s\\^ 7E   e%5 6 7r file_pathreturnc&|j|j}|jdk(r7|jj}|jjj}n"|j}|jj}t ||gzS)Nrg) relative_toparentrIpartsstemlist)r/rqr^normalized_pathmodule_basename module_parents rget_module_qualnamezFreezer.get_module_qualnames $//0@0G0GH   = 0 .4499O+2299??M-22O+2288MM"o%666r file_contentc(t}t||dS)Nexec) PATH_MARKERcompile)r/r} path_markers rcompile_stringzFreezer.compile_strings" |[&99rcx|jsJ|jdk7r|j|dy|jtvr|j|dy|j|d|j ||}dj |}d|z}t|5}|j|j}dddtj}t|} |jdk(r| } |jjtd j ||| |y#1swYrxYw) zw Compile a Python source file to frozen bytecode. Append the result to `self.frozen_modules`. z.pyNNreF__M_rg.)is_filesuffixr:rIrmr|rAr?rreadmarshaldumpsr\r.appendr) r/r1r^module_qualnamemodule_mangled_namersrc_filecor!r s rrbzFreezer.compile_files||~~ ;;%  HHT3   99 ! HHT3   s2249IJ"ii8++ $Z 68$$X]]_5B 6==$8} 99 %5D "" /2FD( K  6 6s  D00D9N)r"r#r$boolr0rr%r:rNrYrrDrcrrarr|typesCodeTyperrbr)rrr+r+fs%%    C  " )l6667D7D77$7T7T7dSVi7(:3:5>>:    rr+ctjd}|jddd|jddd |jd d d |jddd |jdddd|j}t |j }|j D]y}t|}|jrJtj|dz s2t|jdD]}|j||h|j||{|j|j|j|j|j |j"y)NzCompile py source) descriptionpaths*zPaths to freeze.)nargshelpz --verbose store_truezPrint debug logs)actionrz --install-dirz --install_dirz#Root directory for all output files)rz--ossz5If it's OSS build, add a fake _PyImport_FrozenModulesz --symbol-namez --symbol_namez6The name of the frozen module array symbol to generate_PyImport_FrozenModules_torch)rdefaultrg)argparseArgumentParser add_argument parse_argsr+r,rrr`existssortedglobrcrN install_dirrYrVrW)parserrrMpr1mods rmainrs^  $ $1D EF s1CD  L?QR /T  D   E/     D A ZZ 'Aw ;;=TM-A!B diin- )sC( ) NN4 & 'T%%&LL!!488T-=-=>r__main__)rrN)__doc__rrrBrr@r dataclassesrpathlibrtypingrrrQrRrSrUrTrmr>rrr+rr"r)rrrs. !-   #))*CD    $  R R j$?N zFr