16 lines
374 B
Python
16 lines
374 B
Python
from enum import Enum
|
|
|
|
|
|
class ShipDirection(Enum):
|
|
VERTICAL = 'VERTICAL'
|
|
HORIZONTAL = 'HORIZONTAL'
|
|
UNKNOWN = 'UNKNOWN'
|
|
|
|
@staticmethod
|
|
def from_string(raw_value):
|
|
if raw_value:
|
|
value = raw_value.upper()
|
|
if value in ShipDirection.__members__:
|
|
return ShipDirection[value]
|
|
return ShipDirection.UNKNOWN
|