o qoi( @sZdZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZddlmZddlmZGd d d ejZd ed ed e fddZejddGdddZded ejfddZdeded e fddZded ede ejdedef ddZ d ej!d!ed dfd"d#Z"Gd$d%d%Z#dS)&z;Utilities to use command line flags with Fiddle Buildables.N)AnyDictOptionalTuple)config)daglish)daglish_extensions)module_reflection) serializationc@s*eZdZdZdZdZdedefddZdS) ImportDottedNameDebugContextz2Context of importing a sumbol, for error messages.namereturncCs,|tjkr d|S|tjksJd|S)Nz Could not init a buildable from zCould not load fiddler )r BASE_CONFIGFIDDLER)selfrrP/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/absl_flags/utils.py error_prefix*s   z)ImportDottedNameDebugContext.error_prefixN)__name__ __module__ __qualname____doc__rrstrrrrrrr $s r rmoderc Cs*|d}t|dkrt||dtt|dddD]l}z\td|d|}t||dD]B\}}zt ||}Wq6t yxdd d t |D}d|d||}|||} t||d |d | d |dw|WSt y|dkrYqwt d|d)aReturns the Python object with the given dotted name. Args: name: The dotted name of a Python object, including the module name. mode: Whether we're looking for a base config function or a fiddler. Returns: The named value. Raises: ValueError: If `name` is not a dotted name. ModuleNotFoundError: If no dotted prefix of `name` can be imported. AttributeError: If the imported module does not contain a value with the indicated name. .r z3: Expected a dotted name including the module name.r rN, css"|] }|dst|VqdS)_N) startswithrepr).0nrrr Qs   z%import_dotted_name..z : module z has no attribute ; available names: zNo module named ) splitlen ValueErrorrrange importlib import_modulejoin enumerategetattrAttributeErrordirModuleNotFoundError) rr name_piecesivaluej name_pieceavailable_names module_name failing_namerrrimport_dotted_name1sD      r:T)frozenc@sbeZdZUdZeed<eeedfed<ee eefed<e dZ e deddfd d Zd S) CallExpressiona(Parsed components of a call expression (or bare function name). Examples: >>> CallExpression.parse("fn('foo', False, x=[1, 2])") CallExpression(func_name='fn', args=('foo', False'), kwargs={'x': [1, 2]}) >>> CallExpression.parse("fn") # Bare function name: empty args/kwargs. CallExpression(func_name='fn', args=()), kwargs={}) Attributes: func_name: The name fo the function that should be called. args: Parsed values of positional arguments for the function. kwargs: Parsed values of keyword arguments for the function. func_name.argskwargsz+(?P[\w\.]+)(?:\((?P.*)\))?r4rc Cst|j|}|durtd||ddur"t|ddiSt|}t|tj rGt |j dkrGt|j dtj rGt|j dj tjsNtd||j dj }g}|jD]1}t|tjretdz |t|WqYty}ztd t|d |d |d}~wwi}|jD]2} | jdurtd z t| j || j<Wqty}ztd t| j d |d |d}~wwt|dt||S) aReturns a CallExpression parsed from a string. Args: value: A string containing positional and keyword arguments for a function. Must consist of an open paren followed by comma-separated argument values followed by a close paren. Argument values must be literal constants (i.e., must be parsable with `ast.literal_eval`). var-positional and var-keyword arguments are not supported. Raises: SyntaxError: If `value` is not a simple call expression with literal arguments. Nz2Expected a function name or call expression; got: r>r=rr rz*args is not supported.z.Expected arguments to be simple literals; got z (while parsing )z**kwargs is not supported.)re fullmatch _PARSE_RE SyntaxErrorgroupr<astparse isinstanceModuler'bodyExprr4Callr>Starredappend literal_evalr(unparsekeywordsargtuple) clsr4mnode call_noder>rRexcr?kwargrrrrG{sp        zCallExpression.parseN)rrrrr__annotations__rrrrrAcompilerC classmethodrGrrrrr<ds  r<pathcCs(|ds|dsd|}t|S)z@Parses a path into a list of either attributes or index lookups.[r)r r parse_path)r]rrrr_s  r_r4cCsv|dkrdS|dkrdSzt|WSty:}ztd|d|d|d|d |d |d |d }~ww) zFParses a string `value` (e.g. from the command line) to a Python type.falseFtrueTzCould not parse literal value "z" while trying to set "zR". Does a string need to be quoted? For example, you might want to pass --fdl_set=z='z' instead of --fdl_set==rN)lowerrFrO Exceptionr()r4r]errr parse_values0   rf function_namemodule allow_importsfailure_msg_prefixc Cst||r t||S|r-zt||dWSty,}z t|d|d||d}~wwt|}t|d|dd|d)aReturns function that produces `fdl.Buildable` from its name. Args: function_name: The name of the function. mode: Whether we're looking for a base config function or a fiddler. module: A common namespace to use as the basis for finding configs and fiddlers. May be `None`; if `None`, only fully qualified Fiddler imports will be used (or alternatively a base configuration can be specified using the `--fdl_config_file` flag.) allow_imports: If true, then fully qualified dotted names may be used to specify configs or fiddlers that should be automatically imported. failure_msg_prefix: Prefix string to prefix log messages in case of failures. Returns: The named value. )r z: Nr%rr)hasattrr.r:r1r(r find_base_config_like_thingsr,)rgrrhrirjrer7rrrresolve_function_references$    rncfg assignmentc Cs|jddd\}}t|^}}tt|}z |D]}||}qWnty7}z td|d|d}~wwt||d} z&t |t j rOt ||j | WdSt |t jr]| ||j<WdStd|d ty|}z td |d |d|d}~ww) zSet an attribute's value. Args: cfg: A `fdl.Buildable` whose attribute is to be overridden. assignment: String representing attribute's override expression. Of the form `attribute=value`. rbr )maxsplitzInvalid path "z".N)r4r]zUnexpected path element rzCould not set "z" to ")r&r_typingcastrfollowrdr(rfrHrAttrsetattrrKeykey) rorpr]r4parentslastwalkparentre literal_valuerrr set_values,     r~c@sTeZdZdZ d dejdeejde fddZ d de deejdejfd d Z dS) ZlibJSONSerializerz5Serializer that uses JSON, zlib, and base64 encoding.Nro pyref_policyrcCs"ttt||dS)Nascii)base64urlsafe_b64encodezlibcompressr dump_jsonencodedecode)rrorrrr serializes zZlibJSONSerializer.serialize serializedcCstjtt||dS)N)r)r load_jsonr decompressrurlsafe_b64decoder)rrrrrr deserialize'szZlibJSONSerializer.deserialize)N) rrrrr Buildablerr PyrefPolicyrrrrrrrrs$  r)$rrFr dataclassesenumr*rAtypesrrrrrrr fiddle._srcrrrr fiddle._src.experimentalr Enumr rr: dataclassr<Pathr_rf ModuleTypeboolrnrr~rrrrrsF      3Q *