python休假管理_Python计算休假日从一个轮班工施

import datetime

sched1 = {'Mon':'working','Tues':'working','Wed':'off','Thur':'working','Fri':'working','Sat':'working','Sun':'off'}

sched2 = {'Mon':'off','Tues':'working','Wed':'working','Thur':'working','Fri':'working','Sat':'off','Sun':'working'}

pickdate1 = int(input("Enter a date in the year (YYYY)): "))

pickdate2 = int(input("Enter a date in the year (MM): "))

pickdate3 = int(input("Enter a date in the year (DD): "))

date = datetime.date(pickdate1,pickdate2,pickdate3)

weekno = datetime.date(pickdate1,pickdate2,pickdate3).isocalendar()[1]

weekday = datetime.date.isoweekday(date)

if weekno % 2 == 0:

if weekday == 1:

print ("You are: %s" % (sched2['Mon']))

elif weekday == 2:

print ("You are: %s" % (sched2['Tues']))

elif weekday == 3:

print ("You are: %s" % (sched2['Wed']))

elif weekday == 4:

print ("You are: %s" % (sched2['Thur']))

elif weekday == 5:

print ("You are: %s" % (sched2['Fri']))

elif weekday == 6:

print ("You are: %s" % (sched2['Sat']))

elif weekday == 7:

print ("You are: %s" % (sched2['Sun']))

elif weekno % 2 != 0:

if weekday == 1:

print ("You are: %s" % (sched1['Mon']))

elif weekday == 2:

print ("You are: %s" % (sched1['Tues']))

elif weekday == 3:

print ("You are: %s" % (sched1['Wed']))

elif weekday == 4:

print ("You are: %s" % (sched1['Thur']))

elif weekday == 5:

print ("You are: %s" % (sched1['Fri']))

elif weekday == 6:

print ("You are: %s" % (sched1['Sat']))

elif weekday == 7:

print ("You are: %s" % (sched1['Sun']))