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
|
#!/usr/bin/env python3
"""
# rotate - Copyright (c) 2009-2022 Michael Buesch <m@bues.ch>
# Licensed under the
# GNU General Public license version 2 or (at your option) any later version
"""
import sys
import getopt
def rotateChar(c, count):
count %= 26
c = ord(c)
if c >= ord('a') and c <= ord('z'):
start = ord('a')
end = ord('z')
elif c >= ord('A') and c <= ord('Z'):
start = ord('A')
end = ord('Z')
else: # Do not rotate
return chr(c)
c += count
if (c < start):
c = end - (start - c - 1)
elif (c > end):
c = start + (c - end - 1)
assert(c >= start and c <= end)
return chr(c)
def rotateString(string, count):
s = ""
for c in string:
s += rotateChar(c, count)
return s
def test():
count = 0
for i in range(-100, 101):
s = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123456789!@#$%^&*()_+|~"
if (rotateString(rotateString(s, i), -i) != s):
print("Selftest FAILED at", i)
return 1
count += 1
print(count, "selftests passed")
return 0
def usage():
print("Usage: %s [OPTIONS]" % sys.argv[0])
print("")
print("-h|--help Print this help text")
print("-c|--count COUNT Rotate by COUNT")
print("-s|--string STR Rotate STR (no prompt)")
print("-f|--file FILE Rotate the contents of FILE")
def main(argv):
opt_count = 13
opt_string = None
opt_file = None
try:
(opts, args) = getopt.getopt(argv[1:],
"hc:s:f:t",
[ "help", "count=", "string=", "test", ])
for (o, v) in opts:
if o in ("-h", "--help"):
usage()
return 0
if o in ("-c", "--count"):
opt_count = int(v)
if o in ("-s", "--string"):
opt_string = v
if o in ("-f", "--file"):
opt_file = v
if o in ("-t", "--test"):
return test()
except (getopt.GetoptError, ValueError):
usage()
return 1
if opt_file:
try:
data = file(opt_file, "r").read()
except IOError as e:
print("Failed to read file:", e.strerror)
return 1
sys.stdout.write(rotateString(data, opt_count))
return 0
if opt_string:
print(rotateString(opt_string, opt_count))
else:
while 1:
try:
string = input("rot> ")
except (EOFError, KeyboardInterrupt):
break
if not string:
break
print(rotateString(string, opt_count))
return 0
if __name__ == "__main__":
sys.exit(main(sys.argv))
# vim: ts=4 sw=4 expandtab
|
