Python：通过API获得2019年节假日并自动排班
前言：翻到了一位大神做的节假日api，https://blog.csdn.net/xinit1/article/details/72833988 作者：xinit1
获取指定日期的节假日信息
1、接口地址：http://api.goseek.cn/Tools/holiday?date=数字日期
2、返回数据：正常工作日对应结果为 0, 法定节假日对应结果为 1, 节假日调休补班对应的结果为 2，休息日对应结果为 3
3、节假日数据说明：本接口包含2017年起的中国法定节假日数据，数据来源国务院发布的公告，每年更新1次，确保数据最新
4、示例：
http://api.goseek.cn/Tools/holiday?date=20170528
返回数据：
{“code”:10000,“data”:1}
正好自己需要做简单的周末和节假日排班表，之前都是EXCEL下载人家做好的日历表，再手动找出哪些是节假日，最后再按照一定的规则分配加班人，现在可以用Python自动化完成了。
具体操作如下：
1、新建一个Excel文件，输入’2019/1/1’格式，然后下拉365天，接着用=TEXT(C2,“yyyymmdd”)来获得标准日期格式（如“20190101”）
2、调用API接口，获得每一天的返回数据
def get_holiday():
'''根据日期判断日期类型：正常工作日对应结果为 0, 法定节假日对应结果为 1, 节假日调休补班对应的结果为 2，休息日对应结果为 3'''
url = r'http://api.goseek.cn/Tools/holiday?date='
try:
for i in range(2, 367):
data = ws.Range('A' + str(i)).Value
new_url = url + data
r = requests.get(new_url)
time.sleep(1)
r.encoding = 'utf-8'
d = json.loads(r.text)
answer = d['data']
ws.Range('B' + str(i)).Value = answer
print('{}:{}'.format(data, answer))
except Exception as e:
print(e)
finally:
wb.Close(True)
xl.quit()
3、我定义的加班规则如下，优先级从大到小：
1）、2019年1月1号由A来做，手动输入了；
2）、如果每个月的1号是休息日，就由A来做；
3）、如果前天是休息日且由某人来做，那今天就不要让他做了，从剩下的人中选择加班天数较少的人来做；
4）、如果昨天是休息日且由某人来做，那今天就仍然由他做；
5）、以上三种情况都不满足的话，就选加班天数较少的人来做；
4、完整代码如下
import requests
import json
from win32com.client import Dispatch
from os import path
import time
def get_holiday():
'''根据日期判断日期类型：正常工作日对应结果为 0, 法定节假日对应结果为 1, 节假日调休补班对应的结果为 2，休息日对应结果为 3'''
url = r'http://api.goseek.cn/Tools/holiday?date='
try:
for i in range(2, 367):
data = ws.Range('A' + str(i)).Value
new_url = url + data
r = requests.get(new_url)
time.sleep(1)
r.encoding = 'utf-8'
d = json.loads(r.text)
answer = d['data']
ws.Range('B' + str(i)).Value = answer
print('{}:{}'.format(data, answer))
except Exception as e:
print(e)
finally:
wb.Close(True)
xl.quit()
def get_min(a, b, c):
'''获得a, b, c三者里的最小值'''
if (b<= a) and (b<= c):
return 'B'
elif (c<= a) and (c<= b):
return 'C'
else:
return 'A'
def get_abc(this_person, a, b, c):
'''根据获得的加班人姓名，进行赋值动作并输出新的a,b,c'''
ws.Range('E' + str(i)).Value = this_person
print('{}是休息日，由{}来加班'.format(date, this_person))
if this_person == 'A': a += 1
elif this_person == 'B': b += 1
elif this_person == 'C': c += 1
return a, b, c
xl = Dispatch("Excel.Application")
xl.Visible = False #True是显示， False是隐藏
xl.DisplayAlerts = 0
wb_path = path.abspath('.')+"\\" + r'2019年日历表.xlsx'
wb = xl.Workbooks.Open(wb_path)
ws = wb.Sheets('Sheet1')
a = 0 #代表加班人员A的目前加班天数，下同
b = 0
c = 0
person_dict = {'A': a, 'B': b, 'C': c}
for i in range(3, 367):
value = int(ws.Range('B' + str(i)).Value)
date = ws.Range('A' + str(i)).Value
day = int(date[-2:])
if (day == 1): if_first_day = 1 #判断是否为每个月的第一天
else: if_first_day = 0
if (value == 0) or (value == 2): #如果今天是工作日就pass
print('{}是工作日'.format(date))
continue
else:
if if_first_day == 1: #如果是每个月第一天或第二天就A来做
this_person = 'A'
ws.Range('E' + str(i)).Value = this_person
print('{}是休息日，由{}来加班'.format(date, this_person))
a += 1
elif int(ws.Range('B' + str(i-2)).Value) in (1, 3): #如果前天也是节假日或休息日，那今天不要让那个人加班了
qiantian_person = ws.Range('E' + str(i-2)).Value
person_list = ['A', 'B', 'C']
person_list.remove(qiantian_person)
if person_dict[person_list[0]] >= person_dict[person_list[1]]:
this_person = person_list[1]
else:
this_person = person_list[0]
a, b, c = get_abc(this_person, a, b, c)
elif int(ws.Range('B' + str(i-1)).Value) in (1, 3): #如果昨天也是节假日和休息日，就由昨天加班的人来做
pre_person = ws.Range('E' + str(i-1)).Value
this_person = pre_person
a, b, c = get_abc(this_person, a, b, c)
else:
this_person = get_min(a, b, c)
a, b, c = get_abc(this_person, a, b, c)
wb.Close(True)
xl.quit()
最后成果：
