39 lines
804 B
Python
39 lines
804 B
Python
|
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
|