U @fU @sddlmZz ddlZWnek r4ddlZYnXddlZzddlmZWnek rfdZYnXddlZddddgZ dd d hZ Gd d d Z d dZ ddZ ddZddZddZiZddZddZddZd1ddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zgdfd-d.Ze d/krej!"dee#ej!d$Z$e%e$&ej!dd0Z'e(e'W5QRXdS)2) annotationsNregister_at_forkinjectimport_patched monkey_patchis_monkey_patched __builtins____file____name__c@s*eZdZdZd ddZddZddZd S) SysModulesSaverzClass that captures some subset of the current state of sys.modules. Pass in an iterator of module names to the constructor.cCsi|_t|j|dSN)_savedimp acquire_locksave)self module_namesr r J/var/www/html/chatgpt/venv/lib/python3.8/site-packages/eventlet/patcher.py__init__szSysModulesSaver.__init__cGs"|D]}tj|d|j|<qdS)z&Saves the named modules to the object.N)sysmodulesgetr)rrmodnamer r rr szSysModulesSaver.savec Cs^zN|jD]>\}}|dk r(|tj|<q z tj|=Wq tk rHYq Xq W5tXdS)zRRestores the modules that the saver knows about into sys.modules. N)r release_lockritemsrrKeyError)rrmodr r rrestore%s   zSysModulesSaver.restoreN)r )r __module__ __qualname____doc__rrrr r r rr s r c Gsd|}|tjkrtj|S|s>ttttt}tdd|D}|||D]\}}|tj|<q^tj |dt tj D] }| |drtj |dqzVt|ii|ddd}|dk rt|D]}|tkrt||||<q|tj|<W5| X|S)aBase method for "injecting" greened modules into an imported module. It imports the module specified in *module_name*, arranging things so that the already-imported modules in *additional_modules* are used when *module_name* makes its imports. **Note:** This function does not create or change any sys.modules item, so if your greened module use code like 'sys.modules["your_module_name"]', you need to update sys.modules by yourself. *new_globals* is either None or a globals dictionary that gets populated with the contents of the *module_name* module. This is useful when creating a "green" version of some other module. *additional_modules* should be a collection of two-element tuples, of the form (, ). If it's not specified, a default selection of name/module pairs is used, which should cover all use cases but may be slower because there are inevitably redundant or unnecessary imports. Z__patched_module_cSsg|] \}}|qSr r ).0namemr r r [szinject..N.)rr_green_os_modules_green_select_modules_green_socket_modules_green_thread_modules_green_time_modulesr rpoplistkeys startswithr __import__splitdir __excludegetattr) module_nameZ new_globalsadditional_modulesZ patched_namesaverr$rZimported_module_namemoduler r rr6s>        cOst|df|t|S)zImports a module in a way that ensures that the module uses "green" versions of the standard library modules, so that everything works nonblockingly. The only required argument is the name of the module to be imported. N)rtupler)r7r8Zkw_additional_modulesr r rr}s cs4s"tttttfdd}|S)a Decorator that returns a version of the function that patches some modules for the duration of the function call. This is deeply gross and should only be used for functions that import network libraries within their function bodies that there is no way of getting around.csFt}D]\}}|||tj|<q z||WS|XdSr)r rrrr)argskwr9r$rr8funcr rpatcheds   zpatch_function..patched)r)r*r+r,r-)r?r8r@r r>rpatch_functions rAcsfdd}|S)auKind of the contrapositive of patch_function: decorates a function such that when it's called, sys.modules is populated only with the unpatched versions of the specified modules. Unlike patch_function, only the names of the modules need be supplied, and there are no defaults. This is a gross hack; tell your kids not to import inside function bodies!cs>t}D]}t|tj|<q z||WS|XdSr)r originalrrr)r<r=r9r$r?rr rr@s z)_original_patch_function..patchedr )r?rr@r rCr_original_patch_functionsrDcCsd|}|tjkrtj|St|f}tj|dddd}||krh||}||t|tj|<zNt|ii| ddd}|dkrt |d st |j j d|j _ |tj|<W5|Xtj|S) zd This returns an unpatched version of a module; this is useful for Eventlet itself (i.e. tpool).Z__original_module_N_thread threading)rFqueuer'r()QueuerG _threading)rrrr r.rrBrr2r3hasattrrDrHr)r original_namer9deps dependencyZreal_modr r rrBs(      rBc Kstjddddddddd h }d |kr4d|kr4tz|d }Wntk rVYn X||d<|d d }|D]}||krttd |qt|d krd|k}|D]8}|dkr| |d|dkr| |d| ||q|drt dst g}dt fdtfdtfdtfdtfdtfdtfd tffD]2\}}||r0t |s0||7}dt |<q0|drt dsz ddlm} | dt d<Wntk rYnXtd} tz|D]\}} tj |} | d krt|} | j D](} t!| | d }|d k rt"| | |qt!| dg}|D]} t#| | r.t$| | q.|dkrt%r| j&j'| fdd}|qW5tXdd l(}td}||j)_*dd l+}|j,|_-dd l.}|j/|_0d S)aGlobally patches certain system modules to be greenthread-friendly. The keyword arguments afford some control over which modules are patched. If no keyword arguments are supplied, all possible modules are patched. If keywords are set to True, only the specified modules are patched. E.g., ``monkey_patch(socket=True, select=True)`` patches only the select and socket modules. Most arguments patch the single module of the same name (os, time, select). The exceptions are socket, which also patches the ssl module if present; and thread, which patches thread, threading, and Queue. It's safe to call monkey_patch multiple times. osselectsocketthreadtimeZpsycopgMySQLdbbuiltins subprocess __builtin__allNz4monkey_patch() got an unexpected keyword argument %rTFr)psycopg2_patcherrFZ __deleted__cs4dgfdd}fdd}t||ddS)Ncsdd<jd<dS)N_activer)rYr  _global_dict_patched_prefork_activer r before_forkNs z?monkey_patch..fix_threading_active..before_forkcsdd<dS)NrrYr r )r[r]r r after_forkRsz>monkey_patch..fix_threading_active..after_fork)beforeafter_in_parentr)r[r\r^r_r rZrfix_threading_activeFsz*monkey_patch..fix_threading_activerE)1eventletZhubsZget_hubAssertionErrorr.rr0 TypeErrorvalues setdefaultalready_patchedr_green_existing_locksr)r*r+r,r-_green_MySQLdb_green_builtins_green_subprocess_modulesZeventlet.supportrXZmake_psycopg_green ImportErrorrBrrrrrr2Z __patched__r6setattrrJdelattrrcurrent_thread __globals__Zimportlib._bootstrap _bootstraprErF_PyRLockRLockrG_PySimpleQueue SimpleQueue)onZ accepted_argsbZ default_onkrZmodules_to_patchr$Zmodules_functionrXrIrZorig_mod attr_nameZ patched_attrZdeletedrb importlibrQrFrGr r rrs                     cCs|tkpt|ddtkS)a[Returns True if the given module is monkeypatched currently, False if not. *module* can be either the module itself or its name. Based entirely off the name of the module, so if you import a module some other way than with the import keyword (including import_patched), this might not be correct about that particular module.r N)rhr6)r:r r rrnscsddl}ddl}ddl}ddl}t||jjfdd}t t j |d|j kr`dS| tfdd|D}|rddl}|d}|d |d d dS) aMake locks created before monkey-patching safe. RLocks rely on a Lock and on Python 2, if an unpatched Lock blocks, it blocks the native thread. We need to replace these with green Locks. This was originally noticed in the stdlib logging module.rNcs t|Sr)_convert_py3_rlock)Zold_lock)tidr rupgradesz&_green_existing_locks..upgradeZPYTEST_CURRENT_TESTcsh|]}t|r|qSr ) isinstance)r#o) rlock_typer r s z(_green_existing_locks..rcz{} RLock(s) were not greened,z= to fix this error make sure you run eventlet.monkey_patch() z#before importing any other modules.)gcrNrFeventlet.green.threadtypertgreenrQ get_ident_upgrade_instancesrrenvironZcollectlenZ get_objectsloggingLoggererrorformat)rrNrFrcr~Zremaining_rlocksrloggerr )rr}rrizs(      ric s,dkr idkri|t|<fdd}t|trnt|D] \}}||}|dk rL|||<qLt|trt|D] \} }||}|dk r||| <qz t|} Wntk rYnfXz6t| D]$\}}||}|dk rt|||qWn,ddl } | d} | j dddYnXdS) a| Starting with a Python object, find all instances of ``klass``, following references in ``dict`` values, ``list`` items, and attributes. Once an object is found, replace all instances with ``upgrade(found_object)``, again limited to the criteria above. In practice this is used only for ``threading.RLock``, so we can assume instances are hashable. NcsXt|krdSt|r@|kr*|S|}||<|Snt|dSdSr)idrr)objnewklass old_to_newr~visitedr rupgrade_or_traverses  z/_upgrade_instances..upgrade_or_traverserrczAn exception was thrown while monkey_patching for eventlet. to fix this error make sure you run eventlet.monkey_patch() before importing any other modules.T)exc_info) rrdictr/r enumeratevarsrernrr exception) containerrr~rrrryvriZcontainer_varsrrr rrrs>        rcCsddl}ddlm}|}t|dr0t|ds8td||_d}|rb|| d}qD|rv| d}|r||_ |S) z Convert a normal RLock to one implemented in Python. This is necessary to make RLocks work with eventlet, but also introduces bugs, e.g. https://bugs.python.org/issue13697. So more of a downgrade, really. rN) allocate_lock_block_ownerzINTERNAL BUG. Perhaps you are using a major version of Python that is unsupported by eventlet? Please file a bug at https://github.com/eventlet/eventlet/issues/newFT) rFrrrsrJ RuntimeErrorr _is_ownedreleaseacquirer)oldr}rFrrZacquiredr r rr|s& r|cCsddlm}d|fgS)NrrNrN)eventlet.greenrNrr r rr)s r)cCs4ddlm}d|fg}ddlm}|d|f|S)Nr)rOrO) selectorsr)rrOrappend)rOrrr r rr* s    r*cCsNddlm}zddlm}d|fd|fgWStk rHd|fgYSXdS)Nr)rP)sslrPr)rrPrrm)rPrr r rr+s   r+cCsddlm}d|fgS)NrrUrU)rrUrr r rrls rlcCs:ddlm}ddlm}ddlm}d|fd|fd|fgS)Nr)rH)rQ)rFrGrErF)rrHrQrF)rHrQrFr r rr,!s   r,cCsddlm}d|fgS)NrrRrR)rrRrr r rr-(s r-cCs6zddlm}d|fgWStk r0gYSXdS)NrrSrS)rrSrmrr r rrj-s   rjcCs6zddlm}d|fgWStk r0gYSXdS)NrbuiltinrT)rrrmrr r rrk5s   rkcs,|dkrj}|fdd|DdS)aCopy properties from *source* (assumed to be a module) to *destination* (assumed to be a dict). *ignore* lists properties that should not be thusly copied. *srckeys* is a list of keys to copy, if the source's __all__ is untrustworthy. Ncs*i|]"}|ds|ks|t|qS)__)r1r6)r#r$ignoresourcer r Gs  z$slurp_properties..)__all__update)r destinationrZsrckeysr rrslurp_properties=s r__main__exec)NN)) __future__r_imprrmrrNrrcrr5r rrrArDrBrhrrrirr|r)r*r+rlr,r-rjrkrr argvr.openfcompilereadcoderr r r rsP      G (  * @