def end_ride_helper(session, city, ride_id):
ride = session.query(Ride).filter_by(city=city, id=ride_id).first()
v = session.query(Vehicle).filter_by(city=city, id=ride.vehicle_id).first()
ride.end_address = v.current_location
ride.revenue = MovRGenerator.generate_revenue()
ride.end_time = datetime.datetime.now()
v.status = "available"
def end_ride_helper(session, city, ride_id):
ride = session.query(Ride).filter_by(id=ride_id)
r = ride.first()
vehicle = session.query(Vehicle).filter_by(id=r.vehicle_id)
vehicle.update({'status': 'available'})
v = vehicle.first()
ride.update({
'end_address': v.current_location,
'revenue': MovRGenerator.generate_revenue(),
'end_time': datetime.datetime.now()
})
def add_rides_helper(sess, chunk, n):
users = sess.query(User).filter_by(city=city).all()
vehicles = sess.query(Vehicle).filter_by(city=city).all()
rides = []
for i in range(chunk, min(chunk + chunk_size, num_rides)):
start_time = datetime.datetime.now() - datetime.timedelta(days=random.randint(0, 30))
rides.append(Ride(id=MovRGenerator.generate_uuid(),
city=city,
rider_id=random.choice(users).id,
vehicle_id=random.choice(vehicles).id,
start_time=start_time,
start_address=datagen.address(),
end_address=datagen.address(),
revenue=MovRGenerator.generate_revenue(),
end_time=start_time + datetime.timedelta(minutes=random.randint(0, 60))))
sess.bulk_save_objects(rides)