2019-11-15 20:43:59 +01:00
|
|
|
import enum
|
2019-11-15 20:29:06 +01:00
|
|
|
|
2019-11-15 21:04:16 +01:00
|
|
|
from pydantic import BaseModel
|
2019-11-15 20:29:06 +01:00
|
|
|
|
2019-11-15 21:04:16 +01:00
|
|
|
|
|
|
|
|
class Vector(BaseModel):
|
|
|
|
|
x: int = 0
|
|
|
|
|
y: int = 0
|
2019-11-15 20:34:34 +01:00
|
|
|
|
|
|
|
|
|
2019-11-15 20:43:59 +01:00
|
|
|
class Direction(enum.Enum):
|
|
|
|
|
NORTH = "N"
|
|
|
|
|
SOUTH = "S"
|
|
|
|
|
EAST = "E"
|
|
|
|
|
WEST = "W"
|
|
|
|
|
|
|
|
|
|
|
2019-11-15 21:04:16 +01:00
|
|
|
class Rover(BaseModel):
|
|
|
|
|
pos: Vector = Vector(x=0, y=0)
|
|
|
|
|
planet_size: Vector = Vector(x=100, y=100)
|
2019-11-15 20:43:59 +01:00
|
|
|
dir: Direction = Direction.NORTH
|
