-
Notifications
You must be signed in to change notification settings - Fork 4
/
main.py
146 lines (133 loc) · 4.28 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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
#!/usr/bin/python
import sys
import argparse
import time
from turkish_ids import TurkishID
# ################################
# Eyup Can Elma - https://elma.dev
# Dec 22, 2018 - 11:52:05 PM
# ################################
def check_args(args=None):
parser = argparse.ArgumentParser(
description='Turkish ID Number Validator and Generator App',
prefix_chars='-',
)
parser.add_argument(
'-f', '--guess',
metavar='123456789',
help=(
"Give a TC ID number without it's two rightmost digits then "
"I'll guess the exact TC ID number according to the algorithm."
),
type=str,
)
parser.add_argument(
'-v', '--validate',
metavar='12345678901',
help=(
"Validate a number to be sure that if it's a valid TC ID "
"number or not."
),
type=str,
)
parser.add_argument(
'-g', '--generate',
metavar='[1, ~]',
help=(
"Generate valid TC ID numbers up to the limit given with this. "
"You must specify a sample TC ID number with --start-from "
"or -s parameter to start generating new TC ID numbers from."
),
type=int,
)
parser.add_argument(
'-r', '--relatives',
metavar='[1, ~]',
help=(
"Find the relatives' TC ID numbers according to the TC ID number "
"given by the parameter -s/--start-from."
),
type=int,
)
parser.add_argument(
'-o', '--older',
action='store_true',
help=(
"Tells -r/--relatives argument to search for older relatives."
),
)
parser.add_argument(
'-y', '--younger',
action='store_true',
help=(
"Tells -r/--relatives argument to search for younger relatives."
),
)
parser.add_argument(
'-s', '--start-from',
metavar='12345678901',
help=(
"Sample TC ID number to start generating new ones from. "
"This has to be used with the -g/--generate parameter or the "
"-r/--relatives parameter."
),
type=str,
)
return parser.parse_args()
def main():
turkish_id = TurkishID()
argument = check_args(sys.argv[1:])
if argument.validate:
TC_ID = argument.validate
v = 'a' if turkish_id.is_valid_tc_id_num(TC_ID) else 'NOT a'
print('{tc_id} is {v} valid TC ID number.'.format(tc_id=TC_ID, v=v))
return 0
elif argument.generate:
if not argument.start_from:
print(
'Please also specify a sample TC ID number as a start point '
'with the -s or --start-from parameter '
'so we can generate new ones from.'
)
return 2
values = turkish_id.generate_valid_tc_ids(
argument.generate, argument.start_from)
print(*values, sep='\n')
return 0
elif argument.relatives:
if not argument.start_from:
print(
'Please also specify a sample TC ID number as a start point '
'with the -s or --start-from parameter '
'so we can find the relatives from.'
)
return 2
elif not argument.older and not argument.younger:
print(
'You must specify whether you want to search for the older '
'relatives or the younger ones by using -o/--older or '
'-y/--younger parameters.'
)
return 2
values = turkish_id.find_relatives_tc_ids(
limit=argument.relatives,
start=argument.start_from,
younger=argument.younger is True,
)
print(*values, sep='\n')
elif argument.guess:
if len(argument.guess) != 9:
print('The length of the given number must be exactly 9 digits.')
return 2
result = turkish_id.guess_tc_id_number(argument.guess)
print(result)
return 0
else:
print('Huh. What am I supposed to do?')
return 2
if __name__ == '__main__':
start_time = time.time()
result = main()
end_time = time.time()
print('Finished in {0:.2f} seconds.'.format(end_time-start_time))
sys.exit(result)