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