判断指定日期是否为交易日-python

import requests
import json
import requests.packages.urllib3.util.ssl_
import os
import time

requests.packages.urllib3.util.ssl_.DEFAULT_CIPHERS = 'ALL'  # 防止SSL限制

def isExist(timestr,save_file_path):
    with open(save_file_path, 'a+') as f:
    	f.seek(0)
    	lines = f.readlines()
    	if timestr in lines:
    		return 1
    	else:
    		return 0

def get_trading_date(_year_i,_month_i,save_file_path):
    """通过爬虫抓取深交所的交易日历
    :param month_date: 日期,例 2020-01、2022-12
    """
    month_date = "{}-{}".format(_year_i, _month_i + 1)
    target_url = "http://www.szse.cn/api/report/exchange/onepersistenthour/monthList?month={}".format(month_date)
    send_headers = {
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36",
        "Connection": "keep-alive",
        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
        "Accept-Language": "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7"}
    time.sleep(2)  # 限制频率
    req = requests.get(target_url, headers=send_headers)
    json_state = json.loads(req.text)
    for dict_value in json_state['data']:
        #保存结果:工作日非交易日
        date_str = dict_value['jyrq']
        if dict_value['jybz'] == "1":
            with open(save_file_path, 'a') as file:
                file.write(date_str)
                file.write("\n")
    return 0

def main():
    #timestr为需要判断的日期,默认为当天
    timestr = time.strftime('%Y-%m-%d',time.localtime(time.time()))+'\n'
    year = time.strftime('%Y',time.localtime(time.time()))
    save_file_path = year + "交易日.txt"
    #1.判断文件存在后,返回是否为交易日
    if(os.path.isfile(save_file_path)):
        return isExist(timestr,save_file_path)
    #2.更新文件,每年一次
    print("无本地文件,进行下载")
    for _month_i in range(12):
        get_trading_date(year, _month_i, save_file_path)
    #3.判断是否工作日非交易日
    return isExist(timestr,save_file_path)


if __name__ == '__main__':
    res = main()
    if(1 == res):
        print("当天是交易日")

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值