Madra_smieciarka/classes/Household.py

from classes.Garbage import *


class Household:
    def __init__(self):
        self.id: int = 0
        self.image: object = None
        self.position: tuple = None
        self.garbage: object = Garbage()

    def setPosition(self, position: tuple):
        self.position = position
        return self

    def getPosition(self):
        return self.position

    def setImage(self, image: object):
        self.image = image
        return self

    def getImage(self):
        return self.image

    def setId(self, newID: int):
        self.id = newID
        return self

    def getId(self):
        return self.id

    def setGarbage(self, garbage: object):
        self.garbage = garbage
        return self

    def getGarbage(self):
        return self.garbage


def generateHousehold(i, image, position):
    """
    :param int i:
    :param object image:
    :param tuple position:
    :return:
    """
    house = Household()
    house.setId(i).setImage(image).setPosition(position)
    return house