from app.extensions import db
from app.models.vehicle import Vehicle
from .vehicle_data import vehicle_data

def create_vehicles(length = None):
    if length:
        for vehicle in vehicle_data[:length]:
            db.session.add(Vehicle(**vehicle))
    else:
        for vehicle in vehicle_data:
            db.session.add(Vehicle(**vehicle))
    db.session.commit()