brynq-sdk-task-scheduler 4.0.5__tar.gz → 4.0.7__tar.gz

{brynq_sdk_task_scheduler-4.0.5 → brynq_sdk_task_scheduler-4.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_task_scheduler
-Version: 4.0.5
+Version: 4.0.7
 Summary: Code to execute tasks in BrynQ.com with the task scheduler
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_task_scheduler-4.0.5 → brynq_sdk_task_scheduler-4.0.7}/brynq_sdk_task_scheduler/task_scheduler.py

@@ -69,6 +69,12 @@ class TaskScheduler:
             self.mysql_reachable = True
             try:
                 self.mysql = MySQL()
+                # override connection credentials always because they mightve been retrieved from an interface credential
+                self.mysql.host = os.getenv("MYSQL_HOST")
+                self.mysql.user = os.getenv("MYSQL_USER")
+                self.mysql.password = os.getenv("MYSQL_PASSWORD")
+                self.mysql.database = os.getenv("MYSQL_DATABASE")
+                self.mysql.port = 3306 if os.getenv("MYSQL_PORT") is None else int(os.getenv("MYSQL_PORT"))
                 self.mysql.ping()
             except Exception as e:
                 self.mysql_reachable = False
@@ -248,8 +254,12 @@ class TaskScheduler:
                                       filter=f'WHERE `reload_id` = {self.run_id}')
                     if self.email_after_errors:
                         self.email_errors(failed=True)
-                    # Remove the temp values from the variables table
-                    self.mysql.raw_query(f'UPDATE `task_variables` SET temp_value = null WHERE task_id = {self.task_id}', insert=True)
+
+                    # Remove the temp values from the variables table (legacy), if it doesn't exist, ignore the error. In that case it will already be flushed by the API endpoint
+                    try:
+                        self.mysql.raw_query(f'UPDATE `task_variables` SET temp_value = null WHERE task_id = {self.task_id}', insert=True)
+                    except Exception as e:
+                        pass
 
                     # Start the chained tasks if it there are tasks which should start if this one is failed
                     self.start_chained_tasks(finished_task_status='FAILED')
@@ -302,8 +312,11 @@ class TaskScheduler:
                                   values=['FinishedSuccess', f'{datetime.datetime.now()}'],
                                   filter=f'WHERE `reload_id` = {self.run_id}')
 
-            # Remove the temp values from the variables table
-            self.mysql.raw_query(f'UPDATE `task_variables` SET temp_value = null WHERE task_id = {self.task_id}', insert=True)
+            # Remove the temp values from the variables table if it exists (legacy)
+            try:
+                self.mysql.raw_query(f'UPDATE `task_variables` SET temp_value = null WHERE task_id = {self.task_id}', insert=True)
+            except Exception as e:
+                pass
 
             # Start the new task if it there is a task which should start if this one is finished
             self.start_chained_tasks(finished_task_status='SUCCESS')
@@ -375,7 +388,7 @@ class TaskScheduler:
                 logs = []
             logs.append(payload)
             with open(log_file_path, 'w') as f:
-                json.dump(logs, f)
+                json.dump(logs, f, default=str)
         except Exception as e:
             print(f"Error saving log locally: {e}")
 

{brynq_sdk_task_scheduler-4.0.5 → brynq_sdk_task_scheduler-4.0.7}/brynq_sdk_task_scheduler.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-task-scheduler
-Version: 4.0.5
+Version: 4.0.7
 Summary: Code to execute tasks in BrynQ.com with the task scheduler
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_task_scheduler-4.0.5 → brynq_sdk_task_scheduler-4.0.7}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_task_scheduler',
-    version='4.0.5',
+    version='4.0.7',
     description='Code to execute tasks in BrynQ.com with the task scheduler',
     long_description='Code to execute tasks in the BrynQ.com platform with the task scheduler',
     author='BrynQ',