B ѻd@s`dZddlZddlZddlmZejdej dZ d ddZ iZ d dd Z dd d Zdd d gZdS)z)Generate __all__ from a module docstring.N) tf_inspectz ^@@(\w+)$)flagscsn|dkrtj|g}tddttj|Dt}x,|D]$}|fddt|jDq>Wt |S)aoGenerates `__all__` from the docstring of one or more modules. Usage: `make_all(__name__)` or `make_all(__name__, [sys.modules(__name__), other_module])`. The doc string modules must each a docstring, and `__all__` will contain all symbols with `@@` references, where that symbol currently exists in the module named `module_name`. Args: module_name: The name of the module (usually `__name__`). doc_string_modules: a list of modules from which to take docstring. If None, then a list containing only the module named `module_name` is used. Returns: A list suitable for use as `__all__`. Ncss|]\}}|VqdS)N).0name_rrQ/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/util/all_util.py .szmake_all..cs$g|]}|dkr|dqS))group)rm) cur_membersrr 2szmake_all..) _sysmodulesset _tf_inspect getmembersupdate_reference_patternfinditer__doc__list) module_namedoc_string_modulesresultsZ doc_moduler)r rmake_alls  rcCsF|tkrtd||dd}t|\}}|s6|}t|||dS)aReveals a symbol that was previously removed by `remove_undocumented`. This should be used by tensorflow internal tests only. It explicitly defeats the encapsulation afforded by `remove_undocumented`. It throws an exception when the symbol was not hidden in the first place. Args: symbol_name: a string representing the full absolute path of the symbol. target_module: if specified, the module in which to restore the symbol. z Symbol %s is not a hidden symbol.N)_HIDDEN_ATTRIBUTES LookupErrorsplitsetattr) symbol_name target_moduleZsymbol_basenameZoriginal_module attr_valuerrrreveal_undocumented>s   r&c Cstttj|}t||}||p$g7}|t|}tj|}x@|D]8}|drTqD|d|}|t||ft|<t||qDWdS)aRemoves symbols in a module that are not referenced by a docstring. Args: module_name: the name of the module (usually `__name__`). allowed_exception_list: a list of names that should not be removed. doc_string_modules: a list of modules from which to take the docstrings. If None, then a list containing only the module named `module_name` is used. Furthermore, if a symbol previously added with `add_to_global_allowlist`, then it will always be allowed. This is useful for internal tests. Returns: None rrN) rdirrrr startswithgetattrrdelattr) rZallowed_exception_listrZcurrent_symbolsZ should_haveZ extra_symbolsr$Z extra_symbolZfully_qualified_namerrrremove_undocumentedRs        r+)N)N)NN)rre_resysrtensorflow.python.utilrrcompile MULTILINErrrr&r+__all__rrrrs  !