Using (self,*cars) instead of (self,cars) for multiple instances

Question

I tried using *cars and it returns the car object ID instead of the model year and make. Any ideas on how I would get the loop to use the def str method?

class Dealership: def init(self): self.cars = []

def iter(self): yield from self.cars

def add_car(self, *car): self.cars.append(car)

car_one = Car('Model T', 1908) car_two = Car('Phantom', 2020, 'Rolls Royce')

my_dealership = Dealership() my_dealership.add_car(car_one, car_two)

for car in my_dealership: print(car)

Answer 1 · 2023-01-23T20:15:48Z

January 23, 2023 8:15pm

You look like you understand Python class, nice job!

All you need is a little bit of Python magic "dunder" methods to get your job done:

__init__, __str__, __iter__

class Car:
    def __init__(self, make, year=0, brand=''):
        self.make = make
        self.year = year
        self.brand = brand

    def __str__(self):
        return f"{self.make} {self.year} {self.brand}"        

class Dealership:
    def __init__(self):
        self.cars = []

    def __iter__(self):
        yield from self.cars

    def add_car(self, *cars):
        for car in cars:
            self.cars.append(car)

car_one = Car('Model T', 1908 )
car_two = Car('Phantom', 2020, 'Rolls Royce')

my_dealership = Dealership()
my_dealership.add_car(car_one, car_two)

for car in my_dealership:
    print(car)

Welcome to the Treehouse Community

Looking to learn something new?

Sam Appleton

Sam Appleton

Using (self,*cars) instead of (self,cars) for multiple instances

1 Answer

Jeff Muday

Jeff Muday