





import argparse
import os
import re


# All paths are set with the intent you should run this script from the root of the repo with the command
# python utils/check_dummies.py
_lowerCamelCase :    str  							= '''src/diffusers'''

# Matches is_xxx_available()
_lowerCamelCase :    Tuple  							= re.compile(r'''is\_([a-z_]*)_available\(\)''')
# Matches from xxx import bla
_lowerCamelCase :    Union[str, Any]  							= re.compile(r'''\s+from\s+\S*\s+import\s+([^\(\s].*)\n''')


_lowerCamelCase :    List[Any]  							= '''
{0} = None
'''

_lowerCamelCase :    str  							= '''
class {0}(metaclass=DummyObject):
    _backends = {1}

    def __init__(self, *args, **kwargs):
        requires_backends(self, {1})

    @classmethod
    def from_config(cls, *args, **kwargs):
        requires_backends(cls, {1})

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, {1})
'''


_lowerCamelCase :    str  							= '''
def {0}(*args, **kwargs):
    requires_backends({0}, {1})
'''




def _a (  SCREAMING_SNAKE_CASE__      :  List[str]       )	->    str:

							'''simple docstring'''






							SCREAMING_SNAKE_CASE__       :      List[str]     =						_re_backend.findall(SCREAMING_SNAKE_CASE__       )
							if len(SCREAMING_SNAKE_CASE__       ) == 0:
														return None

							return "_and_".join(SCREAMING_SNAKE_CASE__       )




def _a (  )	->    Optional[int]:

							'''simple docstring'''






							with open(os.path.join(SCREAMING_SNAKE_CASE__			,				"__init__.py"       )			,				"r"			,				encoding="utf-8"			,				newline="\n"       ) as f:
														SCREAMING_SNAKE_CASE__       :      List[str]     =						f.readlines()

							# Get to the point we do the actual imports for type checking
							SCREAMING_SNAKE_CASE__       :      Any     =						0
							SCREAMING_SNAKE_CASE__       :      str     =						{}
							# Go through the end of the file
							while line_index < len(SCREAMING_SNAKE_CASE__       ):
														# If the line contains is_backend_available, we grab all objects associated with the `else` block
														SCREAMING_SNAKE_CASE__       :      Optional[int]     =						find_backend(lines[line_index]       )
														if backend is not None:
																					while not lines[line_index].startswith("else:"       ):
																												line_index += 1
																					line_index += 1
																					SCREAMING_SNAKE_CASE__       :      List[str]     =						[]
																					# Until we unindent, add backend objects to the list
																					while line_index < len(SCREAMING_SNAKE_CASE__       ) and len(lines[line_index]       ) > 1:
																												SCREAMING_SNAKE_CASE__       :      Optional[int]     =						lines[line_index]
																												SCREAMING_SNAKE_CASE__       :      Dict     =						_re_single_line_import.search(SCREAMING_SNAKE_CASE__       )
																												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

																					if len(SCREAMING_SNAKE_CASE__       ) > 0:
																												SCREAMING_SNAKE_CASE__       :      List[Any]     =						objects
														else:
																					line_index += 1

							return backend_specific_objects




def _a (  SCREAMING_SNAKE_CASE__      :  List[str]			,				SCREAMING_SNAKE_CASE__      :  Tuple       )	->    int:

							'''simple docstring'''






							if name.isupper():
														return DUMMY_CONSTANT.format(SCREAMING_SNAKE_CASE__       )
							elif name.islower():
														return DUMMY_FUNCTION.format(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       )
							else:
														return DUMMY_CLASS.format(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       )




def _a (  SCREAMING_SNAKE_CASE__      :  Any=None       )	->    Optional[int]:

							'''simple docstring'''






							if backend_specific_objects is None:
														SCREAMING_SNAKE_CASE__       :      Dict     =						read_init()
							# For special correspondence backend to module name as used in the function requires_modulename
							SCREAMING_SNAKE_CASE__       :      Optional[Any]     =						{}

							for backend, objects in backend_specific_objects.items():
														SCREAMING_SNAKE_CASE__       :      str     =						"[" + ", ".join(f'''"{b}"''' for b in backend.split("_and_"       )       ) + "]"
														SCREAMING_SNAKE_CASE__       :      Tuple     =						"# This file is autogenerated by the command `make fix-copies`, do not edit.\n"
														dummy_file += "from ..utils import DummyObject, requires_backends\n\n"
														dummy_file += "\n".join([create_dummy_object(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       ) for o in objects]       )
														SCREAMING_SNAKE_CASE__       :      Dict     =						dummy_file

							return dummy_files




def _a (  SCREAMING_SNAKE_CASE__      :  Dict=False       )	->    Union[str, Any]:

							'''simple docstring'''






							SCREAMING_SNAKE_CASE__       :      List[str]     =						create_dummy_files()
							# For special correspondence backend to shortcut as used in utils/dummy_xxx_objects.py
							SCREAMING_SNAKE_CASE__       :      int     =						{"torch": "pt"}

							# Locate actual dummy modules and read their content.
							SCREAMING_SNAKE_CASE__       :      Tuple     =						os.path.join(SCREAMING_SNAKE_CASE__			,				"utils"       )
							SCREAMING_SNAKE_CASE__       :      str     =						{
							    backend: os.path.join(SCREAMING_SNAKE_CASE__			,				f'''dummy_{short_names.get(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       )}_objects.py'''       )
							    for backend in dummy_files.keys()
							}

							SCREAMING_SNAKE_CASE__       :      Any     =						{}
							for backend, file_path in dummy_file_paths.items():
														if os.path.isfile(SCREAMING_SNAKE_CASE__       ):
																					with open(SCREAMING_SNAKE_CASE__			,				"r"			,				encoding="utf-8"			,				newline="\n"       ) as f:
																												SCREAMING_SNAKE_CASE__       :      Any     =						f.read()
														else:
																					SCREAMING_SNAKE_CASE__       :      List[Any]     =						""

							for backend in dummy_files.keys():
														if dummy_files[backend] != actual_dummies[backend]:
																					if overwrite:
																												print(
																												    f'''Updating diffusers.utils.dummy_{short_names.get(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       )}_objects.py as the main '''
																												    "__init__ has new objects."       )
																												with open(dummy_file_paths[backend]			,				"w"			,				encoding="utf-8"			,				newline="\n"       ) as f:
																																			f.write(dummy_files[backend]       )
																					else:
																												raise ValueError(
																												    "The main __init__ has objects that are not present in "
																												    f'''diffusers.utils.dummy_{short_names.get(SCREAMING_SNAKE_CASE__			,				SCREAMING_SNAKE_CASE__       )}_objects.py. Run `make fix-copies` '''
																												    "to fix this."       )


if __name__ == "__main__":
		_lowerCamelCase :    Union[str, Any]  							= argparse.ArgumentParser()
		parser.add_argument('''--fix_and_overwrite''', action='''store_true''', help='''Whether to fix inconsistencies.''')
		_lowerCamelCase :    Tuple  							= parser.parse_args()

		check_dummies(args.fix_and_overwrite)













def _a (  SCREAMING_SNAKE_CASE__      :  str       )	->    str:

							'''simple docstring'''






							if not all(char in "01" for char in bin_string       ):
														raise ValueError("Non-binary value was passed to the function"       )
							if not bin_string:
														raise ValueError("Empty string was passed to the function"       )
							SCREAMING_SNAKE_CASE__       :      List[Any]     =						""
							while len(SCREAMING_SNAKE_CASE__       ) % 3 != 0:
														SCREAMING_SNAKE_CASE__       :      str     =						"0" + bin_string
							SCREAMING_SNAKE_CASE__       :      List[Any]     =						[
							    bin_string[index : index + 3]
							    for index in range(len(SCREAMING_SNAKE_CASE__       )       )
							    if index % 3 == 0
							]
							for bin_group in bin_string_in_3_list:
														SCREAMING_SNAKE_CASE__       :      List[Any]     =						0
														for index, val in enumerate(SCREAMING_SNAKE_CASE__       ):
																					oct_val += int(2 ** (2 - index) * int(SCREAMING_SNAKE_CASE__       )       )
														oct_string += str(SCREAMING_SNAKE_CASE__       )
							return oct_string


if __name__ == "__main__":
		from doctest import testmod

		testmod()






