소스 검색

修改mysql查询引擎

anweiguo 3 달 전
부모
커밋
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):