3 (hk @szddlZddlZddlZddlmZddlZddlmZyejdj Z Wnej k rbdZ YnXd ddZ dd Z dS) N) ModuleType)filterzpython-monkey-businessTcs|dkrHtttfr }dn(tttfrHtddDrH}ddkrbtjt|dSdkrpj t|ttfrtdd|Drfdd|DSt|ttfst dyt |}WnDt k rt dd d jtdt |d d|j j giYnXr,t |d dkr,dSytj|Wnt k rT|YnXtjfd d}||_|_tjrtj|rt|dr|jrt|}nt|}t||t |S)a  Decorator for monkeypatching functions on modules and classes. Example:: # This replaces FooClass.bar with our method @monkeybiz.patch(FooClass) def bar(original_bar, *args, **kwargs): print "Patched!" return original_bar(*args, **kwargs) # This replaces FooClass.bar and foomodule.bar with our method @monkeybiz.patch([FooClass, foomodule]) def bar(original_bar, *args, **kwargs): #... The first argument to ``monkeybiz.patch`` can be either a module, a class, or a list of modules and/or classes. The decorator also takes optional ``name`` and ``avoid_doublewrap`` keyword arguments. If ``name`` is omitted, the name of the function being patched will be the name of the function being decorated. If ``avoid_doublewrap`` is True (the default), then functions and methods can only be patched once using this function. Use ``monkeybiz.unpatch()`` to revert a monkey-patched function to its original. NcSsg|]}t|ttfqS) isinstancertype).0irrF/tmp/pip-install-q3hcpn_q/python-monkey-business/monkeybiz/__init__.py /szpatch..)objnameavoid_doublewrapcSsg|]}t|ttfqSr)rrr)rrrrr r 9scsg|]}t|dqS))funcr r r )patch)ro)r rr rr r :sz_Argument passed to @patch decorator must be a class or module, or a list of classes and modulesz%(func_repr)s does not existZ func_repr. __module__wrappercsf||S)Nr)argskwargs)roriginal_callablerr rWszpatch..wrapperim_self)rrrlisttupleall functoolspartialr__name__ ValueErrorgetattrAttributeError TypeErrorjoinrsixget_method_functionwrapsoriginalrPY2inspectisclasshasattrr classmethod staticmethodsetattr)rr r r callrr)r rr rr rsR       rcCst||t||jdS)z4 Undo the effects of patch(func, obj, name) N)r-rr&)r r rrr unpatchlsr/)NNNT)rr( pkg_resourcestypesrr#Z six.movesrget_distributionversion __version__DistributionNotFoundrr/rrrr s    \