



'''simple docstring'''

from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
    is_vision_available,
)


__lowerCAmelCase						      =       {
    '''configuration_layoutlmv3''': [
        '''LAYOUTLMV3_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''LayoutLMv3Config''',
        '''LayoutLMv3OnnxConfig''',
    ],
    '''processing_layoutlmv3''': ['''LayoutLMv3Processor'''],
    '''tokenization_layoutlmv3''': ['''LayoutLMv3Tokenizer'''],
}

try:
						if not is_tokenizers_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						__lowerCAmelCase						      =       ['''LayoutLMv3TokenizerFast''']

try:
						if not is_torch_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						__lowerCAmelCase						      =       [
						    '''LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LIST''',
						    '''LayoutLMv3ForQuestionAnswering''',
						    '''LayoutLMv3ForSequenceClassification''',
						    '''LayoutLMv3ForTokenClassification''',
						    '''LayoutLMv3Model''',
						    '''LayoutLMv3PreTrainedModel''',
						]

try:
						if not is_tf_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						__lowerCAmelCase						      =       [
						    '''TF_LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LIST''',
						    '''TFLayoutLMv3ForQuestionAnswering''',
						    '''TFLayoutLMv3ForSequenceClassification''',
						    '''TFLayoutLMv3ForTokenClassification''',
						    '''TFLayoutLMv3Model''',
						    '''TFLayoutLMv3PreTrainedModel''',
						]

try:
						if not is_vision_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						__lowerCAmelCase						      =       ['''LayoutLMv3FeatureExtractor''']
						__lowerCAmelCase						      =       ['''LayoutLMv3ImageProcessor''']


if TYPE_CHECKING:
						from .configuration_layoutlmva import (
						    LAYOUTLMV3_PRETRAINED_CONFIG_ARCHIVE_MAP,
						    LayoutLMvaConfig,
						    LayoutLMvaOnnxConfig,
						)
						from .processing_layoutlmva import LayoutLMvaProcessor
						from .tokenization_layoutlmva import LayoutLMvaTokenizer

						try:
												if not is_tokenizers_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .tokenization_layoutlmva_fast import LayoutLMvaTokenizerFast

						try:
												if not is_torch_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_layoutlmva import (
												    LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LIST,
												    LayoutLMvaForQuestionAnswering,
												    LayoutLMvaForSequenceClassification,
												    LayoutLMvaForTokenClassification,
												    LayoutLMvaModel,
												    LayoutLMvaPreTrainedModel,
												)

						try:
												if not is_tf_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_tf_layoutlmva import (
												    TF_LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LIST,
												    TFLayoutLMvaForQuestionAnswering,
												    TFLayoutLMvaForSequenceClassification,
												    TFLayoutLMvaForTokenClassification,
												    TFLayoutLMvaModel,
												    TFLayoutLMvaPreTrainedModel,
												)

						try:
												if not is_vision_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .feature_extraction_layoutlmva import LayoutLMvaFeatureExtractor
												from .image_processing_layoutlmva import LayoutLMvaImageProcessor

else:
						import sys

						__lowerCAmelCase						      =       _LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)









'''simple docstring'''

import os
import random
import sys

from . import cryptomath_module as cryptomath
from . import rabin_miller

__lowerCAmelCase						      =       3



def 					__lowerCamelCase					(				lowerCAmelCase_  )      ->				int:
		print('Generating primitive root of p'  )
		while True:
				_a					:				List[Any]				=							random.randrange(3						,      lowerCAmelCase_  )
				if pow(lowerCAmelCase_						,      2						,      lowerCAmelCase_  ) == 1:
						continue
				if pow(lowerCAmelCase_						,      lowerCAmelCase_						,      lowerCAmelCase_  ) == 1:
						continue
				return g



def 					__lowerCamelCase					(				lowerCAmelCase_  )      ->				tuple[tuple[int, int, int, int], tuple[int, int]]:
		print('Generating prime p...'  )
		_a					:				int				=							rabin_miller.generate_large_prime(lowerCAmelCase_  )  # select large prime number.
		_a					:				List[str]				=							primitive_root(lowerCAmelCase_  )  # one primitive root on modulo p.
		_a					:				Any				=							random.randrange(3						,      lowerCAmelCase_  )  # private_key -> have to be greater than 2 for safety.
		_a					:				List[Any]				=							cryptomath.find_mod_inverse(pow(lowerCAmelCase_						,      lowerCAmelCase_						,      lowerCAmelCase_  )						,      lowerCAmelCase_  )

		_a					:				Tuple				=							(key_size, e_a, e_a, p)
		_a					:				str				=							(key_size, d)

		return public_key, private_key



def 					__lowerCamelCase					(				lowerCAmelCase_						,      lowerCAmelCase_  )      ->				None:
		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()

		_a	, _a					:				Dict				=							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 					__lowerCamelCase					(				)      ->				None:
		print('Making key files...'  )
		make_key_files('elgamal'						,      2048  )
		print('Key files generation successful'  )


if __name__ == "__main__":
							main()




