Started using Bootstrap for styling, moved app factory, added tests, started work on login and registration templates, and more
This commit is contained in:
@@ -2,6 +2,7 @@
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import Enum
|
||||
from sqlalchemy.ext.hybrid import hybrid_property
|
||||
|
||||
# Import db
|
||||
from accounts.models import db
|
||||
@@ -13,23 +14,44 @@ class Dog(db.Model):
|
||||
__tablename__ = 'dogs'
|
||||
|
||||
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
||||
owner = db.relationship('User', back_populates='dogs')
|
||||
owner_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
||||
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')
|
||||
birth_date = db.Column(db.DateTime, default=datetime(1970, 1, 1))
|
||||
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 __str__(self):
|
||||
return self.name
|
||||
|
||||
def __repr__(self):
|
||||
return f'{self.name}'
|
||||
return self.to_dict()
|
||||
|
||||
@hybrid_property
|
||||
def status(self):
|
||||
return self.owner.status
|
||||
|
||||
def get_age(self):
|
||||
current_date = datetime.now()
|
||||
age = (current_date.year - self.birth_date.year
|
||||
- ((current_date.month, current_date.day) < (self.birth_date.month, self.birth_date.day)))
|
||||
|
||||
return age
|
||||
|
||||
def to_dict(self):
|
||||
return {
|
||||
'id': self.id,
|
||||
'owner': self.owner,
|
||||
'owner_id': self.owner_id,
|
||||
'name': self.name,
|
||||
'breed': self.breed,
|
||||
'owner': self.owner
|
||||
'breed_size': self.breed_size,
|
||||
'birth_date': self.birth_date,
|
||||
'age': self.get_age(),
|
||||
'visits': self.visits,
|
||||
'status': self.status,
|
||||
'created_at': self.created_at,
|
||||
'updated_at': self.updated_at
|
||||
}
|
||||
|
||||
|
||||
@@ -2,7 +2,8 @@ from enum import Enum
|
||||
from enum import auto
|
||||
|
||||
class BreedSize(Enum):
|
||||
SMALL = auto()
|
||||
MEDIUM = auto()
|
||||
LARGE = auto()
|
||||
SMALL = 'Small'
|
||||
MEDIUM = 'Medium'
|
||||
LARGE = 'Large'
|
||||
XLARGE = 'X-large'
|
||||
|
||||
|
||||
Reference in New Issue
Block a user