ls-scheduler/Scheduler.py at master · zfine87/ls-scheduler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import csv, time, datetime

#Class for HLTV player
class Shift:

  #Constructor
  def __init__(self, name, start, end, day, rank):
    self.name = name
    self.start = start
    self.end = end
    self.day = day
    self.rank = rank

class Intern:

  def __init__(self, name, rank):
    self.name = name
    self.rank = rank

def getRankings():
  rankList = []

  with open('rankings.csv', newline='') as csvfile:

    spamreader = csv.reader(csvfile, delimiter=',', quotechar='|')
    next(spamreader, None)
    for each in spamreader:
      rankList.append(Intern(each[0], each[1]))

  return rankList

def getSchedule(rankList):
  shiftList = []

  with open('schedules.csv', newline='') as csvfile:
    spamreader = csv.reader(csvfile, delimiter=',', quotechar='|')
    next(spamreader, None)
    for each in spamreader:
      shiftList.append(Shift(each[1], datetime.datetime.strptime(str(each[0]) + " " +str(each[2]), "%m/%d/%Y %I:%M %p" ), datetime.datetime.strptime(str(each[0]) + " " +str(each[3]), "%m/%d/%Y %I:%M %p" ), datetime.datetime.strptime(each[0], "%m/%d/%Y"), next((x.rank for x in rankList if x.name == each[1]), None) ) )

  return shiftList


def createSchedule(scheduleList):

  dayList = sorted(list(set([each.day for each in scheduleList])))

  for day in dayList:
    print(str(day))
    oldWorkStrList = []
    workList = []
    change = False

    newScheduleList = [shift for shift in scheduleList if shift.day == day]

    newScheduleList.sort(key=lambda x: x.start, reverse=False)
    time = day
    while(time < day + datetime.timedelta(days=1)):
      #Check for and remove ending shifts
      leavingList = [shift.name for shift in workList if shift.end == time]
      newWorkList = [shift for shift in workList if shift.end != time]
      if(len(newWorkList) != len(workList)):
        change = True

      workList = newWorkList

      #Check for and add new shifts
      for shift in newScheduleList:
        if shift.start == time:
          addShift(workList, shift)
          change = True

      if(change):
        oldWorkStrList = printWorkList(workList, time, oldWorkStrList, leavingList)


      change = 0
      time += datetime.timedelta(minutes=15)

    input()

#Add new shift into list, then reorder based on ranking
def addShift(workList, shift):
  workList.append(shift)
  workList.sort(key=lambda x: x.rank, reverse=False)


def printWorkList(workList, time, oldWorkStrList, leavingList):
  workStrList = []
  print()
  print(time.strftime("At %I:%M %p"))
  size = len(workList)
  for i in range(0, int(size/3)):
    if(workList[i].name + " in Group 2" not in oldWorkStrList):
      workStrList.append(workList[i].name + " in Group 2")

  for i in range(int(size/3), int((size/3) * 2)):
    if(workList[i].name + " in Group 3" not in oldWorkStrList):
      workStrList.append(workList[i].name + " in Group 3")

  for i in range(int((size/3)*2), int((size/3) * 3)):
    if(workList[i].name + " in Group 1" not in oldWorkStrList):
      workStrList.append(workList[i].name + " in Group 1")

  for each in leavingList:
    print(each + " shift ends")

  for each in workStrList:
    print(each)

  return list(set(workStrList + oldWorkStrList))


# getSchedule()
createSchedule(getSchedule(getRankings()))