


import random



def     __lowerCAmelCase						(							snake_case	:    int						,      snake_case	:    float						,      snake_case	:    bool = False )       ->							dict:
   __lowerCamelCase:				dict     =	{i: [] for i in range(snake_case )}

   # if probability is greater or equal than 1, then generate a complete graph
   if probability >= 1:
      return complete_graph(snake_case )
   # if probability is lower or equal than 0, then return a graph without edges
   if probability <= 0:
      return graph

   # for each couple of nodes, add an edge from u to v
   # if the number randomly generated is greater than probability probability
   for i in range(snake_case ):
      for j in range(i + 1						,      snake_case ):
         if random.random() < probability:
            graph[i].append(snake_case )
            if not directed:
               # if the graph is undirected, add an edge in from j to i, either
               graph[j].append(snake_case )
   return graph



def     __lowerCAmelCase						(							snake_case	:    int )       ->							dict:
   return {
       i: [j for j in range(snake_case ) if i != j] for i in range(snake_case )
   }


if __name__ == "__main__":
   import doctest

   doctest.testmod()






_A		:				Tuple									=       '''ABCDEFGHIJKLMNOPQRSTUVWXYZ'''



def     __lowerCAmelCase						(							)       ->							None:
   __lowerCamelCase:				Optional[int]     =	input("""Enter message: """ )
   __lowerCamelCase:				Dict     =	input("""Enter key [alphanumeric]: """ )
   __lowerCamelCase:				List[Any]     =	input("""Encrypt/Decrypt [e/d]: """ )

   if mode.lower().startswith("""e""" ):
      __lowerCamelCase:				Optional[int]     =	"""encrypt"""
      __lowerCamelCase:				Optional[int]     =	encrypt_message(snake_case						,      snake_case )
   elif mode.lower().startswith("""d""" ):
      __lowerCamelCase:				Union[str, Any]     =	"""decrypt"""
      __lowerCamelCase:				Optional[Any]     =	decrypt_message(snake_case						,      snake_case )

   print(f'\n{mode.title()}ed message:' )
   print(snake_case )



def     __lowerCAmelCase						(							snake_case	:    str						,      snake_case	:    str )       ->							str:
   return translate_message(snake_case						,      snake_case						,      """encrypt""" )



def     __lowerCAmelCase						(							snake_case	:    str						,      snake_case	:    str )       ->							str:
   return translate_message(snake_case						,      snake_case						,      """decrypt""" )



def     __lowerCAmelCase						(							snake_case	:    str						,      snake_case	:    str						,      snake_case	:    str )       ->							str:
   __lowerCamelCase:				Any     =	[]
   __lowerCamelCase:				Optional[int]     =	0
   __lowerCamelCase:				Any     =	key.upper()

   for symbol in message:
      __lowerCamelCase:				int     =	LETTERS.find(symbol.upper() )
      if num != -1:
         if mode == "encrypt":
            num += LETTERS.find(key[key_index] )
         elif mode == "decrypt":
            num -= LETTERS.find(key[key_index] )

         num %= len(snake_case )

         if symbol.isupper():
            translated.append(LETTERS[num] )
         elif symbol.islower():
            translated.append(LETTERS[num].lower() )

         key_index += 1
         if key_index == len(snake_case ):
            __lowerCamelCase:				Union[str, Any]     =	0
      else:
         translated.append(snake_case )
   return "".join(snake_case )


if __name__ == "__main__":
   main()


