forked from AllAlgorithms/python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmorse_code.py
More file actions
69 lines (58 loc) · 1.29 KB
/
morse_code.py
File metadata and controls
69 lines (58 loc) · 1.29 KB
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
#!/usr/bin/env python
"""
Pythonic Implementation of Morse Code encoding
"""
__author__ = "Aditya Krishnakumar"
# The alphabet dictionary for morse codes
morseAlphabet = {
"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": "-----"
}
# Lambda function for decoding the code to alphabet
inverseAlphabet = reduce(lambda a, b: dict(a.items() + b.items()),
[{
morseAlphabet[k]: k
} for k in morseAlphabet.keys()], {})
def encode(_text):
return ' '.join([morseAlphabet[_c.upper()] for _c in _text[:]])
def decode(_text):
return ''.join([inverseAlphabet[_c] for _c in _text.split(' ')])
# Dry runner
def test():
print decode(encode("TEST")) == "TEST"
if __name__ == "__main__":
test()