chess_rpg_backend/game/services/deck_handler.py

import random

from game.models import Deck, Player, HeroTypes, Hero, HeroInDeck


def create_first_deck(player: Player):
    deck = Deck.objects.create(player=player)
    positions = []

    for x in range(8):
        for y in range(2):
            positions.append((x, y))
    positions.remove((3, 0))
    positions.remove((4, 0))
    random.shuffle(positions)

    types = (
        ["KING", "WIZARD"]
        + ["ARCHER" for _ in range(4)]
        + ["WARRIOR" for _ in range(6)]
    )
    for _ in range(4):
        t = random.choice(HeroTypes.choices[:3])[0]
        if t == "WIZARD" and types.count("WIZARD") > 1:
            t = random.choice(HeroTypes.choices[:2])[0]
        types.append(t)

    counter = 0
    for t in types:
        hero = Hero()
        hero.player = player
        hero.type = t

        # set random position on deck for heroes
        if t == "KING":
            pos_x = 4
            pos_y = 0
        elif t == "WIZARD":
            pos_x = 3
            pos_y = 0
        else:
            pos_x = positions[counter][0]
            pos_y = positions[counter][1]

            counter += 1

        hero.health = random.randint(0, 10)
        hero.attack = random.randint(0, 10)
        hero.speed = random.randint(0, 10)

        hero.save()
        HeroInDeck.objects.create(deck=deck, hero=hero, x=pos_x + 1, y=pos_y + 1)