wiki_crawler/backend/config.py

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()