acquire-lead/config/settings.py

"""项目配置：从 .env 读取数据库与采集参数。"""

from pathlib import Path

from dotenv import load_dotenv
import os

ROOT_DIR = Path(__file__).resolve().parent.parent
load_dotenv(ROOT_DIR / ".env")

DB_HOST = os.getenv("DB_HOST", "localhost")
DB_PORT = int(os.getenv("DB_PORT", "3306"))
DB_USER = os.getenv("DB_USER", "root")
DB_PASSWORD = os.getenv("DB_PASSWORD", "")
DB_NAME = os.getenv("DB_NAME", "acquirelead")

SEARCH_KEYWORD = os.getenv("SEARCH_KEYWORD", "怎么学好英语？")
COMMENT_LIMIT = int(os.getenv("COMMENT_LIMIT", "10"))
VIDEO_LIMIT = int(os.getenv("VIDEO_LIMIT", "5"))

BROWSER_DATA_DIR = ROOT_DIR / "browser_data" / "douyin"
DOUYIN_URL = "https://www.douyin.com"

# leads.status
LEAD_STATUS_COLLECTED = 1

# 表名前缀（迁移后库表为 douyin_videos / douyin_leads 等）
TABLE_PREFIX = os.getenv("TABLE_PREFIX", "douyin_")


def database_url() -> str:
    from urllib.parse import quote_plus

    password = quote_plus(DB_PASSWORD)
    return (
        f"mysql+pymysql://{DB_USER}:{password}"
        f"@{DB_HOST}:{DB_PORT}/{DB_NAME}?charset=utf8mb4"
    )