Python - Ayuda, con un programa para que me de el costo multiplicado por la cantidad del producto

class Producto:

    def __init__(self, codigo, nombre, costo, ):

        self.__codigo = codigo

        self.__nombre = nombre

        self.__costo = costo

    @property

    def codigo(self):

        return self.__codigo

    @codigo.setter

    def codigo(self, valor):

        self.__codigo = valor

    @property

    def nombre(self):

        return self.__nombre

    @nombre.setter

    def nombre(self, valor):

        self.__nombre = valor

    @property

    def costo(self):

        return self.__costo

    @costo.setter

    def costo(self, valor):

        self.__costo = valor

    def __str__(self):

        return 'codigo: ' + str(self.__codigo) + ' , nombre: ' + str(self.__nombre) + ', costo '+ str(self.__costo)

p1 = Producto(100, "Leche", 30, )

p2 = Producto(500, "Huevo", 30, )

p3 = Producto(40, "Azucar", 55, )

p4 = Producto(1, "Jamon", 135, )

print(p1)

print(p2)

print(p3)

print(p4)

def total(self):

        cantidad = int(input("Cantidad: "))

        return f"El costo total es: {cantidad * self.costo}"

p1 = Producto(100, "Leche", 30, )

p3 = Producto(40, "Azucar", 55, )

print(p1.total())

print(p3.total())