SUPyF2 Objects/Classes-Exericse - 08. Vehicle

"""
Objects and Classes - Exericse
Check your code: https://judge.softuni.bg/Contests/Practice/Index/1734#7

SUPyF2 Objects/Classes-Exericse - 08. Vehicle (not included in final score)

Problem:
Create a class Vehicle. The __init__ method should receive: type, model, price. You should also set the owner to None.
The class should have the following methods:
• buy(money, owner) - if the person has enough money and the vehicle has no owner, returns:
"Successfully bought a {type}. Change {change}" and sets the owner to the given one.
If the money is not enough, return: "Sorry, not enough money".
If the car already has an owner, return: "Car already sold"
• sell() - if the car has an owner, set it to None again. Otherwise, return: "Vehicle has no owner"
• __repr__() - returns "{model} {type} is owned by: {owner}" if the vehicle has an owner.
Otherwise, return: "{model} {type} is on sale: {price}"

Example:
Test Code:
vehicle_type = "car"
model = "BMW"
price = 30000
vehicle = Vehicle(vehicle_type, model, price)
vehicle.buy(15000, "Peter")
vehicle.buy(35000, "George")
print(vehicle)
vehicle.sell()
print(vehicle)

Output:
Sorry, not enough money
Successfully bought a car. Change: 5000.00
BMW car is owned by: George
BMW car is on sale: 30000
"""


class Vehicle:
    def __init__(self, type: str, model: str, price: float):
        self.type = type
        self.model = model
        self.price = price
        self.owner = None

    def buy(self, money, owner):
        if money >= self.price and self.owner is None:
            self.owner = owner
            return f"Successfully bought a {self.type}. Change: {(money - self.price):.2f}"
        elif money < self.price:
            return f"Sorry, not enough money"
        elif self.owner:
            return f"Car already sold"

    def sell(self):
        if not self.owner:
            return "Vehicle has no owner"
        self.owner = None

    def __repr__(self):
        if self.owner:
            return f"{self.model} {self.type} is owned by: {self.owner}"
        return f"{self.model} {self.type} is on sale: {self.price}"


vehicle_type = "car"
model = "BMW"
price = 30000
vehicle = Vehicle(vehicle_type, model, price)
vehicle.buy(15000, "Peter")
vehicle.buy(35000, "George")
print(vehicle)
vehicle.sell()
print(vehicle)


vehicle_type = "car"
model = "BMW"
price = 30000
vehicle = Vehicle(vehicle_type, model, price)
vehicle.buy(15000, "Peter")
vehicle.buy(35000, "George")
print(vehicle)
vehicle.sell()
print(vehicle)