-
Notifications
You must be signed in to change notification settings - Fork 1
/
CateringItems.py
36 lines (33 loc) · 1.21 KB
/
CateringItems.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
33
34
35
36
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on Mon Jul 22 02:32:42 2019
@author: L.A.B
"""
class CateringItems:
def __init__(self, name):
self.name = name
self.items = {self.name : dict()}
def add(self, item, value = 1):
if issubclass(type(item), CateringProduct) or issubclass(type(item), CateringWorker):
self.items[self.name][item.name] = item
elif issubclass(type(item), CateringItems):
self.items[self.name][item.name] = item.items[item.name]
else:
self.items[self.name][item] = value
def rem(self, item):
if item in self.items[self.name].keys():
self.items[self.name].pop(item)
if isinstance(item, CateringItems):
if item in self.items[self.name].keys():
self.items[self.name].pop(item.name)
def get(self, name = '', full = 0):
if name:
try:
return self.items[self.name][name]
except NameError:
print("Objecto não existe em {n}".format(n = self.name))
if full:
return self.items
else:
return self.items[self.name]