Fix last version of Peewee and others methods deprecated

oct/core/turrets_manager.py

@@ -100,7 +100,7 @@ def write(self, turret):
 
         :param dict turret_data: the data of the turret to write
         """
-        with db.execution_context():
+        with db:
             turret.save()
 
     def publish(self, message, channel=None):

oct/results/models.py

@@ -74,10 +74,10 @@ def set_database(db_url, proxy, config):
     db_config = config.get('results_database', {}).get('params', {})
 
     if 'testing' in config and config['testing'] is True:
-        database = connect('sqlite:////tmp/results.sqlite', check_same_thread=False, threadlocals=True)
+        database = connect('sqlite:////tmp/results.sqlite', check_same_thread=False)
     else:
         if os.path.isfile(db_url) or os.path.isdir(os.path.dirname(db_url)):
             db_url = "sqlite:///" + db_url
-            db_config.update(check_same_thread=False, threadlocals=True)
+            db_config.update(check_same_thread=False)
         database = connect(db_url, **db_config)
     proxy.initialize(database)

oct/results/report.py

@@ -40,7 +40,7 @@ def _init_dataframes(self):
         """Initialise the main dataframe for the results and the custom timers dataframes
         """
         df = pd.read_sql_query("SELECT elapsed, epoch, scriptrun_time, custom_timers FROM result ORDER BY epoch ASC",
-                               db.get_conn())
+                               db.connection())
 
         self._get_all_timers(df)
         self.main_results = self._get_processed_dataframe(df)

oct/results/stats_handler.py

@@ -23,7 +23,7 @@ def init_stats(output_dir, config):
 
     tables_to_create = [t for t in [Result, Turret] if not t.table_exists()]
 
-    db.get_conn()
+    db.connection()
     db.create_tables(tables_to_create)
 
 
@@ -42,7 +42,7 @@ def write_result(self, data):
         self.results.append(data)
 
         if len(self.results) >= 150:  # 150 rows for SQLite default limit
-            with db.execution_context():
+            with db:
                 with db.atomic():
                     Result.insert_many(self.results).execute()
                 del self.results[:]
@@ -52,7 +52,7 @@ def write_remaining(self):
         """
         if not self.results:
             return
-        with db.execution_context():
+        with db:
             with db.atomic():
                 Result.insert_many(self.results).execute()
         del self.results[:]