diff --git a/app/models.py b/app/models.py index 813dd46..27254f7 100644 --- a/app/models.py +++ b/app/models.py @@ -10,6 +10,7 @@ class User(UserMixin, db.Model): email = db.Column(db.String(120), index=True, unique=True) password_hash = db.Column(db.String(64)) posts = db.relationship('Post', backref='author', lazy='dynamic') + search_history = db.relationship('Search', backref='user', lazy='dynamic') def set_password(self, password): self.password_hash = generate_password_hash(password) @@ -22,6 +23,15 @@ class User(UserMixin, db.Model): def __repr__(self): return ''.format(self.username) +class Search(db.Model): + id = db.Column(db.Integer, primary_key=True) + query = db.Column(db.String(120)) + timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow) + user_id = db.Column(db.Integer, db.ForeignKey('user.id')) + + def __repr__(self): + return ''.format(self.query if len(self.query) < 10 else self.query[:10] + '...') + class Post(db.Model): id = db.Column(db.Integer, primary_key=True) body = db.Column(db.String(140))