From 33adf11d8f9f0779bd9597db5469ba2d5894946f Mon Sep 17 00:00:00 2001 From: yeyemogui Date: Mon, 30 May 2022 22:46:30 +0800 Subject: [PATCH] 1) use pymysql to replace the obsoleted MySQLDb; 2) fix the empty return value issue which caused by API stock_zh_a_hist cannot support time format Y-M-D --- jobs/guess_indicators_daily_job.py | 4 ++-- libs/common.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/jobs/guess_indicators_daily_job.py b/jobs/guess_indicators_daily_job.py index 47a87319..f9b59d83 100644 --- a/jobs/guess_indicators_daily_job.py +++ b/jobs/guess_indicators_daily_job.py @@ -259,8 +259,8 @@ def apply_guess(tmp, stock_column): date = tmp["date"] code = tmp["code"] date_end = datetime.datetime.strptime(date, "%Y%m%d") - date_start = (date_end + datetime.timedelta(days=-100)).strftime("%Y-%m-%d") - date_end = date_end.strftime("%Y-%m-%d") + date_start = (date_end + datetime.timedelta(days=-100)).strftime("%Y%m%d") + date_end = date_end.strftime("%Y%m%d") # print(code, date_start, date_end) # open, high, close, low, volume, price_change, p_change, ma5, ma10, ma20, v_ma5, v_ma10, v_ma20, turnover # 使用缓存方法。加快计算速度。 diff --git a/libs/common.py b/libs/common.py index 55cd8ddf..eb855fd9 100644 --- a/libs/common.py +++ b/libs/common.py @@ -8,7 +8,7 @@ import time import sys import os -import MySQLdb +import pymysql from sqlalchemy import create_engine from sqlalchemy.types import NVARCHAR from sqlalchemy import inspect @@ -45,11 +45,11 @@ def engine_to_db(to_db): # 通过数据库链接 engine。 def conn(): try: - db = MySQLdb.connect(MYSQL_HOST, MYSQL_USER, MYSQL_PWD, MYSQL_DB, charset="utf8") + db = pymysql.connect(host=MYSQL_HOST, port=3306, user=MYSQL_USER, password=MYSQL_PWD, db=MYSQL_DB, charset="utf8") # db.autocommit = True except Exception as e: print("conn error :", e) - db.autocommit(on=True) + db.autocommit(True) return db.cursor()