from enum import Enum class ShapeColor(Enum): Unknown = 'unknown' Red = 'red' Green = 'green' Blue = 'blue' Yellow = 'yellow' White = 'white' Black = 'black' @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 class ShapeType(Enum): Unknown = 'unknown' Rectangle = 'rectangle' Circle = 'circle' Square = 'square' @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