16 lines
389 B
Python
16 lines
389 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
|