hnc-daniil/HNC/Exercises/Ship_Battle/ShipDirection.py

from enum import Enum


class ShipDirection(Enum):
    VERTICAL = "VERTICAL"
    HORIZONTAL = "HORIZONTAL"
    UNKNOWN = "UNKNOWN"

    @staticmethod
    def from_string(raw_value):
        if raw_value:
            value = raw_value.upper()
            if value in ShipDirection.__members__:
                return ShipDirection[value]
        return ShipDirection.UNKNOWN