| 1234567891011121314151617181920 |
- 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)
|
