# 导入PyWenCai模块
import pandas as pd
import pywencai
import json 
from datetime import datetime, timedelta

def 获取涨停板数据(start_date_str,end_date_str):
    # 获取今天的日期
    start_date = datetime.strptime(start_date_str, '%Y%m%d')
    end_date = datetime.strptime(end_date_str, '%Y%m%d')
    current_date = start_date
    while current_date <= end_date:
        if current_date.weekday() < 5:  # 跳过周六和周日 (weekday() 返回 0-6, 0 是周一, 6 是周日)
            交易日 = current_date.strftime('%Y%m%d')
            print(f'正在获取 {交易日} 的涨停板数据...')
            try:
                res = pywencai.get(query=f'涨停 {交易日}',loop=True,sort_key='股票代码',sort_order='asc', log=True);
                if any(交易日 in col for col in res.columns):
                    res.to_csv(f'x:\\zt{交易日}.csv')
                    print(f'成功获取 {交易日} 的涨停板数据并已保存为 x:\\zt{交易日}.csv')
            except Exception as e:
                print(f"发生错误: {e}")
        current_date += timedelta(days=1)  # 增加一天

#获取涨停板数据('20241008','20241031')
#获取涨停板数据('20240902','20240930')
#获取涨停板数据('20240801','20240830')
#获取涨停板数据('20240701','20240731')
#获取涨停板数据('20240601','20240630')
#获取涨停板数据('20240501','20240531')
#获取涨停板数据('20240401','20240430')
#获取涨停板数据('20240301','20240331')
#获取涨停板数据('20240201','20240229')
获取涨停板数据('20240102','20240131')
print("\r\n\r\n")