106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
|
from schemas.rooms import AnonymousIn_schema, RoomIn_schema
|
||
|
|
from schemas.users import User_schema
|
||
|
|
from database.auth.models import User
|
||
|
|
from database.rooms.models import AnonymousMember, Room, RoomOwner, Waiter
|
||
|
|
from services.database import generate_unique_code
|
||
|
|
|
||
|
|
|
||
|
|
async def create_room_with_user_db(room: RoomIn_schema, user: User_schema):
|
||
|
|
code = await generate_unique_code(Room)
|
||
|
|
|
||
|
|
room_obj = await Room.create(**room.dict(exclude_unset=True), id_code=code)
|
||
|
|
|
||
|
|
user = await User.get(id=user.id)
|
||
|
|
await room_obj.users.add(user)
|
||
|
|
await RoomOwner.create(room_id=room_obj.id, user_id=user.id)
|
||
|
|
return room_obj
|
||
|
|
|
||
|
|
|
||
|
|
async def create_room_anonymous_db(room: RoomIn_schema, anonymous: AnonymousIn_schema):
|
||
|
|
code = await generate_unique_code(Room)
|
||
|
|
|
||
|
|
room_obj = await Room.create(**room.dict(exclude_unset=True), id_code=code)
|
||
|
|
|
||
|
|
anonymous_code = await generate_unique_code(AnonymousMember)
|
||
|
|
anonymous = await AnonymousMember.create(**anonymous.dict(exclude_unset=True), id_code=anonymous_code, room_id=room_obj.id)
|
||
|
|
|
||
|
|
await RoomOwner.create(room_id=room_obj.id, anonymous_id=anonymous.id)
|
||
|
|
return room_obj
|
||
|
|
|
||
|
|
|
||
|
|
async def get_room_db(room_id: str):
|
||
|
|
room = await Room.get_or_none(id_code=room_id)
|
||
|
|
return room
|
||
|
|
|
||
|
|
|
||
|
|
async def check_user_in_room(room: Room, user: User):
|
||
|
|
return await room.users.filter(id=user.id).count() != 0
|
||
|
|
|
||
|
|
|
||
|
|
async def get_member_by_code(room: Room, code: str):
|
||
|
|
anonymous = await room.anonymousmembers
|
||
|
|
filtered_anonymous = [
|
||
|
|
m for m in anonymous if m.id_code == code]
|
||
|
|
if len(filtered_anonymous) == 0:
|
||
|
|
return None
|
||
|
|
return filtered_anonymous[0]
|
||
|
|
|
||
|
|
|
||
|
|
async def check_user_owner(room: Room, user: User):
|
||
|
|
room_owner = await room.room_owner
|
||
|
|
user_owner = await room_owner.user
|
||
|
|
if user_owner == None:
|
||
|
|
return False
|
||
|
|
return user_owner.id == user.id
|
||
|
|
|
||
|
|
|
||
|
|
async def check_anonymous_owner(room: Room, anonymous: AnonymousMember):
|
||
|
|
room_owner = await room.room_owner
|
||
|
|
anonymous_owner = await room_owner.anonymous
|
||
|
|
if anonymous_owner == None:
|
||
|
|
return False
|
||
|
|
return anonymous_owner.id_code == anonymous.id_code
|
||
|
|
|
||
|
|
|
||
|
|
async def create_waiter_by_user(room: Room, user: User):
|
||
|
|
code = await generate_unique_code(Waiter)
|
||
|
|
return await Waiter.create(room_id=room.id, user_id=user.id, name=user.username, id_code=code)
|
||
|
|
|
||
|
|
|
||
|
|
async def create_waiter_anonymous(room: Room, name: str):
|
||
|
|
code = await generate_unique_code(Waiter)
|
||
|
|
return await Waiter.create(name=name, id_code=code, room_id=room.id)
|
||
|
|
|
||
|
|
|
||
|
|
async def connect_room(room: Room, code):
|
||
|
|
online = room.online
|
||
|
|
await room.update_from_dict({'online': [*online, code]}).save(update_fields=['online'])
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
async def disconnect_room(room: Room, code):
|
||
|
|
online = room.online
|
||
|
|
await room.update_from_dict({'online': [o for o in online if o != code]}).save(update_fields=['online'])
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
async def validate_name_in_room(room: Room, name):
|
||
|
|
anonymous = await room.anonymousmembers
|
||
|
|
if len([a for a in anonymous if a == name]) != 0:
|
||
|
|
return "Pseudo déjà utilisé"
|
||
|
|
if len(name) < 3:
|
||
|
|
return "Pseudo trop court"
|
||
|
|
if len(name) > 30:
|
||
|
|
return "Pseudo trop long"
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
async def check_user_in_room(room_id: str , user: User):
|
||
|
|
room = await Room.get(id_code = room_id)
|
||
|
|
users = await room.users.filter(id=user.id)
|
||
|
|
return users.count() != 0
|
||
|
|
|
||
|
|
async def check_user_in_room(room_id: str , anonymous: AnonymousMember):
|
||
|
|
room = await Room.get(id_code = room_id)
|
||
|
|
members = await room.anonymousmembers.filter(id=anonymous.id)
|
||
|
|
return members.count() != 0
|