





import collections
import os
import re
from pathlib import Path


lowercase    : List[Any]    	=  "src/transformers"


# Matches is_xxx_available()
lowercase    : Optional[Any]    	=  re.compile(R"""is\_([a-z_]*)_available()""")
# Catches a one-line _import_struct = {xxx}
lowercase    : Any    	=  re.compile(R"""^_import_structure\s+=\s+\{([^\}]+)\}""")
# Catches a line with a key-values pattern: "bla": ["foo", "bar"]
lowercase    : Union[str, Any]    	=  re.compile(R"""\s+\"\S*\":\s+\[([^\]]*)\]""")
# Catches a line if not is_foo_available
lowercase    : Optional[int]    	=  re.compile(R"""^\s*if\s+not\s+is\_[a-z_]*\_available\(\)""")
# Catches a line _import_struct["bla"].append("foo")
lowercase    : List[str]    	=  re.compile(R"""^\s*_import_structure\[\"\S*\"\]\.append\(\"(\S*)\"\)""")
# Catches a line _import_struct["bla"].extend(["foo", "bar"]) or _import_struct["bla"] = ["foo", "bar"]
lowercase    : Any    	=  re.compile(R"""^\s*_import_structure\[\S*\](?:\.extend\(|\s*=\s+)\[([^\]]*)\]""")
# Catches a line with an object between quotes and a comma:     "MyModel",
lowercase    : List[Any]    	=  re.compile(R"""^\s+\"([^\"]+)\",""")
# Catches a line with objects between brackets only:    ["foo", "bar"],
lowercase    : Optional[Any]    	=  re.compile(R"""^\s+\[([^\]]+)\]""")
# Catches a line with from foo import bar, bla, boo
lowercase    : Union[str, Any]    	=  re.compile(R"""\s+from\s+\S*\s+import\s+([^\(\s].*)\n""")
# Catches a line with try:
lowercase    : int    	=  re.compile(R"""^\s*try:""")
# Catches a line with else:
lowercase    : Any    	=  re.compile(R"""^\s*else:""")


def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase					):

       '''simple docstring'''





       if _re_test_backend.search(_UpperCamelCase					) is None:
              return None
       _UpperCamelCase              =						[b[0] for b in _re_backend.findall(_UpperCamelCase					)]
       backends.sort()
       return "_and_".join(_UpperCamelCase					)


def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase					):

       '''simple docstring'''





       with open(_UpperCamelCase    ,       '''r'''    ,       encoding='''utf-8'''    ,       newline='''\n'''					) as f:
              _UpperCamelCase              =						f.readlines()

       _UpperCamelCase              =						0
       while line_index < len(_UpperCamelCase					) and not lines[line_index].startswith('''_import_structure = {'''					):
              line_index += 1

       # If this is a traditional init, just return.
       if line_index >= len(_UpperCamelCase					):
              return None

       # First grab the objects without a specific backend in _import_structure
       _UpperCamelCase              =						[]
       while not lines[line_index].startswith('''if TYPE_CHECKING'''					) and find_backend(lines[line_index]					) is None:
              _UpperCamelCase              =						lines[line_index]
              # If we have everything on a single line, let's deal with it.
              if _re_one_line_import_struct.search(_UpperCamelCase					):
                     _UpperCamelCase              =						_re_one_line_import_struct.search(_UpperCamelCase					).groups()[0]
                     _UpperCamelCase              =						re.findall(R'''\[([^\]]+)\]'''    ,       _UpperCamelCase					)
                     for imp in imports:
                            objects.extend([obj[1:-1] for obj in imp.split(''', '''					)]					)
                     line_index += 1
                     continue
              _UpperCamelCase              =						_re_import_struct_key_value.search(_UpperCamelCase					)
              if single_line_import_search is not None:
                     _UpperCamelCase              =						[obj[1:-1] for obj in single_line_import_search.groups()[0].split(''', '''					) if len(_UpperCamelCase					) > 0]
                     objects.extend(_UpperCamelCase					)
              elif line.startswith(''' ''' * 8 + '''"'''					):
                     objects.append(line[9:-3]					)
              line_index += 1

       _UpperCamelCase              =						{'''none''': objects}
       # Let's continue with backend-specific objects in _import_structure
       while not lines[line_index].startswith('''if TYPE_CHECKING'''					):
              # If the line is an if not is_backend_available, we grab all objects associated.
              _UpperCamelCase              =						find_backend(lines[line_index]					)
              # Check if the backend declaration is inside a try block:
              if _re_try.search(lines[line_index - 1]					) is None:
                     _UpperCamelCase              =						None

              if backend is not None:
                     line_index += 1

                     # Scroll until we hit the else block of try-except-else
                     while _re_else.search(lines[line_index]					) is None:
                            line_index += 1

                     line_index += 1

                     _UpperCamelCase              =						[]
                     # Until we unindent, add backend objects to the list
                     while len(lines[line_index]					) <= 1 or lines[line_index].startswith(''' ''' * 4					):
                            _UpperCamelCase              =						lines[line_index]
                            if _re_import_struct_add_one.search(_UpperCamelCase					) is not None:
                                   objects.append(_re_import_struct_add_one.search(_UpperCamelCase					).groups()[0]					)
                            elif _re_import_struct_add_many.search(_UpperCamelCase					) is not None:
                                   _UpperCamelCase              =						_re_import_struct_add_many.search(_UpperCamelCase					).groups()[0].split(''', '''					)
                                   _UpperCamelCase              =						[obj[1:-1] for obj in imports if len(_UpperCamelCase					) > 0]
                                   objects.extend(_UpperCamelCase					)
                            elif _re_between_brackets.search(_UpperCamelCase					) is not None:
                                   _UpperCamelCase              =						_re_between_brackets.search(_UpperCamelCase					).groups()[0].split(''', '''					)
                                   _UpperCamelCase              =						[obj[1:-1] for obj in imports if len(_UpperCamelCase					) > 0]
                                   objects.extend(_UpperCamelCase					)
                            elif _re_quote_object.search(_UpperCamelCase					) is not None:
                                   objects.append(_re_quote_object.search(_UpperCamelCase					).groups()[0]					)
                            elif line.startswith(''' ''' * 8 + '''"'''					):
                                   objects.append(line[9:-3]					)
                            elif line.startswith(''' ''' * 12 + '''"'''					):
                                   objects.append(line[13:-3]					)
                            line_index += 1

                     _UpperCamelCase              =						objects
              else:
                     line_index += 1

    # At this stage we are in the TYPE_CHECKING part, first grab the objects without a specific backend
       _UpperCamelCase              =						[]
       while (
           line_index < len(_UpperCamelCase					)
           and find_backend(lines[line_index]					) is None
           and not lines[line_index].startswith('''else'''					)
       ):
              _UpperCamelCase              =						lines[line_index]
              _UpperCamelCase              =						_re_import.search(_UpperCamelCase					)
              if single_line_import_search is not None:
                     objects.extend(single_line_import_search.groups()[0].split(''', '''					)					)
              elif line.startswith(''' ''' * 8					):
                     objects.append(line[8:-2]					)
              line_index += 1

       _UpperCamelCase              =						{'''none''': objects}
       # Let's continue with backend-specific objects
       while line_index < len(_UpperCamelCase					):
              # If the line is an if is_backend_available, we grab all objects associated.
              _UpperCamelCase              =						find_backend(lines[line_index]					)
              # Check if the backend declaration is inside a try block:
              if _re_try.search(lines[line_index - 1]					) is None:
                     _UpperCamelCase              =						None

              if backend is not None:
                     line_index += 1

                     # Scroll until we hit the else block of try-except-else
                     while _re_else.search(lines[line_index]					) is None:
                            line_index += 1

                     line_index += 1

                     _UpperCamelCase              =						[]
                     # Until we unindent, add backend objects to the list
                     while len(lines[line_index]					) <= 1 or lines[line_index].startswith(''' ''' * 8					):
                            _UpperCamelCase              =						lines[line_index]
                            _UpperCamelCase              =						_re_import.search(_UpperCamelCase					)
                            if single_line_import_search is not None:
                                   objects.extend(single_line_import_search.groups()[0].split(''', '''					)					)
                            elif line.startswith(''' ''' * 12					):
                                   objects.append(line[12:-2]					)
                            line_index += 1

                     _UpperCamelCase              =						objects
              else:
                     line_index += 1

       return import_dict_objects, type_hint_objects


def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase    ,       lowerCAmelCase					):

       '''simple docstring'''





       def find_duplicates(lowerCAmelCase					):
              return [k for k, v in collections.Counter(_UpperCamelCase					).items() if v > 1]

       if list(import_dict_objects.keys()					) != list(type_hint_objects.keys()					):
              return ["Both sides of the init do not have the same backends!"]

       _UpperCamelCase              =						[]
       for key in import_dict_objects.keys():
              _UpperCamelCase              =						find_duplicates(import_dict_objects[key]					)
              if duplicate_imports:
                     errors.append(f'''Duplicate _import_structure definitions for: {duplicate_imports}'''					)
              _UpperCamelCase              =						find_duplicates(type_hint_objects[key]					)
              if duplicate_type_hints:
                     errors.append(f'''Duplicate TYPE_CHECKING objects for: {duplicate_type_hints}'''					)

              if sorted(set(import_dict_objects[key]					)					) != sorted(set(type_hint_objects[key]					)					):
                     _UpperCamelCase              =						'''base imports''' if key == '''none''' else f'''{key} backend'''
                     errors.append(f'''Differences for {name}:'''					)
                     for a in type_hint_objects[key]:
                            if a not in import_dict_objects[key]:
                                   errors.append(f'''  {a} in TYPE_HINT but not in _import_structure.'''					)
                     for a in import_dict_objects[key]:
                            if a not in type_hint_objects[key]:
                                   errors.append(f'''  {a} in _import_structure but not in TYPE_HINT.'''					)
       return errors


def     SCREAMING_SNAKE_CASE			(      ):

       '''simple docstring'''





       _UpperCamelCase              =						[]
       for root, _, files in os.walk(_UpperCamelCase					):
              if "__init__.py" in files:
                     _UpperCamelCase              =						os.path.join(_UpperCamelCase    ,       '''__init__.py'''					)
                     _UpperCamelCase              =						parse_init(_UpperCamelCase					)
                     if objects is not None:
                            _UpperCamelCase              =						analyze_results(*_UpperCamelCase					)
                            if len(_UpperCamelCase					) > 0:
                                   _UpperCamelCase              =						f'''Problem in {fname}, both halves do not define the same objects.\n{errors[0]}'''
                                   failures.append('''\n'''.join(_UpperCamelCase					)					)
       if len(_UpperCamelCase					) > 0:
              raise ValueError('''\n\n'''.join(_UpperCamelCase					)					)


def     SCREAMING_SNAKE_CASE			(      ):

       '''simple docstring'''





       _UpperCamelCase              =						[]
       for path, directories, files in os.walk(_UpperCamelCase					):
              for folder in directories:
                     # Ignore private modules
                     if folder.startswith('''_'''					):
                            directories.remove(_UpperCamelCase					)
                            continue
                     # Ignore leftovers from branches (empty folders apart from pycache)
                     if len(list((Path(_UpperCamelCase					) / folder).glob('''*.py'''					)					)					) == 0:
                            continue
                     _UpperCamelCase              =						str((Path(_UpperCamelCase					) / folder).relative_to(_UpperCamelCase					)					)
                     _UpperCamelCase              =						short_path.replace(os.path.sep    ,       '''.'''					)
                     submodules.append(_UpperCamelCase					)
              for fname in files:
                     if fname == "__init__.py":
                            continue
                     _UpperCamelCase              =						str((Path(_UpperCamelCase					) / fname).relative_to(_UpperCamelCase					)					)
                     _UpperCamelCase              =						short_path.replace('''.py'''    ,       ''''''					).replace(os.path.sep    ,       '''.'''					)
                     if len(submodule.split('''.'''					)					) == 1:
                            submodules.append(_UpperCamelCase					)
       return submodules


lowercase    : Union[str, Any]    	=  [
    "convert_pytorch_checkpoint_to_tf2",
    "modeling_flax_pytorch_utils",
    "models.esm.openfold_utils",
]


def     SCREAMING_SNAKE_CASE			(      ):

       '''simple docstring'''





       from transformers.utils import direct_transformers_import

       _UpperCamelCase              =						direct_transformers_import(_UpperCamelCase					)

       _UpperCamelCase              =						set(transformers._import_structure.keys()					)
       # This contains all the base keys of the _import_structure object defined in the init, but if the user is missing
       # some optional dependencies, they may not have all of them. Thus we read the init to read all additions and
       # (potentiall re-) add them.
       with open(os.path.join(_UpperCamelCase    ,       '''__init__.py'''					)    ,       '''r'''					) as f:
              _UpperCamelCase              =						f.read()
       import_structure_keys.update(set(re.findall(R'''import_structure\[\"([^\"]*)\"\]'''    ,       _UpperCamelCase					)					)					)

       _UpperCamelCase              =						[
           module
           for module in get_transformers_submodules()
           if module not in IGNORE_SUBMODULES and module not in import_structure_keys
       ]

       if len(_UpperCamelCase					) > 0:
              _UpperCamelCase              =						'''\n'''.join(f'''- {module}''' for module in module_not_registered					)
              raise ValueError(
                  '''The following submodules are not properly registed in the main init of Transformers:\n'''
                  f'''{list_of_modules}\n'''
                  '''Make sure they appear somewhere in the keys of `_import_structure` with an empty list as value.'''					)


if __name__ == "__main__":
       check_all_inits()
       check_submodules()













def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase    ,       lowerCAmelCase    ,       lowerCAmelCase = 0    ,       lowerCAmelCase = 0					):
       _UpperCamelCase              =						right or len(lowerCAmelCase					) - 1
       if left > right:
              return -1
       elif list_data[left] == key:
              return left
       elif list_data[right] == key:
              return right
       else:
              return search(lowerCAmelCase    ,       lowerCAmelCase    ,       left + 1    ,       right - 1					)


if __name__ == "__main__":
       import doctest

       doctest.testmod()






