mars-rover/src/rover/rover.py

import random
import warnings

HEADINGS = ['N', 'E', 'S', 'W']

class Location:

    def __init__(self, x, y, heading):
        self.x = x
        self.y = y
        self.heading = heading


class Rover:

    def __init__(self, x=None, y=None, heading=None):
        if all((x, y, heading)):
            self.location = Location(x, y, heading)
        else:
             warnings.warn("Landing at a random place")
             self.location = Location(random.randint(-10, 10),
                                      random.randint(-10, 10),
                                      HEADINGS[random.randint(0,
                                                 len(HEADINGS) - 1)])

    def move(self, command):
        for item in command:
            if item == 'F' and self.location.heading == 'E':
                self.location.x += 1

            if item == 'F' and self.location.heading == 'N':
                self.location.y += 1
            # ... TODO: add all directions and check constraints
            # if going over the borders calculate roll-over ...
            # e.g, if after a few moves X location is E is 11 it should be
            # reported and saved as W -1.

        # separate moving from reporting
        self._report()

    def _report(self):
        """naive implementation that prints to stdout, can be changed to
        send a radio message"""
        print(f"Location ({self.location.x}, {self.location.y}) Heading {self.location.heading}")