Create argparse_for_cli.py

Amogh Singhal · web-flow · commit 645f47ea8b0f · 2018-03-04T11:47:22.000+05:30
diff --git a/argparse_for_cli.py b/argparse_for_cli.py
@@ -0,0 +1,26 @@
+import argparse
+import sys
+
+def main():
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--x', type=float, default=1.0,
+                      help='What is the first number?')
+  parser.add_argument('--y', type=float, default=1.0,
+                      help='What is the second number?')
+  parser.add_argument('--operation', type=str, default='add',
+                      help='What is the operation(add, sub, mul or div)?')
+  args = parser.parse_args()
+  sys.stdout.write(str(calc(args)))
+  
+def calc(args):
+  if args.operation == 'add':
+    return args.x + args.y
+  if args.operation == 'sub':
+    return args.x - args.y
+  if args.operation == 'mul':
+    return args.x * args.y
+  if args.operation == 'div':
+    return args.x / args.y
+    
+if __name__ == '__main__':
+  main()
