X-Git-Url: https://plomlompom.com/repos/?p=berlin-corona-table;a=blobdiff_plain;f=enhance_table.py;h=2b6bb8dc8ff7fcc4f51cb279a44671294f4a0d79;hp=11f65eb5ef7d737c5f494d2fd694b0dce7bc12f7;hb=d90d54eb25c326364575d062543664a214506210;hpb=810dacb371f6acdadad59185b609f46cdb9093b1;ds=sidebyside

diff --git a/enhance_table.py b/enhance_table.py
index 11f65eb..2b6bb8d 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -1,5 +1,11 @@
 #!//usr/bin/env python3
 
+import sys
+if len(sys.argv) != 2:
+    print('Expecting infections table file path as only argument.')
+    exit(1)
+infections_table = sys.argv[1]
+
 # District population numbers as per Wikipedia.
 district_pops = {
   'CW': 342332,
@@ -17,7 +23,7 @@ district_pops = {
   'sum': 3754418,
 }
 
-f = open('daily_infections_table.txt', 'r')
+f = open(infections_table, 'r')
 lines = f.readlines()
 f.close()