from app.extensions import db, ma

class BatteryChange(db.Model):
    order = db.Column(db.Integer)
    shift_id = db.Column(db.Integer, db.ForeignKey('shift.id'), primary_key=True, nullable=False)
    vehicle_id = db.Column(db.Integer, db.ForeignKey('vehicle.id'), primary_key=True, nullable=False)
    completed = db.Column(db.Boolean, nullable=False, default=False)
    vehicle = db.relationship('Vehicle', backref='change_to_vehicle')

class BatteryChangeSchema(ma.Schema):
    vehicle = ma.Nested('VehicleSchema', only=['license_plate', 'battery_level', 
                                               'in_use', 'model', 'location_lat', 
                                               'location_long', 'id'], many=False)
    class Meta:
        model=BatteryChange
        fields = ('shift_id', 'vehicle_id', 'completed', 'vehicle')