



"""simple docstring"""







import os
import random
import sys

from . import cryptomath_module as cryptomath
from . import rabin_miller

A		=						3

def  __SCREAMING_SNAKE_CASE		(			lowerCamelCase_:							int ):
						"""simple docstring"""






						print("Generating primitive root of p" )
						while True:
												snake_case   :			List[Any]						    = random.randrange(3     ,		lowerCamelCase_ )
												if pow(lowerCamelCase_     ,		2     ,		lowerCamelCase_ ) == 1:
																		continue
												if pow(lowerCamelCase_     ,		lowerCamelCase_     ,		lowerCamelCase_ ) == 1:
																		continue
												return g

def  __SCREAMING_SNAKE_CASE		(			lowerCamelCase_:							int ):
						"""simple docstring"""






						print("Generating prime p..." )
						snake_case   :			List[Any]						    = rabin_miller.generate_large_prime(lowerCamelCase_ )  # select large prime number.
						snake_case   :			Dict						    = primitive_root(lowerCamelCase_ )  # one primitive root on modulo p.
						snake_case   :			Tuple						    = random.randrange(3     ,		lowerCamelCase_ )  # private_key -> have to be greater than 2 for safety.
						snake_case   :			Optional[Any]						    = cryptomath.find_mod_inverse(pow(lowerCamelCase_     ,		lowerCamelCase_     ,		lowerCamelCase_ )     ,		lowerCamelCase_ )

						snake_case   :			List[Any]						    = (key_size, e_a, e_a, p)
						snake_case   :			List[Any]						    = (key_size, d)

						return public_key, private_key

def  __SCREAMING_SNAKE_CASE		(			lowerCamelCase_:							str     ,		lowerCamelCase_:							int ):
						"""simple docstring"""






						if os.path.exists(f'''{name}_pubkey.txt''' ) or os.path.exists(f'''{name}_privkey.txt''' ):
												print("\nWARNING:" )
												print(
												    f'''"{name}_pubkey.txt" or "{name}_privkey.txt" already exists. \n'''
												    "Use a different name or delete these files and re-run this program." )
												sys.exit()

						snake_case    ,			snake_case   :			str						    = generate_key(lowerCamelCase_ )
						print(f'''\nWriting public key to file {name}_pubkey.txt...''' )
						with open(f'''{name}_pubkey.txt'''     ,		"w" ) as fo:
												fo.write(f'''{public_key[0]},{public_key[1]},{public_key[2]},{public_key[3]}''' )

						print(f'''Writing private key to file {name}_privkey.txt...''' )
						with open(f'''{name}_privkey.txt'''     ,		"w" ) as fo:
												fo.write(f'''{private_key[0]},{private_key[1]}''' )

def  __SCREAMING_SNAKE_CASE		(			):
						"""simple docstring"""






						print("Making key files..." )
						make_key_files("elgamal"     ,		2_0_4_8 )
						print("Key files generation successful" )


if __name__ == "__main__":
			main()








"""simple docstring"""







import importlib
import inspect
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_config_docstrings.py
A		=						'src/transformers'


# This is to make sure the transformers module imported is the one in the repo.
A		=						importlib.util.spec_from_file_location(
    'transformers',
    os.path.join(PATH_TO_TRANSFORMERS, '__init__.py'),
    submodule_search_locations=[PATH_TO_TRANSFORMERS],
)
A		=						spec.loader.load_module()

A		=						transformers.models.auto.configuration_auto.CONFIG_MAPPING

# Regex pattern used to find the checkpoint mentioned in the docstring of `config_class`.
# For example, `[bert-base-uncased](https://huggingface.co/bert-base-uncased)`
A		=						re.compile('\[(.+?)\]\((https://huggingface\.co/.+?)\)')


A		=						{
    'CLIPConfigMixin',
    'DecisionTransformerConfigMixin',
    'EncoderDecoderConfigMixin',
    'RagConfigMixin',
    'SpeechEncoderDecoderConfigMixin',
    'VisionEncoderDecoderConfigMixin',
    'VisionTextDualEncoderConfigMixin',
}

def  __SCREAMING_SNAKE_CASE		(			):
						"""simple docstring"""






						snake_case   :			Union[str, Any]						    = []

						for config_class in list(CONFIG_MAPPING.values() ):
												snake_case   :			List[str]						    = False

												# source code of `config_class`
												snake_case   :			Optional[int]						    = inspect.getsource(lowerCamelCase_ )
												snake_case   :			Optional[int]						    = _re_checkpoint.findall(lowerCamelCase_ )

												for checkpoint in checkpoints:
																		# Each `checkpoint` is a tuple of a checkpoint name and a checkpoint link.
																		# For example, `('bert-base-uncased', 'https://huggingface.co/bert-base-uncased')`
																		snake_case    ,			snake_case   :			Tuple						    = checkpoint

																		# verify the checkpoint name corresponds to the checkpoint link
																		snake_case   :			Any						    = f'''https://huggingface.co/{ckpt_name}'''
																		if ckpt_link == ckpt_link_from_name:
																								snake_case   :			List[str]						    = True
																								break

												snake_case   :			Any						    = config_class.__name__
												if not checkpoint_found and name not in CONFIG_CLASSES_TO_IGNORE_FOR_DOCSTRING_CHECKPOINT_CHECK:
																		configs_without_checkpoint.append(lowerCamelCase_ )

						if len(lowerCamelCase_ ) > 0:
												snake_case   :			List[str]						    = "\n".join(sorted(lowerCamelCase_ ) )
												raise ValueError(f'''The following configurations don\'t contain any valid checkpoint:\n{message}''' )


if __name__ == "__main__":
			check_config_docstrings_have_checkpoints()



