Exercise 1.33

drblanc · drblanc · commit 4d872026644d · 2020-06-17T16:00:25.000-04:00
diff --git a/Work/pcost.py b/Work/pcost.py
@@ -1,7 +1,8 @@
 # pcost.py
 #
-# Exercise 1.32
+# Exercise 1.33
 
+import sys
 import csv
 
 def portfolio_cost(filename):
@@ -23,6 +24,11 @@ def portfolio_cost(filename):
     f.close()
     return total_cost
 
-cost = portfolio_cost('Data/portfolio.csv')
+if len(sys.argv) == 2:
+    filename = sys.argv[1]
+else:
+    filename = 'Data/portfolio.csv'
+
+cost = portfolio_cost(filename)
 print('Total cost:', cost)
 
