3.4

Author: kavun

Work/fileparse.py

@@ -1,3 +1,30 @@
 # fileparse.py
 #
 # Exercise 3.3
+
+import csv
+
+def parse_csv(filename, select: list):
+    '''
+    Parse a CSV file into a list of records
+    '''
+    with open(filename) as f:
+        rows = csv.reader(f)
+
+        # Read the file headers
+        headers = next(rows)
+        select_indices = [headers.index(name) for name in select] \
+            if select \
+            else [i for i in range(len(headers))]
+        if select:
+            headers = select
+        records = []
+        for row in rows:
+            if not row:    # Skip rows with no data
+                continue
+            record = dict(zip(
+                headers,
+                [row[i] for i in select_indices]))
+            records.append(record)
+
+    return records

Work/fileparse_test.py

@@ -0,0 +1,8 @@
+from fileparse import parse_csv
+
+def test_parse_csv_select_3_4():
+    assert parse_csv('Data/portfolio.csv', select=['name','shares']) == \
+        [{'name': 'AA', 'shares': '100'}, {'name': 'IBM', 'shares': '50'},
+         {'name': 'CAT', 'shares': '150'}, {'name': 'MSFT', 'shares': '200'},
+         {'name': 'GE', 'shares': '95'}, {'name': 'MSFT', 'shares': '50'},
+         {'name': 'IBM', 'shares': '100'}]