26 lines
900 B
Python
26 lines
900 B
Python
import os
|
|
|
|
class Settings:
|
|
# 数据库配置
|
|
DB_USER: str = "postgres"
|
|
DB_PASS: str = "DXC_welcome001"
|
|
DB_HOST: str = "8.155.144.6"
|
|
DB_PORT: str = "25432"
|
|
DB_NAME: str = "wiki_crawler"
|
|
|
|
DASHSCOPE_API_KEY: str = "sk-8b091493de594c5e9eb42f12f1cc5805"
|
|
FIRECRAWL_API_KEY: str = "fc-8a2af3fb6a014a27a57dfbc728cb7365"
|
|
@property # property 方法，意义：将方法转换为属性，调用时不需要加括号
|
|
def DATABASE_URL(self) -> str:
|
|
url = f"postgresql+psycopg2://{self.DB_USER}:{self.DB_PASS}@{self.DB_HOST}:{self.DB_PORT}/{self.DB_NAME}"
|
|
return url
|
|
|
|
def API_KEY(self, type: str) -> str:
|
|
if type == "dashscope":
|
|
return self.DASHSCOPE_API_KEY
|
|
elif type == "firecrawl":
|
|
return self.FIRECRAWL_API_KEY
|
|
else:
|
|
raise ValueError(f"Unknown API type: {type}")
|
|
|
|
settings = Settings() |