improve

coldStartTasks/ai_pipeline/extract_video_best_frame.py

@@ -168,7 +168,7 @@ class ExtractVideoBestFrame:
                 datetime.datetime.now(),
                 task_id,
                 const.PROCESSING_STATUS,
-            )
+            ),
         )
         return update_rows
 
@@ -208,7 +208,7 @@ class ExtractVideoBestFrame:
                     "task_id": task["id"],
                     "track_back": traceback.format_exc(),
                     "error": str(e),
-                }
+                },
             )
             update_task_queue_status(
                 db_client=self.db_client,
@@ -219,7 +219,9 @@ class ExtractVideoBestFrame:
             )
             return None
 
-    def upload_video_to_gemini_ai(self, max_processing_pool_size: int = const.POOL_SIZE) -> None:
+    def upload_video_to_gemini_ai(
+        self, max_processing_pool_size: int = const.POOL_SIZE
+    ) -> None:
         # upload video to gemini ai
         roll_back_lock_tasks_count = self._roll_back_lock_tasks(task="upload")
         log(
@@ -269,8 +271,8 @@ class ExtractVideoBestFrame:
                         data={
                             "task_id": task["id"],
                             "file_name": file_name,
-                            "state": state
-                        }
+                            "state": state,
+                        },
                     )
 
                 case "FAILED":
@@ -297,8 +299,8 @@ class ExtractVideoBestFrame:
                         data={
                             "task_id": task["id"],
                             "file_name": file_name,
-                            "state": state
-                        }
+                            "state": state,
+                        },
                     )
 
                 case "ACTIVE":
@@ -335,8 +337,8 @@ class ExtractVideoBestFrame:
                                 "task_id": task["id"],
                                 "file_name": file_name,
                                 "state": state,
-                                "best_frame_tims_ms": best_frame_tims_ms
-                            }
+                                "best_frame_tims_ms": best_frame_tims_ms,
+                            },
                         )
                     except Exception as e:
                         log(
@@ -347,7 +349,7 @@ class ExtractVideoBestFrame:
                                 "task_id": task["id"],
                                 "track_back": traceback.format_exc(),
                                 "error": str(e),
-                            }
+                            },
                         )
                         update_task_queue_status(
                             db_client=self.db_client,
@@ -366,7 +368,7 @@ class ExtractVideoBestFrame:
                     "task_id": task["id"],
                     "track_back": traceback.format_exc(),
                     "error": str(e),
-                }
+                },
             )
             update_task_queue_status(
                 db_client=self.db_client,
@@ -407,8 +409,8 @@ class ExtractVideoBestFrame:
                     "task_id": task["id"],
                     "video_oss_path": task["video_oss_path"],
                     "time_millisecond_str": time_str,
-                    "response": response
-                }
+                    "response": response,
+                },
             )
             if response["success"] and response["data"]:
                 cover_oss_path = response["data"]
@@ -429,8 +431,8 @@ class ExtractVideoBestFrame:
                 data={
                     "task_id": task["id"],
                     "video_oss_path": task["video_oss_path"],
-                    "time_millisecond_str": time_str
-                }
+                    "time_millisecond_str": time_str,
+                },
             )
             update_task_queue_status(
                 db_client=self.db_client,