def rooms_occupied_by(self, people): s = Storage(self.name, self.token, True) room_numbers, rooms, person_names, persons, current_timestamp = s.get_stats() log = s.get_log() people_wanted = { (t[1]+name):True for t, name in people } rooms = {} result = {} for event in log: if not event.who in people_wanted or event.room == -1: continue if event.action == 'A': if event.room in rooms: people_preset = rooms[event.room] else: people_present = {} rooms[event.room] = people_present people_present[event.who] = True if people_present == people_wanted: result[event.room] = True elif event.action == 'L': if event.who in rooms[event.room]: rooms[event.room].pop(event.who) if len(result) == 0: return result = [e for e in result.keys()] result.sort() result2 = [str(e) for e in result] print(','.join(result2))
def rooms_entered_by(self, who, is_employee): who = 'E' + who if is_employee else 'G' + who s = Storage(self.name, self.token, True) room_numbers, rooms, person_names, persons, _ = s.get_stats() log = s.get_log() print(','.join([str(e.room) for e in log if e.who == who and e.action == 'A' and e.room != -1 ]))