rewriting aqa 2015 code

rewriting_aqa_code.py

class pieces(object):
    piece_dict = {}
    def piece(func):
        def predicate():
            piece_dict[func.__name__] = func

        return predicate
    return func


class vec2d(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __add__(self, vec):
        try:
            return vec2d(
                self.x + vec.x,
                self.y + vec.y)
        except:
            raise ValueError("cannot add non type vec2d to vec2d")

    def __sub__(self, vec):
        try:
            return vec2d(
                self.x - vec.x,
                self.y - vec.y)
        except:
            raise ValueError("cannot add non type vec2d to vec2d")

    def __eq__(self, vec):
        return self.x == vec.x and self.y == vec.y

    def __str__(self):
        return "x: {0.x}, y: {0.y}".format(self)

class piece(object):
    def __init__(self, x, y, name, owner):
        self.location = vec2d(x,y)
        self.name = name
        self.owner = owner

    def get_loc(self):
        return self.location

    def get_loc_xy(self):
        return self.location.x, self.location.y

    def test_location(self, location):
        if self.location == location:
            return self # return self if at location

    def check_self_locations(self, *location):
        return not all(map(lambda x: self.location == x, locations))

class somepiece(piece):
    def __init__(self, x, y, owner):
        super().__init__(x, y, "somepiece", owner)

    def check_move(self, *otherPieces):
        # +1y -1y +1x -1x if any of own class