Update datetime usage

stream_translator_gpt/common.py

@@ -1,6 +1,6 @@
 import os
 from abc import ABC, abstractmethod
-from datetime import datetime
+from datetime import datetime, timezone
 
 import google.generativeai as genai
 import numpy as np
@@ -37,7 +37,7 @@ def work(cls, **kwargs):
 
 
 def sec2str(second: float):
-    dt = datetime.utcfromtimestamp(second)
+    dt = datetime.fromtimestamp(second, tz=timezone.utc)
     result = dt.strftime('%H:%M:%S')
     result += ',' + str(int(second * 10 % 10))
     return result

stream_translator_gpt/llm_translator.py

@@ -4,7 +4,7 @@
 import time
 import re
 from collections import deque
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 
 import google.generativeai as genai
 from google.api_core.exceptions import InternalServerError, ResourceExhausted, TooManyRequests
@@ -50,7 +50,7 @@ def _parse_json_completion(completion):
 
 
 def _is_task_timeout(task: TranslationTask, timeout: float) -> bool:
-    return datetime.utcnow() - task.start_time > timedelta(seconds=timeout)
+    return datetime.now(timezone.utc) - task.start_time > timedelta(seconds=timeout)
 
 
 class LLMClint():
@@ -175,7 +175,7 @@ def __init__(self, llm_client: LLMClint, timeout: int, retry_if_translation_fail
 
     def _trigger(self, translation_task: TranslationTask):
         if not translation_task.start_time:
-            translation_task.start_time = datetime.utcnow()
+            translation_task.start_time = datetime.now(timezone.utc)
         translation_task.translation_failed = False
         thread = threading.Thread(target=self.llm_client.translate, args=(translation_task,))
         thread.daemon = True
@@ -225,7 +225,7 @@ def __init__(self, llm_client: LLMClint, timeout: int, retry_if_translation_fail
 
     def _trigger(self, translation_task: TranslationTask):
         if not translation_task.start_time:
-            translation_task.start_time = datetime.utcnow()
+            translation_task.start_time = datetime.now(timezone.utc)
         translation_task.translation_failed = False
         thread = threading.Thread(target=self.llm_client.translate, args=(translation_task,))
         thread.daemon = True