import os
from dotenv import load_dotenv

load_dotenv()

SOURCE_DB = {
    'host': os.environ['SOURCE_DB_HOST'],
    'port': int(os.environ['SOURCE_DB_PORT']),
    'user': os.environ['SOURCE_DB_USER'],
    'password': os.environ['SOURCE_DB_PASSWORD'],
    'database': os.environ['SOURCE_DB_NAME'],
    'charset': 'utf8mb4',
}

HK_SONGS_DB = {
    'host': os.environ['TARGET_DB_HOST'],
    'port': int(os.environ['TARGET_DB_PORT']),
    'user': os.environ['TARGET_DB_USER'],
    'password': os.environ['TARGET_DB_PASSWORD'],
    'database': os.environ['TARGET_DB_NAME'],
    'charset': 'utf8mb4',
}

CRAWLER_DB = {
    'host': os.environ['CRAWLER_DB_HOST'],
    'port': int(os.environ.get('CRAWLER_DB_PORT', '5432')),
    'user': os.environ['CRAWLER_DB_USER'],
    'password': os.environ['CRAWLER_DB_PASSWORD'],
    'database': os.environ['CRAWLER_DB_NAME'],
    'ssl_context': None if os.environ.get('CRAWLER_DB_SSL', 'false').lower() != 'true' else True,
}

OSS_CONFIG = {
    'access_key_id': os.environ['OSS_ACCESS_KEY_ID'],
    'access_key_secret': os.environ['OSS_ACCESS_KEY_SECRET'],
    'endpoint': os.environ['OSS_ENDPOINT'],
    'bucket_name': os.environ['OSS_BUCKET_NAME'],
    'base_url': f"https://{os.environ['OSS_BUCKET_NAME']}.{os.environ['OSS_ENDPOINT']}",
}

PLATFORM_QQ = '1'
PLATFORM_KUGOU = '2'
PLATFORM_NETEASE = '4'
PLATFORMS = [PLATFORM_QQ, PLATFORM_KUGOU, PLATFORM_NETEASE]

BATCH_SIZE = 1