feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/service/generate_image/service_generate_single_logo.py

@@ -11,18 +11,16 @@ import json
 import logging
 import time
 
-import cv2
 import numpy as np
 import redis
+import tritonclient.grpc as grpcclient
 from PIL import Image
-from minio import Minio
 from tritonclient.utils import np_to_triton_dtype
 
-from app.core.config import *
-import tritonclient.grpc as grpcclient
+from app.core.config import settings, GI_RABBITMQ_QUEUES, GSL_MODEL_NAME, GSL_MODEL_URL
 from app.schemas.generate_image import GenerateSingleLogoImageModel
 from app.service.generate_image.utils.mq import publish_status
-from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_SDXL_image
+from app.service.generate_image.utils.upload_sd_image import upload_SDXL_image
 
 logger = logging.getLogger()
 
@@ -30,7 +28,7 @@ logger = logging.getLogger()
 class GenerateSingleLogoImage:
     def __init__(self, request_data):
         self.grpc_client = grpcclient.InferenceServerClient(url=GSL_MODEL_URL)
-        self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
+        self.redis_client = redis.StrictRedis(host=settings.REDIS_HOST, port=settings.REDIS_PORT, db=settings.REDIS_DB, decode_responses=True)
         self.batch_size = 1
         self.category = "single_logo"
         self.negative_prompts = "bad, ugly"
@@ -93,12 +91,12 @@ class GenerateSingleLogoImage:
             raise Exception(str(e))
         finally:
             dict_generate_data, str_generate_data = self.read_tasks_status()
-            if not DEBUG:
+            if not settings.DEBUG:
                 publish_status(str_generate_data, GI_RABBITMQ_QUEUES)
 
 
 def infer_cancel(tasks_id):
-    redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
+    redis_client = redis.StrictRedis(host=settings.REDIS_HOST, port=settings.REDIS_PORT, db=settings.REDIS_DB, decode_responses=True)
     data = {'tasks_id': tasks_id, 'status': 'REVOKED', 'message': "revoked", 'data': 'revoked'}
     generate_data = json.dumps(data)
     redis_client.set(tasks_id, generate_data)