hnc-eduard/HNS/Excercises/13082023 ДЗ по фигурам JSON/enums.py

from enum import Enum


class ShapeType(Enum):
    Rectangle = 'rectangle'
    Circle = 'circle'
    Square = 'square'
    Unknown = 'unknown'

    @staticmethod
    def from_string(raw_value):
        if raw_value:
            value = raw_value.lower().capitalize()
            if value in ShapeType.__members__:
                return ShapeType[value]
        return ShapeType.Unknown


class ShapeColor(Enum):
    Red = 'red'
    Yellow = 'yellow'
    Blue = 'blue'
    Green = 'green'
    Black = 'black'
    Unknown = 'unknown'

    @staticmethod
    def from_string(raw_value):
        if raw_value:
            value = raw_value.lower().capitalize()
            if value in ShapeColor.__members__:
                return ShapeColor[value]
        return ShapeColor.Unknown