def withdrawResource(self, resource): """Removes a resource from the square and returns it.""" if resource.getResource() == 0: return elif self._resources[resource.getType()] <= resource.getResource(): self._resources[resource.getType()] == 0 else: self._resources[resource.getType()] -= resource.getResource()
def addResource(self, resource): """Adds a resource to this square.""" if resource.getResource() == 0: return else: self._resources[resource.getType()] += resource.getResource()