不灭的焱

节假日api调用来自于：http://tool.bitefu.net/jiari/

交易日：为 非节假日 且 非周六周日

代码如下：

# stock_util.py

import urllib.request as request
import datetime


def stock_get_date_type(query_date):
    """
    获取日期类型
    :param query_date: 日期，如 2020-10-01
    :return 0:工作日，1:周末，2:假日，-1:出错
    """
    url = 'http://tool.bitefu.net/jiari/?d=' + query_date
    resp = request.urlopen(url, timeout=3)
    content = resp.read()
    if content:
        try:
            day_type = int(content)
        except ValueError:
            return -1
        else:
            return day_type

    return -1


def stock_is_trade_date(query_date):
    """
    是否为 交易日
    :param query_date: 日期，如 2020-10-01
    :return: 1:是，0:不是
    """
    weekday = datetime.datetime.strptime(query_date, '%Y-%m-%d').isoweekday()
    if weekday <= 5 and stock_get_date_type(query_date) == 0:
        return 1
    else:
        return 0