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

app/service/generate_image/service_generate_relight_image.py

@@ -18,11 +18,11 @@ import tritonclient.grpc as grpcclient
 from PIL import Image
 from tritonclient.utils import np_to_triton_dtype
 
-from app.core.config import *
+from app.core.config import settings, GRI_MODEL_URL, GRI_MODEL_NAME_SINGLE, GRI_MODEL_NAME_OVERALL, GRI_RABBITMQ_QUEUES
 from app.schemas.generate_image import GenerateRelightImageModel
 from app.service.generate_image.utils.mq import publish_status
 from app.service.generate_image.utils.upload_sd_image import upload_SDXL_image
-from app.service.utils.oss_client import oss_get_image
+from app.service.utils.new_oss_client import oss_get_image
 
 logger = logging.getLogger()
 
@@ -30,7 +30,7 @@ logger = logging.getLogger()
 class GenerateRelightImage:
     def __init__(self, request_data):
         self.grpc_client = grpcclient.InferenceServerClient(url=GRI_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.category = "relight_image"
         self.batch_size = 1
         self.prompt = request_data.prompt
@@ -134,9 +134,10 @@ class GenerateRelightImage:
             raise Exception(str(e))
         finally:
             dict_gen_product_data, str_gen_product_data = self.read_tasks_status()
-            if not DEBUG:
+            if not settings.DEBUG:
                 publish_status(str_gen_product_data, GRI_RABBITMQ_QUEUES)
 
+
 def pre_processing_image(image_url):
     image = oss_get_image(bucket=image_url.split('/')[0], object_name=image_url[image_url.find('/') + 1:], data_type="PIL")
     # 目标图片的尺寸
@@ -178,8 +179,9 @@ def pre_processing_image(image_url):
     # image = cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA)
     return image
 
+
 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'}
     gen_product_data = json.dumps(data)
     redis_client.set(tasks_id, gen_product_data)