o ,wi @sdZddlZddlZddlZddlZddlmZmZedZ edZ edZ de de fd d Z de de fd d Zde dee fd dZejGdddZdS)aManages a namespace to avoid conflicting names. Generally, when generating code, we do not want variables names to conflict. This class helps manage names in scope, and for simplicity we try to avoid conflicts between globals/non-locals/locals, even if they might work when run. N)OptionalSetz (?<=.)([A-Z])z [^a-z_0-9]+z[a-z][a-z_0-9]+namereturncCstdd|S)z:Converts a camel or studly-caps name to a snake_case name.cSsd|dS)N_r)group)mr Z/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fiddle/_src/codegen/namespace.py$sz camel_to_snake..)_CAMEL_TO_SNAKE_REsublowerrr r r camel_to_snake"srcCs td|S)z@Removes invalid characters such as whitespace, parenthesis, etc.)_PY_VAR_NAME_INVALID_REr rr r r valid_name_chars's rcCs(tt|}t|}|r|dSdS)zCReturns a sanitized, lower-case version of `name` for py variables.rN)rr_PY_VAR_NAME_REsearchr)rmatchr r r py_var_name,s  rc@sdeZdZUdZejdddZeee d<dede fdd Z d edefd d Z d edefddZ dS) NamespacezManages active Python instance names. By default, the namespace will be populated with Python keywords. If you do not want this, then initialize `names` manually to the empty set. cCs ttjS)N)setkeywordkwlistr r r r r <s zNamespace.)default_factorynameskeyrcCs ||jvS)zReturns True if a name is already defined. In the default Namespace() constructor, names include keywords. Args: key: Name to check. )r)selfrr r r __contains__>s zNamespace.__contains__rcCs*||jvr td|d|j||S)zBAdds a name and returns it, raising an error if it already exists.z Tried to add z) (e.g. an import), but it already exists!)r ValueErroradd)rrr r r r"Hs   z Namespace.addprefixcCsf|t|}||jvr||StjddD]}|d||jvr.||d|Sqtd)aGenerates a new name for a given instance. This method adds the new name to the namespace. Example: self.names = ["foo"] get_new_name("bar", "shared_") == "shared_bar" Example 2: self.names = ["shared_foo", "bar"] get_new_name("Foo", "shared_") == "shared_foo_2" Args: base_name: Base name to derive a new name for. This will be converted to snake case automatically. prefix: Prefix to prepend to any names. Returns: A new unique name. )startrz.pytype helper -- itertools.count() is infinite)rrr" itertoolscountAssertionError)r base_namer#rir r r get_new_namePs   zNamespace.get_new_nameN)__name__ __module__ __qualname____doc__ dataclassesfieldrrstr__annotations__boolr r"r+r r r r r3s  r)r/r0r&rretypingrrcompiler rrr2rrr dataclassrr r r r s