Skip to content

Commit 7bc48b1

Browse files
committed
Improve insufficient argument handling
1 parent 453f6bf commit 7bc48b1

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

big_number_computation.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,10 @@ def big_number_computation(a, o, b):
5151

5252

5353
if __name__ == '__main__':
54-
if sys.argv[2] != '+' and sys.argv[2] != '-' and sys.argv[2] != '*':
54+
if len(sys.argv) < 4:
55+
print('Not enough arguments to perform computation.', file=sys.stderr)
56+
sys.exit(5)
57+
elif sys.argv[2] != '+' and sys.argv[2] != '-' and sys.argv[2] != '*':
5558
print('Error: Operator must be \'+\' or \'-\'', file=sys.stderr)
5659
sys.exit(2)
5760

date_time_transformation.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,19 @@ def transform_datetime(d, offset, dst_start = None, dst_end = None):
2222
return d
2323

2424
if __name__ == '__main__':
25+
if len(sys.argv) < 2:
26+
print("Not enough arguments.", file=sys.stderr)
27+
sys.exit(1)
28+
2529
d = datetime.datetime.strptime(sys.argv[1], DT_FORMAT)
2630
offset = int(sys.argv[2]) if len(sys.argv) > 2 else None
2731

28-
if len(sys.argv) > 5:
32+
if len(sys.argv) == 4:
2933
print('Warning: Not enough arguments to use daylight saving time mode.', file=sys.stderr)
3034

3135
if offset is not None:
3236
if offset < -12 or offset > +14:
33-
print('Error: Timezone must be within -12 and +14')
37+
print('Error: Timezone must be within -12 and +14.', file=sys.stderr)
3438
sys.exit(2)
3539

3640
d = transform_datetime(

0 commit comments

Comments
 (0)