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})>" )