g dZddlZddlZddlZddlmZmZdhZddhZGddZ Gd d Z ddd d ed e dedefdZ dS)zETools useful for creating decorators, and other high-level callables.N)TypeCallable__new____init_subclass__ __prepare__c2eZdZdZdZfdZdZdZxZS)_lift_to_methodaaA decorator that ensures that an input callable object implements ``__get__``. It is returned unchanged if so, otherwise it is turned into the default implementation for functions, which makes them bindable to instances. Python-space functions and lambdas already have this behavior, but builtins like ``print`` don't; using this class allows us to do:: wrap_method(MyClass, "maybe_mutates_arguments", before=print, after=print) to simply print all the arguments on entry and exit of the function, which otherwise wouldn't be valid, since ``print`` isn't a descriptor. _methodcjt|dr|St|S)N__get__)hasattrsuperr)clsmethod __class__s M/granary/edev/quantum/lib/python3.11/site-packages/qiskit/utils/classtools.pyrz_lift_to_method.__new__-s0 69 % % Mwws###c ||urdS||_dSNr )selfrs r__init__z_lift_to_method.__init__2s T>> F rcH||jStj|j|Sr)r types MethodType)robjobjtypes rr z_lift_to_method.__get__8s& ;<  c222r) __name__ __module__ __qualname____doc__ __slots__rrr __classcell__)rs@rr r sg  I$$$$$  3333333rr c&eZdZdZdZddZddZdS)_WrappedMethodaDescriptor which calls its two arguments in succession, correctly handling instance- and class-method calls. It is intended that this class will replace the attribute that ``inner`` previously was on a class or instance. When accessed as that attribute, this descriptor will behave it is the same function call, but with the ``function`` called before or after. )_method_decorator_method_has_getr _before_afterNct|ttfrt||_nt|t|r |j|_nVt |ddt vr t|_n1t |ddtvr t|_n t|_|||fnd}|||fnd}t|t|r+|j |_ ||j z|_ |j |z|_ n||_ ||_ ||_ t|j d|_ dS)Nrr ) isinstance classmethod staticmethodtyper&getattr_MAGIC_STATICMETHODS_MAGIC_CLASSMETHODSr r r(r)rr')rrbeforeafters rrz_WrappedMethod.__init__LsN f{L9 : : 5%)&\\D " " T + + 5%+%=D " " VZ . .2F F F%1D " " VZ . .2E E E%0D " "%4D "6<6H$((0022b494E''..002 fd4jj ) ) "!>DL!FN2DL -%/DKK!DLDK!DL 't|Y??rcjrjnjtjfd}|S)NcjD]}||i||i|}jD]}||i||Sr)r(r r))argskwargscallbackretvalrrrrs routz#_WrappedMethod.__get__..outls L @ @.  g..?????VT,V,,F K @ @.  g..?????Mr)r'r r functoolswraps)rrrr;rs``` @rr z_WrappedMethod.__get__fst8<7K]%%c7333QUQ]          !   r)NNr)rrr r!r"rr r+rrr%r%AsRYI@@@@4rr%)r3r4rnamer3r4c ptj||}t||t|||dS)aWrap the functionality the instance- or class method ``cls.name`` with additional behavior ``before`` and ``after``. This mutates ``cls``, replacing the attribute ``name`` with the new functionality. This is useful when creating class decorators. The method is allowed to be defined on any parent class instead. If either ``before`` or ``after`` are given, they should be callables with a compatible signature to the method referred to. They will be called immediately before or after the method as appropriate, and any return value will be ignored. Args: cls: the class to modify. name: the name of the method on the class to wrap. before: a callable that should be called before the method that is being wrapped. after: a callable that should be called after the method that is being wrapped. Raises: ValueError: if the named method is not defined on the class or any parent class. N)inspectgetattr_staticsetattrr%)rr>r3r4rs r wrap_methodrCxs:2 #C . .F C~ffe<<=====r) r!r<r@rtypingrrr1r2r r%strrCr+rrrFsLK !!!!!!!!"{*M:!3!3!3!3!3!3!3!3H44444444n=ATX>>>T>>>>>>>>>r