class Post(SearchableMixin, lws_db.Model):
__searchable__ = ['body']
id = lws_db.Column(lws_db.Integer, primary_key=True)
body = lws_db.Column(lws_db.String(140))
timestamp = lws_db.Column(lws_db.DateTime, index=True, default=datetime.utcnow)
user_id = lws_db.Column(lws_db.Integer, lws_db.ForeignKey('user.id'))
language = lws_db.Column(lws_db.String(5))
def __repr__(self):
return '<Post {}>'.format(self.body)
class Ingredient(lws_db.Model):
id = lws_db.Column(lws_db.Integer, primary_key=True)
name = lws_db.Column(lws_db.String(140))
def __repr__(self):
return '<Ingredient {}>'.format(self.name)
class Menu(lws_db.Model):
id = lws_db.Column(lws_db.Integer, primary_key=True)
name = lws_db.Column(lws_db.String(140))
timestamp = lws_db.Column(lws_db.DateTime, index=True, default=datetime.utcnow)
confirmed = lws_db.Column(lws_db.Boolean, default=False)
dishes = lws_db.relationship('Dish', secondary=menus, lazy='dynamic',
backref=lws_db.backref('menus', lazy='dynamic'))
def __repr__(self):
return '<Menu {}>'.format(self.name)
def add_dish(self, dish):
if not self.has_dish(dish):
self.dishes.append(dish)
def remove_dish(self, dish):
if self.has_dish(dish):
self.dishes.remove(dish)
def has_dish(self, dish):
return self.dishes.filter(
menus.c.dish_id == dish.id).count() > 0
def get_ingredients(self):
ingredients = []
for dish in self.dishes:
for ingredient in dish.ingredients:
if ingredient not in ingredients:
ingredients.append(ingredient)
return ingredients
class StockShare(lws_db.Model):
id = lws_db.Column(lws_db.Integer, primary_key=True)
name = lws_db.Column(lws_db.String(10))
buy_price = lws_db.Column(lws_db.Float)
sell_price = lws_db.Column(lws_db.Float)
buy_timestamp = lws_db.Column(lws_db.DateTime, index=True)
sell_timestamp = lws_db.Column(lws_db.DateTime)
def __repr__(self):
return '<StockShare id={0}, {1}>'.format(self.id, self.name)
class Dish(lws_db.Model):
id = lws_db.Column(lws_db.Integer, primary_key=True)
name = lws_db.Column(lws_db.String(140))
ingredients = lws_db.relationship('Ingredient', secondary=ingredients, lazy='dynamic',
backref=lws_db.backref('dishes', lazy='dynamic'))
def __repr__(self):
return '<Dish {}>'.format(self.name)
def add_ingredient(self, ingredient):
if not self.has_ingredient(ingredient):
self.ingredients.append(ingredient)
def remove_ingredient(self, ingredient):
if self.has_ingredient(ingredient):
self.ingredients.remove(ingredient)
def has_ingredient(self, ingredient):
return self.ingredients.filter(
ingredients.c.ingredient_id == ingredient.id).count() > 0
class User(UserMixin, lws_db.Model):
id = lws_db.Column(lws_db.Integer, primary_key=True)
username = lws_db.Column(lws_db.String(64), index=True, unique=True)
email = lws_db.Column(lws_db.String(120), index=True, unique=True)
password_hash = lws_db.Column(lws_db.String(128))
posts = lws_db.relationship('Post', backref='author', lazy='dynamic')
about_me = lws_db.Column(lws_db.String(140))
last_seen = lws_db.Column(lws_db.DateTime, default=datetime.utcnow)
followed = lws_db.relationship(
'User', secondary=followers,
primaryjoin=(followers.c.follower_id == id),
secondaryjoin=(followers.c.followed_id == id),
backref=lws_db.backref('followers', lazy='dynamic'), lazy='dynamic')
def __repr__(self):
return '<User {}>'.format(self.username)
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
def avatar(self, size):
digest = md5(self.email.lower().encode('utf-8')).hexdigest()
return 'https://www.gravatar.com/avatar/{}?d=identicon&s={}'.format(
digest, size)
def follow(self, user):
if not self.is_following(user):
self.followed.append(user)
def unfollow(self, user):
if self.is_following(user):
self.followed.remove(user)
def is_following(self, user):
return self.followed.filter(
followers.c.followed_id == user.id).count() > 0
def followed_posts(self):
followed = Post.query.join(
followers, (followers.c.followed_id == Post.user_id)).filter(
followers.c.follower_id == self.id)
own = Post.query.filter_by(user_id=self.id)
return followed.union(own).order_by(Post.timestamp.desc())
def get_reset_password_token(self, expires_in=600):
return jwt.encode(
{'reset_password': self.id, 'exp': time() + expires_in},
current_app.config['SECRET_KEY'], algorithm='HS256').decode('utf-8')
@staticmethod
def verify_reset_password_token(token):
try:
id = jwt.decode(token, current_app.config['SECRET_KEY'],
algorithms=['HS256'])['reset_password']
except:
return
return User.query.get(id)