Browse Source

:loud_sound: add ai-worker file logger

Jeremy Zheng 8 months ago
parent
commit
259b2a95ff

+ 6 - 4
ai-translate/ai_translate/__main__.py

@@ -2,6 +2,7 @@ import logging
 import argparse
 import argparse
 import sys
 import sys
 import os
 import os
+import datetime
 
 
 from . import launch
 from . import launch
 from .utils import is_stopped
 from .utils import is_stopped
@@ -27,11 +28,12 @@ def main():
     args = parser.parse_args()
     args = parser.parse_args()
 
 
     if args.debug:
     if args.debug:
-        logging.basicConfig(
-            level=logging.DEBUG if args.debug else logging.INFO, format='%(levelname)-5s %(asctime)s(%(pathname)s %(lineno)d): %(message)s')
+        now = datetime.datetime.now()
+        logging.basicConfig(filename=now.strftime("%Y%m%d%H%M%S.log"),
+                            level=args.debug, format='%(levelname)-5s %(asctime)s(%(pathname)s %(lineno)d): %(message)s')
     else:
     else:
-        logging.basicConfig(
-            level=logging.DEBUG if args.debug else logging.INFO, format='%(levelname)-5s %(asctime)s(%(module)s): %(message)s')
+        logging.basicConfig(level=logging.INFO,
+                            format='%(levelname)-5s %(asctime)s(%(module)s): %(message)s')
     is_stopped()
     is_stopped()
     try:
     try:
         launch(args.name, args.queue, args.config)
         launch(args.name, args.queue, args.config)

+ 1 - 1
deploy/roles/mint-v2.1/templates/containers/ai-translate.sh.j2

@@ -8,7 +8,7 @@ source $HOME/python3/bin/activate
 
 
 cd $WORK_DIR/
 cd $WORK_DIR/
 # https://github.com/iapt-platform/mint/blob/df8e1cf7ade16d17add360e7a869540c1ddaf1b9/api-v8/config/mint.php#L129
 # https://github.com/iapt-platform/mint/blob/df8e1cf7ade16d17add360e7a869540c1ddaf1b9/api-v8/config/mint.php#L129
-python3 -m ai_translate -c config.toml -n ai.translate -q ai_translate_v2
+python3 -m ai_translate -d -c config.toml -n ai.translate -q ai_translate_v2
 
 
 ai_translate_v2
 ai_translate_v2
 exit 0
 exit 0