a }=ic@sjdZddlZddlZddlmZejZddZefddZefdd Zd d Z Gd d d e Z ddZ dS)a`Decorator and context manager for saving and restoring flag values. There are many ways to save and restore. Always use the most convenient method for a given use case. Here are examples of each method. They all call ``do_stuff()`` while ``FLAGS.someflag`` is temporarily set to ``'foo'``:: from absl.testing import flagsaver # Use a decorator which can optionally override flags via arguments. @flagsaver.flagsaver(someflag='foo') def some_func(): do_stuff() # Use a decorator which can optionally override flags with flagholders. @flagsaver.flagsaver((module.FOO_FLAG, 'foo'), (other_mod.BAR_FLAG, 23)) def some_func(): do_stuff() # Use a decorator which does not override flags itself. @flagsaver.flagsaver def some_func(): FLAGS.someflag = 'foo' do_stuff() # Use a context manager which can optionally override flags via arguments. with flagsaver.flagsaver(someflag='foo'): do_stuff() # Save and restore the flag values yourself. saved_flag_values = flagsaver.save_flag_values() try: FLAGS.someflag = 'foo' do_stuff() finally: flagsaver.restore_flag_values(saved_flag_values) We save and restore a shallow copy of each Flag object's ``__dict__`` attribute. This preserves all attributes of the flag, such as whether or not it was overridden from its default value. WARNING: Currently a flag that is saved and then deleted cannot be restored. An exception will be raised. However if you *add* a flag after saving flag values, and then restore flag values, the added flag will be deleted with no errors. N)flagscOs|stfi|St|dkrZt|drZ|r6td|d}t|rPtdt|iS|D]l}t|t rxt|dkrtd|f|\}}t|t j std|f|j |vrtd|j |||j <q^tfi|S)z7The main flagsaver interface. See module doc for usage.rz?It's invalid to specify both positional and keyword parameters.z2@flagsaver.flagsaver cannot be applied to a class.z+Expected (FlagHolder, value) pair, found %rzCannot set --%s multiple times) _FlagOverriderlencallable ValueErrorinspectisclass TypeError_wrap isinstancetuplerZ FlagHoldername)argskwargsfuncargZholdervaluerg/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/absl/testing/flagsaver.py flagsaverFs*     rcsfddDS)a_Returns copy of flag values as a dict. Args: flag_values: FlagValues, the FlagValues instance with which the flag will be saved. This should almost never need to be overridden. Returns: Dictionary mapping keys to values. Keys are flag names, values are corresponding ``__dict__`` members. E.g. ``{'key': value_dict, ...}``. csi|]}|t|qSr)_copy_flag_dict).0r flag_valuesrr kz$save_flag_values..rrrrrsave_flag_valuesas rcCs^t|}|D]L}||}|dur.t||q ||j|dkrN|d||_|||_q dS)aRestores flag values based on the dictionary of flag values. Args: saved_flag_values: {'flag_name': value_dict, ...} flag_values: FlagValues, the FlagValues instance from which the flag will be restored. This should almost never need to be overridden. N_value)listgetdelattrr__dict__)Zsaved_flag_valuesrZnew_flag_namesrZsavedrrrrestore_flag_valuesns  r$cstfdd}|S)a[Creates a wrapper function that saves/restores flag values. Args: func: function object - This will be called between saving flags and restoring flags. overrides: {str: object} - Flag names mapped to their values. These flags will be set after saving the original flag state. Returns: return value from func() cs@tfi|i|WdS1s20YdS)z/Wrapper function that saves and restores flags.N)r)rrr overridesrr_flagsaver_wrappersz!_wrap.._flagsaver_wrapper) functoolswraps)rr&r'rr%rr s r c@s0eZdZdZddZddZddZdd Zd S) rzOverrides flags for the duration of the decorated function call. It also restores all original values of flags after decorated method completes. cKs||_d|_dSN) _overrides_saved_flag_values)selfr&rrr__init__sz_FlagOverrider.__init__cCst|rtdt||jS)Nz'flagsaver cannot be applied to a class.)r r r r r+)r-rrrr__call__s z_FlagOverrider.__call__cCs@tt|_ztjfi|jWnt|jtYn0dSr*)rFLAGSr,_set_attributesr+r$)r-rrr __enter__s   z_FlagOverrider.__enter__cCst|jtdSr*)r$r,r0)r-exc_type exc_value tracebackrrr__exit__sz_FlagOverrider.__exit__N)__name__ __module__ __qualname____doc__r.r/r2r6rrrrrs  rcCs&|j}|j|d<t|j|d<|S)aReturns a copy of the flag object's ``__dict__``. It's mostly a shallow copy of the ``__dict__``, except it also does a shallow copy of the validator list. Args: flag: flags.Flag, the flag to copy. Returns: A copy of the flag object's ``__dict__``. r validators)r#copyrr r;)flagr<rrrrs  r) r:r(r Zabslrr0rrr$r objectrrrrrrs/