Example #1
0
                    print("Invalid parameters. \n" + help_message)
            except Exception as e:
                print(e)
                print("Invalid parameters. \n" + help_message)

        elif method_id == "2":  # random number
            key = sys.argv[0]
            test_randomness = False
            if nb_args == 3:
                test_randomness = True if sys.argv[1] == "True" else False

            rdm_generator = RandomNumberGenerator(key,
                                                  use_custom=USE_CUSTOM_RC4)
            print(f"Your random number: {rdm_generator.generate()}")
            if test_randomness:
                rdm_generator.display_random_image()
                rdm_generator.verify_randomness()
                rdm_generator.plot_random_and_semi_random()

        elif method_id == "3":  # full cipher/decipher pipeline
            filename = sys.argv[0]
            algorithm = sys.argv[
                1] if sys.argv[1] == "rc4" or sys.argv[1] == "rc5" else None
            key = sys.argv[2]

            WAV_file_demo = False
            if nb_args == 5:
                WAV_file_demo = True if sys.argv[3] == "True" else False
                if WAV_file_demo and not filename.split(".")[-1] == "wav":
                    print(
                        "Parameter 'WAV_file_demo' is only for WAV files. Parameter ignored."