修改limit_up数据来源

This commit is contained in:
2026-05-05 14:13:11 +08:00
parent 7b8a47ffe6
commit f8395f706e
4 changed files with 507 additions and 222 deletions
+19 -10
View File
@@ -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()