Python3版本下创建计算给定日期范围内工作日方法

眼睛都要挣不开了，代码还有很多需要改进的地方，暂时先把方法放上，明天再整理
# _*_ coding:utf-8 _*_# Author: dai bing# Date: 2020/12/13 22:48# Scripts: calcworkday.py# Desc: 给定两个日期区间，计算区间内的工作日# 需要排除法定节假日以及统计休息日内调整为工作日内的天数# 给定人数计算出需要多少个任务满足报工需求#import datetimeimport timeclass Tasks:    weeks = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期天']    pattern = "%Y-%m-%d"    def __init__(self):        super(Tasks, self).__init__()        self._person = None        self._start_date = None        self._end_date = None        self._statutory_holidays = None        self._non_rest_days = None    @property    def person(self):        return self._person    @person.setter    def person(self, value):        self._person = value    @property    def startDate(self):        return self._start_date    @startDate.setter    def startDate(self, value):        sdate = None        try:            if isinstance(value, str):                sdate = datetime.datetime.strptime(value, self.pattern)            else:                raise Exception        except Exception as e:            print("输入值{}类型为{}，需要输入日期型字符串或日期型数组.".format(value, type(value)))            print(e)            exit(11)        self._start_date = sdate    @property    def endDate(self):        return self._end_date    @endDate.setter    def endDate(self, value):        edate = None        try:            if isinstance(value, str):                edate = datetime.datetime.strptime(value, self.pattern)            else:                raise Exception        except Exception as e:            print("输入值{}类型为{}，需要输入日期型字符串或日期型数组.".format(value, type(value)))            print(e)            exit(12)        self._end_date = edate    @property    def statutoryHolidays(self):        return self._statutory_holidays    @statutoryHolidays.setter    def statutoryHolidays(self, value):        change_statutory_holidays = []        try:            if isinstance(value, list):                for sholiday in value:                    # print(datetime.datetime.strptime(sholiday, pattern))                    change_statutory_holidays.append(datetime.datetime.strptime(sholiday, self.pattern))            elif isinstance(value, str):                change_statutory_holidays.append(datetime.datetime.strptime(value, self.pattern))        except ValueError:            print("输入值{}类型为{}，需要输入日期型字符串或日期型数组.".format(value, type(value)))            exit(13)        self._statutory_holidays = change_statutory_holidays    @property    def nonRestDay(self):        return self._non_rest_days    @nonRestDay.setter    def nonRestDay(self, value):        change_work_day = []        pattern = "%Y-%m-%d"        try:            if isinstance(value, list):                for non_work_day in value:                    change_work_day.append(datetime.datetime.strptime(non_work_day, pattern))            elif isinstance(value, str):                change_work_day.append(datetime.datetime.strptime(value, pattern))        except TypeError:            print("输入值{}类型为{}，需要输入日期型字符串或日期型数组.".format(value, type(value)))            exit(14)        self._non_rest_days = change_work_day    # 计算指定日期区间内的工作日    # statutory_h排除法定节假日    # work_do休息日调休变更为工作日    def getWorkDay(self, statutory_h, work_do):        day_off = 5, 6        # print(day_off)        days_work = [x for x in range(7) if x not in day_off]                # 排除法定节假日        statutory_holidays = statutory_h        # 增加调休工作日        non_rest_days = work_do        # print("statutory_holidays: ", len(statutory_holidays))        # print("work_day_off:", len(work_day_off))        tag_date = self.startDate        flag = 0        while True:            if tag_date > self.endDate:                break            if tag_date.weekday() in days_work and tag_date not in statutory_holidays or tag_date in non_rest_days:                yield tag_date            tag_date += datetime.timedelta(days=1)        # print(days_work)        # print("flag:", flag)    def getjobs(self):        jobs = 1        coefficient = 1        while True:            coefficient = 113 * jobs / int(self.daysCount()) / self.person            jobs += 1            # print("coefficient:{},jobs:{}".format(coefficient, jobs))            if coefficient > 4:                if coefficient < 5:                    print("[{}]{}月份共有{}位需要报工人员，需要{}个任务拆分进行报工。"                          .format(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),                                  datetime.datetime.now().month + 1, self.person, jobs))                    break    def daysCount(self):        """工作日统计,返回数字"""        return len(list(self.getWorkDay(self.statutoryHolidays, self.nonRestDay)))if __name__ == "__main__":    print(__name__)
©著作权归作者所有：来自51CTO博客作者good_joe的原创作品，如需转载，请注明出处，否则将追究法律责任
好知识，才能预见未来

赞赏
0人进行了赞赏支持
好知识，才能预见未来

更多相关文章

随机推荐
