forked from turecekt/vs_project
/
main.py
119 lines (100 loc) · 2.75 KB
/
main.py
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
107
108
109
110
111
112
113
114
115
116
117
118
119
"""Morse script."""
# Pole znaků.
znaky = {
'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': '-----'
}
# Funkce pro enkódování textu.
def encodovani(vstup):
"""Encode text.
>>> encodovani('A')
'.-'
"""
vstup = vstup.upper()
zprava = ''
for znak in vstup:
if znak != ' ':
zprava += znaky[znak] + ' '
else:
zprava += ' '
return zprava[:-1]
# Funkce pro dekódování znaků
def decodovani(vstup):
"""Decode text.
>>> decodovani('.-')
'A'
"""
try:
vstup += " "
desifrovanaZprava = ''
pismeno = ''
i = 0
for letter in vstup:
if (letter != ' '):
i = 0
pismeno += letter
else:
i += 1
if i == 2:
desifrovanaZprava += ' '
else:
desifrovanaZprava += list(znaky.keys())[
list(znaky.values()).index(pismeno)
]
pismeno = ''
return desifrovanaZprava
except ValueError:
return 0
# Funkce pro vypsání textů.
def vypisZnaku():
"""Write the text.
>>> vypisZnaku()
Podporované znaky:
ABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890
"""
print("Podporované znaky:")
znakPodporovan = ""
vypisZnaku = ""
for znakPodporovan in znaky:
vypisZnaku += znakPodporovan
print(vypisZnaku)
# Funkce pro vstup uživatele.
def start():
"""Input."""
print("Zvolte: 'e' pro encode, 'd' pro decode,"
" jinou klávesu pro ukončení programu:")
typ = input()
if typ == "e":
vypisZnaku()
print("Zadejte vstup pro encode:")
vstup = input()
print("Enkodováno:")
print(encodovani(vstup))
elif typ == "d":
print("Zadejte vstup pro decode:")
vstup = input()
res = decodovani(vstup)
if res != 0:
print("Dekodováno:")
print(decodovani(vstup))
else:
print("Chyba - Nelze dekódovat.")
start()
# Funnkce pro spuštění programu
def main():
"""Start program."""
print("Vítejte v encoderu/decoderu pro morseovku")
start()
if __name__ == '__main__':
main()