import os
from addict import Dict

DATABASE = Dict({
    "ORIGINAL_MYSQL_USER": os.environ.get('ORIGINAL_MYSQL_USER', 'poweroa_xzjn'),
    "ORIGINAL_MYSQL_PASS": os.environ.get('ORIGINAL_MYSQL_PASS', 'EPK22xzjn'),
    "ORIGINAL_MYSQL_HOST": os.environ.get('ORIGINAL_MYSQL_HOST',
                                          'rm-uf690m4x98j71szlno.mysql.rds.aliyuncs.com'),
    "ORIGINAL_MYSQL_PORT": os.environ.get('ORIGINAL_MYSQL_PORT', 3306),
    "ORIGINAL_MYSQL_DB": os.environ.get('ORIGINAL_MYSQL_DB', "power_oa"),

    "SLAVE_MYSQL_USER": os.environ.get('SLAVE_MYSQL_USER', 'root'),
    "SLAVE_MYSQL_PASS": os.environ.get('SLAVE_MYSQL_PASS', 'afafa'),
    "SLAVE_MYSQL_HOST": os.environ.get('SLAVE_MYSQL_HOST', '140.143.35.79'),
    "SLAVE_MYSQL_PORT": os.environ.get('SLAVE_MYSQL_PORT', 8886),
    "SLAVE_MYSQL_DB": os.environ.get('SLAVE_MYSQL_DB', "etl_power_oa"),
    # "TRANSFORMATION_MYSQL_USER": os.environ.get('TRANSFORMATION_MYSQL_USER', 'root'),
    # "TRANSFORMATION_MYSQL_PASS": os.environ.get('TRANSFORMATION_MYSQL_PASS', 'Afa@1qaz2wsx'),
    # "TRANSFORMATION_MYSQL_HOST": os.environ.get('TRANSFORMATION_MYSQL_HOST', '188.131.235.243'),
    # "TRANSFORMATION_MYSQL_PORT": os.environ.get('TRANSFORMATION_MYSQL_PORT', 31007),
    # "TRANSFORMATION_MYSQL_DB": os.environ.get('TRANSFORMATION_MYSQL_DB', 'console')
    
    "TRANSFORMATION_MYSQL_USER": os.environ.get('TRANSFORMATION_MYSQL_USER', 'root'),
    "TRANSFORMATION_MYSQL_PASS": os.environ.get('TRANSFORMATION_MYSQL_PASS', '1qaz@WSX'),
    "TRANSFORMATION_MYSQL_HOST": os.environ.get('TRANSFORMATION_MYSQL_HOST', 'sh-cdb-ptpdw86e.sql.tencentcdb.com'),
    "TRANSFORMATION_MYSQL_PORT": os.environ.get('TRANSFORMATION_MYSQL_PORT', 60852),
    "TRANSFORMATION_MYSQL_DB": os.environ.get('TRANSFORMATION_MYSQL_DB', 'console')
})

SLAVE_MYSQL_URL = "mysql+mysqlconnector://{}:{}@{}:{}/{}".format(
    DATABASE.SLAVE_MYSQL_USER, DATABASE.SLAVE_MYSQL_PASS,
    DATABASE.SLAVE_MYSQL_HOST, DATABASE.SLAVE_MYSQL_PORT,
    DATABASE.SLAVE_MYSQL_DB)

SLAVE_CONNECTION = Dict({
    "database": SLAVE_MYSQL_URL,
    "pool": 5,
    "max_overflow": 10,
    "pool_timeout": 30,
    "pool_pre_ping": True
})

KAFKA_HOST = os.environ.get('KAFKA_HOST', '81.68.220.135')
KAFKA_PORT = os.environ.get('KAFKA_HOST', 9092)
if KAFKA_PORT:
    KAFKA_SERVER = ":".join([KAFKA_HOST, str(KAFKA_PORT)])
else:
    KAFKA_SERVER = KAFKA_HOST

KAFKA = Dict({
    "bootstrap_servers": [KAFKA_SERVER],
    "enable_auto_commit": True,
    "auto_commit_interval_ms": 3000,
    "consumer_timeout_ms": float('inf'),
    "auto_offset_reset": "earliest",
    "default_group_id": "etl_",
    "customers": [],
})
