计算二个时间内有多少个交易日

import datetime


class workDays:
    def __init__(self, start_date, end_start, days_off=None):
        """
        weekday()方法将一个日期转换对应成星期,以星期一(0)到星期天(6)结束
        :param start_date: 开始时间,必须是datetime格式
        :param end_start: 结束时间,必须是datetime格式
        :param days_off: 休息日,默认为周六周日
        """
        self.start_date = start_date
        self.end_date = end_start
        self.days_off = days_off

        if self.start_date > self.end_date:  # 防止开始日期和结束日期传入反了
            self.start_date, self.end_date = self.end_date, self.start_date

        if self.days_off is None:  # 休息日为周六、周日
            self.days_off = 5, 6
        # 每周工作日列表
        self.days_work = [i for i in range(7) if i not in self.days_off]

    def workDays(self):  # 计算出自定义范围内有多少个工作日,包含节假日
        tag_date = self.start_date
        while True:
            if tag_date > self.end_date:
                break

            if tag_date.weekday() in self.days_work:  # 判断日期为非周末
                yield tag_date

            tag_date += datetime.timedelta(days=1)

    def holiday(self):  # 计算自定义范围内节假日占多少
        # A股休市日期
        holiday = "20210101,20210211,20210212,20210215,20210216,20210217,20210405,20210503,20210504,20210505,20210614,20210920,20210921,20211001,20211004,20211005,20211006,20211007"
        holidayList = holiday.split(",")
        for i in holidayList:
            date = datetime.datetime(int(i[0:4]), int(i[4:6]), int(i[6:8]))
            if self.start_date <= date <= self.end_date:  # 判断节假日在自定义的范围内
                if date.weekday() in self.days_work:  # 判断为非周末
                    yield date

    def daysCount(self):  # 计算工作日,不包含节假日
        return len(list(self.workDays()))-len(list(self.holiday()))

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值