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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
| class Task:
id_counter = 0
def __init__(self, description: str = "", programmer: str = "", workload: int = 0) -> None:
self.description = description
self.programmer = programmer
self.workload = workload
self.finished = False
Task.id_counter += 1
self.id = Task.id_counter
def is_finished(self):
return self.finished
def mark_finished(self):
self.finished = True
def __str__(self):
status = "FINISHED" if self.finished else "NOT FINISHED"
return f"{self.id}: {self.description} ({self.workload} hours), programmer {self.programmer} {status}"
class OrderBook:
def __init__(self) -> None:
self._all_orders = []
def add_order(self, description, programmer, workload):
task_obj = Task(description, programmer, workload)
self._all_orders.append(task_obj)
def all_orders(self):
return self._all_orders
def programmers(self):
return list(set(order.programmer for order in self._all_orders))
def mark_finished(self, id: int):
for order in self._all_orders:
if order.id == id:
order.mark_finished()
return
raise ValueError("NO ID MATCHED FOR TASK")
def finished_orders(self):
return [order for order in self._all_orders if order.is_finished()]
def unfinished_orders(self):
return [order for order in self._all_orders if not order.is_finished()]
def status_of_programmer(self, programmer: str) -> tuple:
if any(order.programmer == programmer for order in self._all_orders):
task_done = sum([1 for order in self._all_orders if order.programmer == programmer and order.is_finished() == True])
task_undone = sum([1 for order in self._all_orders if order.programmer == programmer and order.is_finished() == False])
num_done_hours = sum([order.workload for order in self._all_orders if order.programmer == programmer and order.is_finished()])
num_undone_hours = sum([order.workload for order in self._all_orders if order.programmer == programmer and not order.is_finished()])
return task_done, task_undone, num_done_hours, num_undone_hours
raise ValueError("STATUS OF PROGRAMMER ERROR")
class OrderBookApplication:
def __init__(self) -> None:
self.__orders = OrderBook()
def help(self):
print("commands: ")
print("1 add order")
print("2 list finished tasks")
print("3 list unfinished tasks")
print("4 mark task as finished")
print("5 programmers")
print("6 status of programmer")
print("0 exit")
def add_order(self):
description = input("description: ")
prog_and_estimate = input("programmer and workload estimate: ")
parts = prog_and_estimate.split(" ")
try:
programmer_name = parts[0]
workload = int(parts[1])
except:
print("erroneous input")
return
self.__orders.add_order(description, programmer_name, workload)
print("added!")
def list_finished_tasks(self):
if len(self.__orders.finished_orders()) == 0:
print("no finished tasks")
for item in self.__orders.finished_orders():
print(item)
def list_unfinished_tasks(self):
this_list = self.__orders.unfinished_orders()
for item in this_list:
print(item)
def mark_task_as_finish(self):
try:
val = int(input("id: "))
if val <= Task.id_counter:
self.__orders.mark_finished(val)
print("marked as finished")
else:
print("erroneous input")
except:
print("erroneous input")
return
def programmers(self):
for programmer in self.__orders.programmers():
print(programmer)
def status_of_programmer(self):
name = input("name: ")
if not name in self.__orders.programmers():
print("erroneous input")
return
parts = self.__orders.status_of_programmer(name)
print(f"tasks: finished {parts[0]} not finished {parts[1]}, hours: done {parts[2]} scheduled {parts[3]}")
def execute(self):
self.help()
while True:
print("")
command = input("command: ")
if command == "0":
break
elif command == "1":
self.add_order()
elif command == "2":
self.list_finished_tasks()
elif command == "3":
self.list_unfinished_tasks()
elif command == "4":
self.mark_task_as_finish()
elif command == "5":
self.programmers()
elif command == "6":
self.status_of_programmer()
else:
self.help()
application = OrderBookApplication()
application.execute()
|