Changed European data source and format

[covid19.git] / publish.py

diff --git a/publish.py b/publish.py
index 8c970560fa9fa4840c5e966cf7cb85e3f98c3ede..8dd4b0d4da946baad0372de1495afa5ff3add313 100644 (file)
--- a/publish.py
+++ b/publish.py
@@ -28,21 +28,9 @@ engine = sqlalchemy.create_engine(connection_string)
 
 
 # %%
-# DEATH_COUNT_THRESHOLD = 10
-COUNTRIES_CORE = tuple('IT DE UK ES IE FR BE'.split())
-# COUNTRIES_NORDIC = 'SE NO DK FI UK'.split()
-# COUNTRIES_FRIENDS = 'IT UK ES BE SI MX'.split()
-# # COUNTRIES_FRIENDS = 'IT UK ES BE SI PT'.split()
+COUNTRIES_CORE = tuple(sorted('ITA DEU GBR ESP IRL FRA BEL'.split()))
+COUNTRIES_FRIENDS = tuple('ITA GBR ESP BEL SVN MEX'.split())
 
-# COUNTRIES_AMERICAS = ['AG', 'AR', 'AW', 'BS', 'BB', 'BZ', 'BM', 'BO', 'BR', 'VG', 'KY', # excluding Canada and USA
-#        'CL', 'CO', 'CR', 'CU', 'CW', 'DM', 'DO', 'EC', 'SV', 'GL', 'GD', 'GT',
-#        'GY', 'HT', 'HN', 'JM', 'MX', 'MS', 'NI', 'PA', 'PY', 'PE', 'PR', 'KN',
-#        'LC', 'VC', 'SX', 'SR', 'TT', 'TC', 'VI', 'UY', 'VE']
-# COUNTRIES_OF_INTEREST = list(set(COUNTRIES_CORE + COUNTRIES_FRIENDS))
-# COUNTRIES_ALL = list(set(COUNTRIES_CORE + COUNTRIES_FRIENDS + COUNTRIES_NORDIC + COUNTRIES_AMERICAS))
-
-
-# # Write results to summary file
 
 # %%
 def singleton_sql_value(engine, query_string):
@@ -56,7 +44,7 @@ last_uk_date = singleton_sql_value(engine, 'select max(date) from uk_data')
 
 
 # %%
-last_intl_date = singleton_sql_value(engine, 'select max(report_date) from weekly_cases')
+last_intl_date = singleton_sql_value(engine, 'select max(date) from weekly_cases')
 
 
 # %%
@@ -95,11 +83,11 @@ with open('covid_summary.md', 'a') as f:
 
 
 # %%
-query_string = f'''select geo_id, country_name, culm_deaths 
-from weekly_cases join countries using (geo_id)
-where geo_id in {COUNTRIES_CORE} 
-    and report_date = '{last_intl_date.isoformat()}'
-order by geo_id'''
+query_string = f'''select country_code, country, culm_deaths 
+from weekly_cases join countries using (country_code)
+where country_code in {COUNTRIES_CORE} 
+    and date = '{last_intl_date.isoformat()}'
+order by country_code'''
 
 with engine.connect() as conn: 
     results = list(conn.execute(query_string))
@@ -244,7 +232,7 @@ transfer_files = [
     'covid_deaths_total_linear.png',
     'cases_and_deaths.png',
     'cases_and_deaths_last_60_days.png',
-    'deaths-radar-2021.png',
+    'deaths_radar_2021.png',
     'covid_deaths_per_week.png',
     'fraction_positive_tests.png',
     'tests_and_cases.png',
@@ -267,7 +255,7 @@ for f in transfer_files:
         os.system(f'scp {f} neil@ogedei:/var/www/scripts.njae.me.uk/covid/')
         print(f'Transferred {f}')
     else:
-        print('Cannot transfer {f}: file does not exist')
+        print(f'Cannot transfer {f}: file does not exist')
 
 # %%