Added heatmap

1 files changed, 5 insertions, 1 deletions

diff --git a/parser.py b/parser.py
index 2918d73..f9e3e81 100644
--- a/parser.py
+++ b/parser.py
@@ -78,4 +78,8 @@ if __name__ == "__main__":
         host = "db"
 
     with database.PayGapDatabase(host = host) as db:
-        parse_csv(db, sys.argv[1])
\ No newline at end of file
+        p = sys.argv[1]
+        if os.path.basename(p) == "National_Statistics_Postcode_Lookup_UK.csv":
+            db.append_counties(p)
+        else:
+            parse_csv(db, p)
\ No newline at end of file