chore(): Expands the definitions of the entities

- Adds create methods to the room, person and cards
- Adds methods to add and remove person from room
This commit is contained in:
Elia el Lazkani 2024-11-23 18:31:53 +01:00
parent 7955c6cd46
commit 307c3ff414
3 changed files with 61 additions and 0 deletions

View file

@ -2,3 +2,24 @@ 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

View file

@ -1,3 +1,6 @@
import uuid
class Person:
"""
Person entity
@ -8,3 +11,11 @@ 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

View file

@ -1,3 +1,6 @@
import uuid
class Room:
"""
Room entity
@ -8,3 +11,29 @@ 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)