Agent/examples/content_finder/db/connection.py

"""共享数据库连接"""

import os

import pymysql


def get_connection():
    """获取数据库连接（与 store_results_mysql、schedule 共用配置）"""
    host = os.getenv("DB_HOST", "rm-t4nh1xx6o2a6vj8qu3o.mysql.singapore.rds.aliyuncs.com")
    port = int(os.getenv("DB_PORT", "3306"))
    user = os.getenv("DB_USER", "content_rw")
    password = os.getenv("DB_PASSWORD", "bC1aH4bA1lB0")
    database = os.getenv("DB_NAME", "content-deconstruction-supply")

    return pymysql.connect(
        host=host,
        port=port,
        user=user,
        password=password,
        database=database,
        charset="utf8mb4",
        cursorclass=pymysql.cursors.DictCursor,
        autocommit=True,
    )