from app import db

class Vehicle(db.Model):
    __tablename__ = 'vehicles'
    vin = db.Column(db.String, primary_key=True)
    make = db.Column(db.String)
    model = db.Column(db.String)
    year = db.Column(db.String)

class WarrantyRegistration(db.Model):
    __tablename__ = 'warranty_registrations'
    id = db.Column(db.Integer, primary_key=True)
    vin = db.Column(db.String, db.ForeignKey('vehicles.vin'))
    customer_name = db.Column(db.String, nullable=False)
    customer_email = db.Column(db.String, nullable=False)
    installer_business = db.Column(db.String, nullable=False)
    installed_product = db.Column(db.String, nullable=False)
    installation_date = db.Column(db.Date, nullable=False)
    make = db.Column(db.String)  # New column
    model = db.Column(db.String)  # New column
    year = db.Column(db.String)  # New column

    vehicle = db.relationship('Vehicle', backref=db.backref('registrations', lazy=True))