Как получить другой элемент Literal?

Question

[Иван Стыслов]

Есть такой класс:

from typing import Literal

class TicTacToePlayer:
    def __init__(self,
                 mark: Literal["cross", "zero"],
                 field: TicTacToeField
                 ):
        self.cells = []
        self.mark = mark
        self.field = field

у него есть поле mark, мне нужно как то отталкиваясь от того, какой mark у данного игрока, получить mark другого игрока
Как бы это по-человечески сделать?
Просто условно есть метод, который должен для игрока определить потенциально хорошие для хода поля, и анализировать нужно по полям другого игрока.
То есть получить что-то вроде, только с литералом

l = [mark1, mark2]
# при условии, что player.mark = mark2
other_player_mark = l[0 if player.mark == mark2 else 1]

Или может стоит это совсем иначе делать? Не используя литерал и подобие enum-ов

Answer 1

[Vindicar]

Литерал означает, что данная переменная/параметр всегда имеет одно из перечисленных значений. Так что...

other_player_mark = 'cross' if mark == 'zero' else 'zero'

Не, ты, конечно можешь извратиться...

import typing as t

Mark = t.Literal['cross', 'zero']

class TicTacToePlayer:
    def __init__(self,
                 mark: Mark,
                 field: TicTacToeField
                 ):
        self.cells = []
        self.mark = mark
        self.field = field
    
    def opposite(self, mark: Mark) -> Mark:
        values = t.get_args(Mark)
        idx = values.index(mark)
        next_idx = int(not idx)
        return t.cast(Mark, values[next_idx])

но лучше не надо!

Comments

[Иван Стыслов]

просто хотелось сделать универсальнее что ли, допустим в литерале будет 3 элемента или больше, получаются, что всевозможные значения для mark нигде не зафиксированы и если я поменяю "cross" на "Cross" в __init__ методе, то мой код с данной записью:

other_player_mark = 'cross' if mark == 'zero' else 'zero'

неминуемо столкнется с ошибкой, которую еще не факт, что получится быстро отследить

[Vindicar]

Иван Стыслов, если будет три элемента или больше, то в чём будет смысл "противоположного"?
Вообще никто не запрещает описать преобразование словарём, если нужна универсальность.
Только не надо забывать про принцип KISS.

[Иван Стыслов]

Vindicar, я нигде не упоминал о том, что ищу противоположный элемент, мой пример - это лишь предположение решения для конкретно случая, про принцип почитаю, спасибо