class ItemModel(BaseModel):
__tablename__ = "item"
name = db.Column(db.String(20), unique=True, nullable=False)
description = db.Column(db.Text, nullable=False)
category_id = db.Column(db.Integer, db.ForeignKey("category.id"))
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
category = db.relationship("CategoryModel")
user = db.relationship("UserModel")
class ItemModel(BaseModel):
__tablename__ = 'item'
name = db.Column(db.String(45))
description = db.Column(db.Text)
updated = db.Column(db.DateTime,
default=db.func.now(),
onupdate=db.func.now())
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
category = db.relationship('CategoryModel')
user = db.relationship('UserModel')
class ItemModel(BaseModel, db.Model):
__tablename__ = 'items'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), unique=True, nullable=False)
description = db.Column(db.String(1000), nullable=False)
category_id = db.Column(db.Integer,
db.ForeignKey('categories.id'),
nullable=False)
creator_id = db.Column(db.Integer,
db.ForeignKey('users.id'),
nullable=False)
class ItemModel(db.Model, BaseModel):
__tablename__ = 'items'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(ITEM_NAME_LENGTH), nullable=False)
description = db.Column(db.String(MAX_DESC_LENGTH))
category_id = db.Column(db.Integer, db.ForeignKey('categories.id'))
user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
category = db.relationship('CategoryModel', lazy='joined')
user = db.relationship('UserModel', lazy='joined')
def __init__(self, **kwargs):
super(ItemModel, self).__init__(**kwargs)
class CategoryModel(BaseModel, db.Model):
__tablename__ = 'categories'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), unique=True, nullable=False)
description = db.Column(db.String(1000), nullable=False)
creator_id = db.Column(db.Integer,
db.ForeignKey('users.id'),
nullable=False)
items = db.relationship('ItemModel', backref='category', lazy='dynamic')
class ReviewModel(db.Model):
# Specify the db table name for sqlalchemy to link with this model
__tablename__ = 'reviews'
# Specify the table columns for sqlalchemy to link with this model
# The column variables must match with the instance attributes except for primary key
id = db.Column(db.Integer, primary_key=True)
comment = db.Column(db.String)
rating = db.Column(db.SmallInteger)
suggested_rating = db.Column(db.SmallInteger)
sentiment_score = db.Column(db.Float)
# Define foreign key in the reviews table that links each entry to a single entry in the companies table.
company_id = db.Column(db.Integer, db.ForeignKey('companies.id'))
company = db.relationship('CompanyModel')
user_agent = db.Column(db.String)
ip_address = db.Column(db.String)
def __init__(self, comment: str, rating: int, suggested_rating: int,
sentiment_score: float, company_id: int, user_agent: str,
ip_address: str):
self.comment = comment
self.rating = rating
self.suggested_rating = suggested_rating
self.sentiment_score = sentiment_score
self.company_id = company_id
self.user_agent = user_agent
self.ip_address = ip_address
def json(self):
return {
'comment': self.comment,
'rating': self.rating,
'suggested_rating': self.suggested_rating,
'sentiment_score': self.sentiment_score,
'company': self.company.name,
'user_agent': self.user_agent,
'ip_address': self.ip_address
}
# Get all reviews for a particular company
@classmethod
def find_by_company(cls, company_id: int):
return cls.query.filter_by(company_id=company_id).all()
def save_to_db(self):
db.session.add(self)
db.session.commit()
def delete_from_db(self):
db.session.delete(self)
db.session.commit()
class Habit(db.Model):
"""Habit Model for 'habits' table"""
__tablename__ = "habits"
user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
name = db.Column(db.String(255), nullable=False)
countable = db.Column(db.Boolean, default=False, nullable=False)
status = db.Column(db.String(32), default=HabitStatus.ACTIVE, nullable=False)
def __repr__(self) -> str:
return (
f"<Habit(id={self.id}, name={self.name}, user_id={self.user_id}, "
+ f"countable={self.countable}, status={self.status}, "
+ f"created_at={self.created_at}, updated_at={self.updated_at})>"
)
class Log(db.Model):
"""Log Model for 'logs' table"""
__tablename__ = "logs"
habit_id = db.Column(db.Integer,
db.ForeignKey("habits.id"),
nullable=False)
date = db.Column(db.Date, nullable=False)
count = db.Column(db.Integer, nullable=True) # defaults to NULL
status = db.Column(db.String(32), default=LogStatus.ACTIVE, nullable=False)
def __repr__(self) -> str:
return (
f"<Log(habit_id={self.habit_id}, date={self.date}, count={self.count}, "
+
f"id={self.id}, created_at={self.created_at}, updated_at={self.updated_at})>"
)