python-scripts/scripts/Morse Code Translator/MorseCodeTranslator.py at 7e8a1b9b63d50ed50d8228ed9384fd4bc09b9d3d · metafy-social/python-scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
MORSE_CODES = {
    "A": ".-",
    "B": "-...",
    "C": "-.-.",
    "D": "-..",
    "E": ".",
    "F": "..-.",
    "G": "--.",
    "H": "....",
    "I": "..",
    "J": ".---",
    "K": "-.-",
    "L": ".-..",
    "M": "--",
    "N": "-.",
    "O": "---",
    "P": ".--.",
    "Q": "--.-",
    "R": ".-.",
    "S": "...",
    "T": "-",
    "U": "..-",
    "V": "...-",
    "W": ".--",
    "X": "-..-",
    "Y": "-.--",
    "Z": "--..",
    "1": ".----",
    "2": "..---",
    "3": "...--",
    "4": "....-",
    "5": ".....",
    "6": "-....",
    "7": "--...",
    "8": "---..",
    "9": "----.",
    "0": "-----",
    ", ": "--..--",
    ".": ".-.-.-",
    "?": "..--..",
    "/": "-..-.",
    "-": "-....-",
    "(": "-.--.",
    ")": "-.--.-",
}


def encrypt(message):
    cipher = ""
    for letter in message:
        if letter != " ":
            cipher += MORSE_CODES[letter] + " "
        else:
            cipher += " "
    return cipher


def decrypt(message):
    message += " "
    decipher = ""
    citext = ""  # stores morse code of a single character
    for letter in message:
        if letter != " ":
            i = 0  # keeps count of the spaces between morse characters
            citext += letter
        else:
            i += 1
            if i == 2:
                decipher += " "
            else:
                decipher += list(MORSE_CODES.keys())[
                    list(MORSE_CODES.values()).index(citext)
                ]
                citext = ""
    return decipher


def main():
    while True:
        print('''
            -----------------------------
            Morse Code Translator
            1. Encrypt
            2. Decrypt

            Enter 'q' to quit.
            -----------------------------
            ''')

        choice = input("Enter your choice: ")
        if choice == "1":
            message = input("Enter message to encrypt: ")
            result = encrypt(message.upper())
            print("> ", result)
        elif choice == "2":
            message = input("Enter message to decrypt: ")
            result = decrypt(message)
            print("> ", result)
        elif choice == "q":
            break
        else:
            print("Invalid choice. Try again.")


if __name__ == "__main__":
    main()