Игра «Инопланетное вторжение»

from random import randint
doska = [] #создаем список
zag = ['|X| ','|1|','|2|','|3|']
#нумерация поля
 
#Далее создаем доску и разметку
for x in range(3): #заполняем доску
    doska.append(['|O|'] * 3)
    
def print_doska(doska): #функция, печатающая доску
    print(' '.join(zag))
    
    i=1
    for row in doska:
        if i < 10:
            print(zag[i], '', ' '.join(row)) #разделитель между символами
        else:
            print(zag[i], ' '.join(row))
        i += 1 
      
print_doska(doska) 
 
 
 
#здесь храним пришельцев
 
alien_0 = {'Цвет': 'крансый', 'очки': 1, 'сила': 'медленный', 'Имя': 'Тоха'}
alien_1 = {'Цвет': 'оранжевый', 'очки': 2, 'сила': 'быстрый', 'Имя': 'Дюха'}
alien_2 = {'Цвет': 'желтый', 'очки': 3, 'сила': 'умный', 'Имя': 'Пашок'}
alien_3 = {'Цвет': 'желтый', 'очки': 4, 'сила': 'умный', 'Имя': 'Васян'}
alien_4 = {'Цвет': 'зеленый', 'очки': 5, 'сила': 'тихий', 'Имя': 'Гошан'}
 
aliens = []
aliens.append(alien_0)
aliens.append(alien_1)
aliens.append(alien_2)
aliens.append(alien_3)
aliens.append(alien_4)

import random
 
 
class AlienInvasion:
 
    def __init__(self, weight: int, height: int):
        self.weight, self.height = weight, height
        self.field = [[0] * weight for _ in range(height)]
        self.aliens = []
 
    def show(self):
        print(*self.field, sep="\n")
 
    def append_alien(self, name: str, color: str, price: int, power: str):
        self.aliens.append({"name": name, "color": color, "price": price, "power": power})
 
    def remove_alien(self, index: int):
        self.aliens.pop(index - 1)
 
    def arrange_aliens(self):
        cells = [(i, j) for j in range(self.weight) for i in range(self.height)]
        for i, alien in enumerate(self.aliens):
            row, column = cells.pop(random.randint(0, len(cells)) - 1)
            self.field[row][column] = i + 1
 
    def get_alien(self, index: int):
        return self.aliens[index - 1]
 
    def select_cell(self, row: int, column: int):
        index = self.field[row][column]
        if index != 0:
            return self.get_alien(index)
        return None
 
 
game = AlienInvasion(5, 5)
game.append_alien("Тоха", "Красный", 1, "Медленный")
game.append_alien("Дюха", "Оранжевый", 2, "Быстрый")
game.append_alien("Пашок", "Желтый", 3, "Умный")
game.arrange_aliens()
game.show()
# Output:
#   [0, 0, 0, 0, 0]
#   [1, 0, 0, 0, 0]
#   [0, 0, 2, 0, 0]
#   [0, 0, 0, 0, 0]
#   [0, 0, 0, 3, 0]
print(game.select_cell(0, 0))
# Output:
#   None
print(game.select_cell(3, 3))
# Output:
#   {"name": "Дюха", "color": "Оранжевый", "price": 2, "power": "Быстрый"}