lei_y601/sqlbot_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat:添加达梦数据库支持以及对接pre环境数据

Browse Source
This commit is contained in:
雷雨
2025-09-24 14:17:07 +08:00
parent 23916fc328
commit c6d045859b
5 changed files with 195 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
main_service.py
View File

@@ -13,17 +13,23 @@ def connect_database(vn):
vn.connect_to_sqlite(config('SQLITE_DATABASE_URL', default=''))
elif db_type == 'mysql':
vn.connect_to_mysql(host=config('MYSQL_DATABASE_HOST', default=''),
port=int(config('MYSQL_DATABASE_PORT', default=3306)),
port=config('MYSQL_DATABASE_PORT', default=3306),
user=config('MYSQL_DATABASE_USER', default=''),
password=config('MYSQL_DATABASE_PASSWORD', default=''),
dbname=config('MYSQL_DATABASE_DBNAME', default=''))
elif db_type == 'postgresql':
database=config('MYSQL_DATABASE_DBNAME', default=''))
elif db_type == 'dameng':
# 待补充
pass
vn.connect_to_dameng(
host=config('DAMENG_DATABASE_HOST', default=''),
port=config('DAMENG_DATABASE_PORT', default=3306),
user=config('DAMENG_DATABASE_USER', default=''),
password=config('DAMENG_DATABASE_PASSWORD', default=''),
)
else:
pass
def load_train_data_ddl(vn: CustomVanna):
vn.train()