34 lines
780 B
Python
34 lines
780 B
Python
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
|