ソースを参照

:bug: fix circular import error

Jeremy Zheng 9 ヶ月 前
コミット
04423218df

+ 1 - 9
ai-translate/ai_translate/__init__.py

@@ -1,8 +1,6 @@
 import logging
 import tomllib
 import json
-import os
-import sys
 
 import pika
 from redis.cluster import RedisCluster
@@ -10,14 +8,8 @@ from types import SimpleNamespace
 
 from .worker import handle_message
 
-logger = logging.getLogger(__name__)
-
 
-def is_stopped():
-    f = ".stop"
-    logger.warning("file %s exists, will be exit", f)
-    if os.path.exists(f):
-        sys.exit(0)
+logger = logging.getLogger(__name__)
 
 
 def open_redis_cluster(config):

+ 2 - 1
ai-translate/ai_translate/__main__.py

@@ -3,7 +3,8 @@ import argparse
 import sys
 import os
 
-from . import launch, is_stopped
+from . import launch
+from .utils import is_stopped
 
 logger = logging.getLogger(__name__)
 

+ 1 - 1
ai-translate/ai_translate/ai_translate.py → ai-translate/ai_translate/service.py

@@ -8,7 +8,7 @@ import time
 
 import requests
 
-from . import is_stopped
+from .utils import is_stopped
 
 logger = logging.getLogger(__name__)
 

+ 12 - 0
ai-translate/ai_translate/utils.py

@@ -0,0 +1,12 @@
+import logging
+import os
+import sys
+
+logger = logging.getLogger(__name__)
+
+
+def is_stopped():
+    f = ".stop"
+    if os.path.exists(f):
+        logger.warning("file %s exists, will be exit", f)
+        sys.exit(0)

+ 2 - 2
ai-translate/ai_translate/worker.py

@@ -1,8 +1,8 @@
 import logging
 
-from .ai_translate import AiTranslateService, SectionTimeout, Message
+from .service import AiTranslateService, SectionTimeout, Message
 from .decode_dataclass import ns_to_dataclass
-from . import is_stopped
+from .utils import is_stopped
 
 logger = logging.getLogger(__name__)