22 lines
338 B
Python
22 lines
338 B
Python
import enum
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class Vector(BaseModel):
|
|
x: int = 0
|
|
y: int = 0
|
|
|
|
|
|
class Direction(enum.Enum):
|
|
NORTH = "N"
|
|
SOUTH = "S"
|
|
EAST = "E"
|
|
WEST = "W"
|
|
|
|
|
|
class Rover(BaseModel):
|
|
pos: Vector = Vector(x=0, y=0)
|
|
planet_size: Vector = Vector(x=100, y=100)
|
|
dir: Direction = Direction.NORTH
|