# Import db
from accounts.models import db

class Visit(db.Model):
    __tablename__ = 'visits'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    date_time = db.Column(db.DateTime, nullable=False)
    owner_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
    owner = db.relationship('User', back_populates='visits')
    dog_id = db.Column(db.Integer, db.ForeignKey('dogs.id'), nullable=False)
    dog = db.relationship('Dog', backref='visits')

    def to_dict(self):
        return {
            'ID': self.id,
            'Date-Time': self.date_time,
            'Owner-ID': self.owner_id,
            'Dog-ID': self.dog_id
        }