Factory method pattern in Python

factory.py

class Car:
  def __init__(self, name, capacity):
    self._name = name
    self._capacity = capacity
  
  def drive(self):
     print("Starting the Car engine!")
  
class Bus:
  def __init__(self, name, capacity):
    self.name = name
    self._capacity = capacity
  
  def drive(self):
    print("Starting the Bus engine!")
  
def get_vehicle(vehicle="Car"):
  '''The Factory method '''
  vehicles = dict(car=Car("BMW", 4), bus=Bus("Yutong", 50))
  
  return vehicles[vehicle]
    

car = get_vehicle('car')
car.drive()

bus = get_vehicle('bus')
bus.drive()