import os


TUSHARE_TOKEN = os.getenv(
    "TUSHARE_TOKEN",
    "0fad3cf498757089e2630028455d5bbe1637475788bcdaa1f2175e93",
)

# 阿里云远程数据库
# DB_CONFIG = {
#     "host": os.getenv("MYSQL_HOST", "8.159.129.156"),
#     "port": int(os.getenv("MYSQL_PORT", "10836")),
#     "user": os.getenv("MYSQL_USER", "yangfan"),
#     "password": os.getenv("MYSQL_PASSWORD", "aA%8023321088"),
#     "database": os.getenv("MYSQL_DATABASE", "level"),
#     "charset": os.getenv("MYSQL_CHARSET", "utf8mb4"),
# }

# mac本地mysql
DB_CONFIG = {
    "host": os.getenv("MYSQL_HOST", "localhost"),
    "port": int(os.getenv("MYSQL_PORT", "3306")),
    "user": os.getenv("MYSQL_USER", "yangfan"),
    "password": os.getenv("MYSQL_PASSWORD", "admin"),
    "database": os.getenv("MYSQL_DATABASE", "level"),
    "charset": os.getenv("MYSQL_CHARSET", "utf8mb4"),
}


def get_db_url():
    return (
        f"mysql+pymysql://{DB_CONFIG['user']}@{DB_CONFIG['host']}:"
        f"{DB_CONFIG['port']}/{DB_CONFIG['database']}?charset={DB_CONFIG['charset']}"
    )


def get_engine():
    from sqlalchemy import create_engine

    return create_engine(
        get_db_url(),
        connect_args={"password": DB_CONFIG["password"]},
        pool_pre_ping=True,
    )