Files
electric-vehicle-battery-ch…/test/utils.py
2023-05-18 18:21:00 -04:00

12 lines
370 B
Python

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()