瀏覽代碼

修改mysql查询引擎

anweiguo 1 月之前
父節點
當前提交
a2ec47ccaf
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      common/database_dml.py

+ 3 - 2
common/database_dml.py

@@ -116,8 +116,9 @@ def get_data_fromMysql(params):
     #数据库读取实测气象
     engine = create_engine(f"mysql+pymysql://{mysql_conn}")
     # 定义SQL查询
-    env_df = pd.read_sql_query(query_sql, engine)
-    return env_df
+    with engine.connect() as conn:
+        df = pd.read_sql_query(query_sql, conn)
+    return df
 
 
 def insert_pickle_model_into_mongo(model, args):