78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
import os
|
|
|
|
import pika
|
|
from dotenv import load_dotenv
|
|
from pydantic import BaseSettings
|
|
|
|
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
|
|
load_dotenv(os.path.join(BASE_DIR, '.env'))
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
PROJECT_NAME = os.getenv('PROJECT_NAME', 'FASTAPI BASE')
|
|
SECRET_KEY = os.getenv('SECRET_KEY', '')
|
|
API_PREFIX = ''
|
|
BACKEND_CORS_ORIGINS = ['*']
|
|
DATABASE_URL = os.getenv('SQL_DATABASE_URL', '')
|
|
ACCESS_TOKEN_EXPIRE_SECONDS: int = 60 * 60 * 24 * 7 # Token expired after 7 days
|
|
SECURITY_ALGORITHM = 'HS256'
|
|
LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
|
|
|
|
|
|
DEBUG = True
|
|
if DEBUG:
|
|
LOGS_PATH = "logs/"
|
|
CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv"
|
|
else:
|
|
LOGS_PATH = "app/logs/"
|
|
CATEGORY_PATH = "app/service/attribute/config/descriptor/category/category_dis.csv"
|
|
|
|
# RABBITMQ_ENV = "" # 生产环境
|
|
# RABBITMQ_ENV = "-dev" # 开发环境
|
|
RABBITMQ_ENV = "-local" # 本地测试环境
|
|
|
|
settings = Settings()
|
|
|
|
# minio 配置
|
|
MINIO_URL = "www.minio.aida.com.hk:9000"
|
|
MINIO_ACCESS = 'vXKFLSJkYeEq2DrSZvkB'
|
|
MINIO_SECRET = 'uKTZT3x7C43WvPN9QTc99DiRkwddWZrG9Uh3JVlR'
|
|
MINIO_SECURE = True
|
|
|
|
# redis 配置
|
|
REDIS_HOST = "10.1.1.240"
|
|
REDIS_PORT = "6379"
|
|
REDIS_DB = "2"
|
|
|
|
# rabbitmq config
|
|
RABBITMQ_PARAMS = {
|
|
"host": "18.167.251.121",
|
|
"port": 5672,
|
|
"credentials": pika.credentials.PlainCredentials(username='rabbit', password='123456'),
|
|
"virtual_host": "/"
|
|
}
|
|
|
|
# attribute service config
|
|
ATT_TRITON_URL = "10.1.1.240:8020"
|
|
|
|
# SR service config
|
|
SR_MODEL_NAME = "super_resolution"
|
|
SR_TRITON_URL = "10.1.1.240:10031"
|
|
SR_MINIO_BUCKET = "aida-users"
|
|
SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", f"SuperResolution{RABBITMQ_ENV}")
|
|
|
|
# GenerateImage service config
|
|
GI_MODEL_NAME = 'stable_diffusion_xl_lcm'
|
|
GI_MODEL_URL = '10.1.1.150:8001'
|
|
GI_MINIO_BUCKET = "aida-users"
|
|
GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}")
|
|
GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg"
|
|
|
|
# SEG service config
|
|
SEG_MODEL_URL = '10.1.1.240:10000'
|
|
SEGMENTATION = {
|
|
"name": "seg_ocrnet_hr18",
|
|
"input": "seg_input__0",
|
|
"output": "seg_output__0",
|
|
}
|