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

app/service/brand_dna/service.py

@@ -9,15 +9,16 @@ import torch.nn.functional as F
 import tritonclient.http as httpclient
 from minio import Minio
 
-from app.core.config import MINIO_URL, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE, DESIGN_MODEL_URL, CATEGORY_PATH
+from app.core.config import DESIGN_MODEL_URL
+from app.core.config import settings
 from app.schemas.brand_dna import BrandDnaModel
-from app.service.attribute.config import local_debug_const, const
+from app.service.attribute.config import const
 from app.service.utils.generate_uuid import generate_uuid
 from app.service.utils.new_oss_client import oss_upload_image, oss_get_image
 
-logger = logging.getLogger()
+minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secret_key=settings.MINIO_SECRET, secure=settings.MINIO_SECURE)
 
-minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
+logger = logging.getLogger()
 
 
 class BrandDna:
@@ -25,7 +26,7 @@ class BrandDna:
         self.sketch_bucket = "test"
         self.image_url = request_item.image_url
         self.is_brand_dna = request_item.is_brand_dna
-        self.attr_type = pd.read_csv(CATEGORY_PATH)
+        self.attr_type = pd.read_csv(settings.CATEGORY_PATH)
         # self.attr_type = pd.read_csv(r"E:\workspace\trinity_client_aida\app\service\attribute\config\descriptor\category\category_dis.csv")
         self.att_client = httpclient.InferenceServerClient(url=DESIGN_MODEL_URL)
         self.seg_client = httpclient.InferenceServerClient(url='10.1.1.243:30000')