from urllib.parse import quote_plus
from sqlalchemy import create_engine
from pqai_agent import configs

def create_sql_engine(config):
    user = config['user']
    passwd = quote_plus(config['password'])
    host = config['host']
    db_name = config['database']
    charset = config.get('charset', 'utf8mb4')
    engine = create_engine(f'mysql+pymysql://{user}:{passwd}@{host}/{db_name}?charset={charset}')
    return engine

def create_ai_agent_db_engine():
    config = configs.get()['database']['ai_agent']
    return create_sql_engine(config)

def create_growth_db_engine():
    config = configs.get()['database']['growth']
    return create_sql_engine(config)