|
@@ -174,6 +174,9 @@ def export_script_file(item: List[QueryWordItem]):
|
|
|
script_df = pd.read_sql(sql, ai_word_engine)
|
|
|
out_df = video_query_word_df.merge(script_df, on='signature', how='inner')
|
|
|
|
|
|
+ # 手动数据库关闭连接
|
|
|
+ ai_word_engine.dispose()
|
|
|
+
|
|
|
# 3 返回流数据
|
|
|
if not out_df.empty:
|
|
|
bio = BytesIO()
|
|
@@ -279,6 +282,10 @@ def get_query_word_task_info_lst(item: QueryWordTaskInfoLst):
|
|
|
f"and ('{item.source_code}' = '[]' or source_code in {tuple(source_code_lst)}) " \
|
|
|
f"and ('{item.search_word}' = '' or query_word = '{item.search_word}')"
|
|
|
df = pd.read_sql(sql, ai_word_engine)
|
|
|
+
|
|
|
+ # 手动数据库关闭连接
|
|
|
+ ai_word_engine.dispose()
|
|
|
+
|
|
|
if not df.empty:
|
|
|
df['source_name'] = df['source_code'].apply(lambda x: source_name_map[x]['name'])
|
|
|
df = df[['source_name', 'query_word', 'stat_date', 'script_num', 'task_status']]
|
|
@@ -338,6 +345,10 @@ def add_script_config(item: AddScriptConfig):
|
|
|
con=ai_word_engine,
|
|
|
if_exists='append',
|
|
|
index=False)
|
|
|
+
|
|
|
+ # 手动数据库关闭连接
|
|
|
+ ai_word_engine.dispose()
|
|
|
+
|
|
|
logger.info(f"request body: {item}, code:0, message: add_script_config success")
|
|
|
return {"code": 0,
|
|
|
"message": "add success",
|
|
@@ -389,6 +400,10 @@ def delete_script_config(item: DeleteScriptConfig):
|
|
|
dtype=column_type)
|
|
|
else:
|
|
|
response.message = f'没有获取到 {item.config_id} 对应的数据'
|
|
|
+
|
|
|
+ # 手动数据库关闭连接
|
|
|
+ ai_word_engine.dispose()
|
|
|
+
|
|
|
logger.info(f"request body: {item}, response: {response}")
|
|
|
except:
|
|
|
response.code = -1
|