



"""simple docstring"""







import os
import random
import sys

from . import cryptomath_module as cryptoMath  # noqa: N812
from . import rabin_miller as rabinMiller  # noqa: N812



def      __UpperCAmelCase      (    )							->      None:
		print('''Making key files...'''	)
		make_key_files('''rsa'''						,     10_24	)
		print('''Key files generation successful.'''	)



def      __UpperCAmelCase      (    __lowerCamelCase	)							->      tuple[tuple[int, int], tuple[int, int]]:
		print('''Generating prime p...'''	)
		lowercase__   :    Optional[Any]				=			rabinMiller.generate_large_prime(__lowerCamelCase	)
		print('''Generating prime q...'''	)
		lowercase__   :    Tuple				=			rabinMiller.generate_large_prime(__lowerCamelCase	)
		lowercase__   :    str				=			p * q

		print('''Generating e that is relatively prime to (p - 1) * (q - 1)...'''	)
		while True:
				lowercase__   :    Optional[int]				=			random.randrange(2 ** (key_size - 1)						,     2 ** (key_size)	)
				if cryptoMath.gcd(__lowerCamelCase						,     (p - 1) * (q - 1)	) == 1:
						break

		print('''Calculating d that is mod inverse of e...'''	)
		lowercase__   :    List[Any]				=			cryptoMath.find_mod_inverse(__lowerCamelCase						,     (p - 1) * (q - 1)	)

		lowercase__   :    str				=			(n, e)
		lowercase__   :    int				=			(n, d)
		return (public_key, private_key)



def      __UpperCAmelCase      (    __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()

		lowercase__		,			lowercase__   :    Union[str, Any]				=			generate_key(__lowerCamelCase	)
		print(f"""\nWriting public key to file {name}_pubkey.txt..."""	)
		with open(f"""{name}_pubkey.txt"""						,     '''w'''	) as out_file:
				out_file.write(f"""{key_size},{public_key[0]},{public_key[1]}"""	)

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


if __name__ == "__main__":
							main()





"""simple docstring"""







from ..utils import DummyObject, requires_backends
class        __A       (	metaclass=A_    ):







		'''simple docstring'''





		lowerCAmelCase			:					List[str]   	=				["torch", "torchsde"]







		def __init__(	self						:  Tuple			,*_snake_case						:  Union[str, Any]			,**_snake_case						:  Any						)					->		Union[str, Any]:






				"""simple docstring"""



				requires_backends(self			,['''torch''', '''torchsde''']						)







		@classmethod
		def 					UpperCAmelCase    (	cls						:  List[str]			,*_snake_case						:  int			,**_snake_case						:  Union[str, Any]						)					->		str:






				"""simple docstring"""



				requires_backends(cls			,['''torch''', '''torchsde''']						)







		@classmethod
		def 					UpperCAmelCase    (	cls						:  List[Any]			,*_snake_case						:  List[Any]			,**_snake_case						:  List[str]						)					->		List[Any]:






				"""simple docstring"""



				requires_backends(cls			,['''torch''', '''torchsde''']						)
