修改limit_up数据来源
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import tushare as ts
|
||||
import pymysql
|
||||
import pandas as pd
|
||||
from datetime import datetime
|
||||
import time
|
||||
import argparse
|
||||
@@ -53,6 +54,14 @@ def str_to_date(date_str):
|
||||
return datetime.strptime(date_str, "%Y%m%d").date()
|
||||
|
||||
|
||||
def mysql_value(value):
|
||||
if pd.isna(value):
|
||||
return None
|
||||
if hasattr(value, "item"):
|
||||
return value.item()
|
||||
return value
|
||||
|
||||
|
||||
# ===== 获取交易日 =====
|
||||
def get_trade_dates(start_date, end_date):
|
||||
_, db_cursor = get_connection()
|
||||
@@ -90,15 +99,15 @@ def insert_data(df):
|
||||
ts_code,
|
||||
symbol,
|
||||
str_to_date(row['trade_date']), # 保持 DATE 类型
|
||||
row['open'],
|
||||
row['high'],
|
||||
row['low'],
|
||||
row['close'],
|
||||
row['pre_close'],
|
||||
row['change'],
|
||||
row['pct_chg'],
|
||||
row['vol'],
|
||||
row['amount']
|
||||
mysql_value(row['open']),
|
||||
mysql_value(row['high']),
|
||||
mysql_value(row['low']),
|
||||
mysql_value(row['close']),
|
||||
mysql_value(row['pre_close']),
|
||||
mysql_value(row['change']),
|
||||
mysql_value(row['pct_chg']),
|
||||
mysql_value(row['vol']),
|
||||
mysql_value(row['amount'])
|
||||
))
|
||||
|
||||
db_cursor.executemany(sql, data)
|
||||
@@ -148,7 +157,7 @@ def run(start_date, end_date):
|
||||
# ===== 输入参数 =====
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="同步日行情")
|
||||
parser.add_argument("--start-date", default="20260430", help="开始日期,格式 YYYYMMDD")
|
||||
parser.add_argument("--start-date", default="20180102", help="开始日期,格式 YYYYMMDD")
|
||||
parser.add_argument("--end-date", default="20260430", help="结束日期,格式 YYYYMMDD")
|
||||
args = parser.parse_args()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user