36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
# Standard library imports
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import Enum
|
|
|
|
# Import db
|
|
from accounts.models import db
|
|
|
|
from accounts.models.dog.breed_size import BreedSize
|
|
from accounts.models.status import Status
|
|
|
|
class Dog(db.Model):
|
|
__tablename__ = 'dogs'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
name = db.Column(db.String, nullable=False)
|
|
breed_size = db.Column(Enum(BreedSize), default=BreedSize.MEDIUM)
|
|
age = db.Column(db.DateTime, default=datetime(1970, 1, 1, 0, 0, 0))
|
|
owner_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
owner = db.relationship('User', back_populates='dogs')
|
|
visits = db.relationship('Visit', back_populates='dogs')
|
|
status = db.Column(Enum(Status), default=Status.ACTIVE)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
updated_at = db.Column(db.DateTime, onupdate=datetime.utcnow)
|
|
|
|
def __repr__(self):
|
|
return f'{self.name}'
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'name': self.name,
|
|
'breed': self.breed,
|
|
'owner': self.owner
|
|
}
|
|
|