不灭的焱

革命尚未成功,同志仍须努力

作者:php-note.com  发布于:2020-10-24 17:33  分类:Python/数据分析  编辑

节假日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