open-douyin/config.py

import os
import socket
import sys
from loguru import logger
# 是否为生产环境， None 则是调试环境（开发环境）
PRODUCE_ENV = os.environ.get("PRODUCE_ENV", None)
WORK_DIR = os.path.dirname(__file__)
LOG_FILE = os.path.join(WORK_DIR,"log", "1.log")


FORMAT = '<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{file}</cyan>:<cyan>{line}</cyan> :<cyan>{function}</cyan> - {message}'
LOG_LEVEL = "DEBUG"
logger.remove()
# logger.add(sys.stderr, format='<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>')
logger.add(sys.stderr, format=FORMAT, level="INFO")
logger.add(LOG_FILE, format=FORMAT, level="DEBUG")
logger.info(f"load config:{ __file__}")

# openssl rand -hex 32
os.environ["SECRET_KEY"]="34581f02dcdbab9dc176d6bb578fb15cc6b8e66159865c14e1fc81cd1d92c2a6"
os.environ["DB_URL"]="postgresql://pg:pg@sv-v:5432/douyin"
os.environ["CLIENT_KEY"] = 'aw6aipmfdtplwtyq'
os.environ["CLIENT_SECRET"] = '53cf3dcd2663629e8a773ab59df0968b'
DOUYIN_OPEN_API="https://open.douyin.com"


# HOST = socket.gethostbyname(socket.gethostname())
# 这个网址 https://open-douyin.magong.site 对应这台服务器的 192.168.1.32:8600 端口，因为这台服务器没有公网ip，所以在本地计算机无法通过  http://192.168.1.32:8600/ 访问到 fastapi 接口，只能通过 https://open-douyin.magong.site/ 访问
HOST = '::'

PORT = 8601 if os.environ.get("USER")=="mrh" else 8600
GRPC_VECTOR="192.168.2.32:18600"
SECRET_KEY = os.environ.get("SECRET_KEY")

DB_URL=os.environ.get("DB_URL")
# 生产环境中绝对不能使用硬编码的 DATA 路径。否则后期负载均衡、数据扩容、迁移将会造成很大影响。
if not PRODUCE_ENV:
    MNT_DOUYIN_DATA="/home/user/code/open-douyin/test/data"
else:
    MNT_DOUYIN_DATA = os.environ["MNT_DOUYIN_DATA"]
logger.info(f"API URL:{HOST}:{PORT}")