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
|
import argparse
def fib(n):
a, b = 0, 1
for _ in range(n):
a, b = b, a+b
return a
def main():
parser = argparse.ArgumentParser(description="Calculate fibonacci")
group = parser.add_mutually_exclusive_group()
group.add_argument('-v', '--verbose', action='store_true')
group.add_argument('-q', '--quit', action='store_true')
parser.add_argument(
'num', help='The fibonacci number you wish to calculate.', type=int)
parser.add_argument(
'-o', '--output', help='Output result into a file', action='store_true')
args = parser.parse_args()
num = args.num
result = fib(num)
output_result = f'The {num}th fibonacci is {result}'
if args.verbose:
print(output_result)
elif args.quit:
print(result)
else:
print(f'Fib({num})={result}')
if args.output:
with open('fibonacci.txt', 'a') as f_h:
f_h.write('\n' + output_result)
if __name__ == "__main__":
main()
|
