exercise 3.16

Author: sunilnahata

Work/pcost.py

@@ -7,14 +7,45 @@
 
 # file = "Data/portfolio.csv"
 
-def portfolio_cost(file):
+def portfolio_cost(filename):
     portfolio = report.read_portfolio(filename)
-    return sum([s['shares']*s['price'] for s in portfolio])
+    return sum([s['shares'] * s['price'] for s in portfolio])
+
+def main(args):
+    if len(args) != 2:
+        raise SystemExit('Usage: %s portfoliofile' % args[0])
+    filename = args[1]
+    print('Total cost:', portfolio_cost(filename))
+
+if __name__ == '__main__':
+    main(sys.argv)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
-if len(sys.argv) == 2:
-    filename = sys.argv[1]
-else:
-    filename = input("Enter a filename: ") 
 
-cost = portfolio_cost(filename)
-print(f"Total cost: {cost}")

Work/report.py

@@ -36,4 +36,13 @@ def portfolio_report(portfoliofile, pricefile):
     report = make_report(portfolio, prices)
     print_report(report)
 
-portfolio_report('Data/portfoliodate.csv', 'Data/prices.csv')
+#portfolio_report('Data/portfoliodate.csv', 'Data/prices.csv)
+
+def main(args):
+    if len(args) != 3:
+        raise SystemExit("Usage: %s portfile pricefile" % args[0])
+    portfolio_report(args[1], args[2])
+
+if __name__ == "__main__":
+    import sys
+    main(sys.argv)