chess_rpg_backend/room/services/room_create.py

from asgiref.sync import sync_to_async
from random import randint

from common.generators import generate_charset
from game.models import Player, Deck
from room.models import Room, PlayerInRoom, GameState, HeroInGame


def sync_create_room(
    deck_id_1: int,
    player_id_1: int,
    player_score_1: int,
    deck_id_2: int,
    player_id_2: int,
    player_score_2: int,
):

    room = Room.objects.create(slug=generate_charset(16))
    player_1 = Player.objects.get(id=player_id_1)
    player_2 = Player.objects.get(id=player_id_2)

    first_player = randint(1, 2)

    p1 = PlayerInRoom.objects.create(
        player=player_1,
        room=room,
        score=player_score_1,
        deck_id=deck_id_1,
        first=first_player == 1,
    )

    p2 = PlayerInRoom.objects.create(
        player=player_2,
        room=room,
        score=player_score_2,
        deck_id=deck_id_2,
        first=first_player == 2,
    )
    for p, d_id in [(p1, deck_id_1), (p2, deck_id_2)]:
        GameState.objects.create(
            room=room, player=p.player, round=0, message="Game started"
        )
        for hero_in_deck in Deck.objects.get(id=d_id).heroes():
            if p.first:
                HeroInGame.objects.create(
                    hero=hero_in_deck.hero,
                    player=p,
                    room=room,
                    x=hero_in_deck.x,
                    y=hero_in_deck.y,
                )
            else:
                HeroInGame.objects.create(
                    hero=hero_in_deck.hero,
                    player=p,
                    room=room,
                    x=hero_in_deck.x,
                    y=8 - hero_in_deck.y,
                )
    return room.slug


@sync_to_async
def create_room(**kwargs):
    return sync_create_room(**kwargs)