Now connect is everywhere

Author: maxnoe

erna/scripts/check_availability.py

@@ -94,7 +94,8 @@ def main(config, verbose, start, end):
     basedir = '/fact/raw'
 
     fact_db = create_engine(db_specification.format(**config['fact_database']))
-    runs = pd.read_sql_query(run_query.format(start=start, end=end), fact_db)
+    with fact_db.connect() as conn:
+        runs = pd.read_sql_query(run_query.format(start=start, end=end), conn)
     runs['night'] = runs['night'].apply(night_int_to_date)
 
     log.info('basedir is: {}'.format(basedir))

migration_scripts/update_existing_rows_drsstep.py

@@ -20,10 +20,10 @@ def main(config):
     database.connect()
 
     engine = create_mysql_engine(**config['fact_database'])
-
-    df = pd.read_sql_table('RunInfo', engine, columns=[
-        'fNight', 'fRunID', 'fDrsStep', 'fROI'
-    ])
+    with engine.connect() as conn:
+        df = pd.read_sql_table('RunInfo', conn, columns=[
+            'fNight', 'fRunID', 'fDrsStep', 'fROI'
+        ])
 
     df.set_index(['fNight', 'fRunID'], inplace=True)