-
Notifications
You must be signed in to change notification settings - Fork 10
/
memory_db.py
32 lines (24 loc) · 817 Bytes
/
memory_db.py
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
class MemoryDB(object):
def __init__(self):
self.employees = {}
self.members = {}
self.id = 0
def next_id(self):
self.id += 1
return self.id
def get_employee(self, id):
return self.employees[id]
def add_employee(self, employee):
employee.id = self.next_id()
self.employees[employee.id] = employee
def remove_employee(self, empId):
self.employees.pop(empId)
def add_union_member(self, member):
member.memberId = self.next_id()
self.members[member.memberId] = member
def get_union_member(self, memberId):
return self.members[memberId]
def remove_union_member(self, memberId):
self.members.pop(memberId)
def get_all_employees(self):
return self.employees.values()