diff --git a/src/entities/cards.py b/src/entities/cards.py index abf164a..2b3be25 100644 --- a/src/entities/cards.py +++ b/src/entities/cards.py @@ -2,3 +2,23 @@ CARDS = { "fibonacci": [0.5, 1, 2, 3, 5, 8, 13, 21, 34, 55, "?"], "tshirt": ["XS", "S", "M", "L", "XL", "XXL", "XXXL", "?"], } + +class Cards: + """ + Cards class object + """ + def __init__(self, card_type = "fibonacci"): + self.card_type = card_type + self.cards = [] + self.create() + + + def create(self): + """ + Method to create a deck of cards + """ + if self.card_type in CARDS.keys(): + self.cards = CARDS[self.card_type] + return self + + return None diff --git a/src/entities/person.py b/src/entities/person.py index bbd4e20..03fb22e 100644 --- a/src/entities/person.py +++ b/src/entities/person.py @@ -1,3 +1,5 @@ +import uuid + class Person: """ Person entity @@ -8,3 +10,12 @@ class Person: Person entity initializer """ self.name = name + self.uuid = None + self.create() + + + def create(self): + if not self.uuid: + self.uuid = uuid.uuid1() + + return self diff --git a/src/entities/room.py b/src/entities/room.py index e694bbc..012588e 100644 --- a/src/entities/room.py +++ b/src/entities/room.py @@ -1,3 +1,5 @@ +import uuid + class Room: """ Room entity @@ -8,3 +10,31 @@ class Room: Room entity initializer """ self.name = name + self.uuid = None + self.attendant = [] + self.create() + + + def create(self): + """ + Method to create a new room object + """ + if not self.uuid: + self.uuid = uuid.uuid1() + + return self + + + def add_person(self, person: uuid.UUID): + """ + Method to add a person to a room + """ + if person not in self.attendant: + self.attendant.append(person) + + def remove_person(self, person: uuid.UUID): + """ + Method to remove a person from a room + """ + if person in self.attendant: + self.attendant.remove(person)