Minor code re-arrangement.

diff --git a/enhance_table.py b/enhance_table.py
index 2a89c4b0113c0331483f4ebd8009831fd2cfd916..774b333a77cf96951609c4363f219f96d158244a 100755 (executable)
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -1,11 +1,5 @@
 #!//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,
@@ -23,6 +17,12 @@ district_pops = {
   'sum': 3754418,
 }
 
+# Read infections table file lines.
+import sys
+if len(sys.argv) != 2:
+    print('Expecting infections table file path as only argument.')
+    exit(1)
+infections_table = sys.argv[1]
 f = open(infections_table, 'r')
 lines = f.readlines()
 f.close()