class Story(db.Model):
id = db.Column(db.Integer, primary_key=True)
story_image = db.Column(db.String(20))
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
def __repr__(self):
return '<Story {}>'.format(self.body)
class Like(db.Model):
id = db.Column(db.Integer, primary_key=True)
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
def __repr__(self):
return '<Like {}>'.format(self.body)
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.String(140))
post_image = db.Column(db.String(20))
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
comments = db.relationship('Comment', backref='post', lazy='dynamic')
likes = db.relationship('Like', backref='post', lazy='dynamic')
def __repr__(self):
return '<Post {}>'.format(self.body)
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
sender_id = db.Column(db.Integer, db.ForeignKey('user.id'))
recipient_id = db.Column(db.Integer, db.ForeignKey('user.id'))
body = db.Column(db.String(140))
image = db.Column(db.String(20))
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
def __repr__(self):
return '<Message {}>'.format(self.body)
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
f_name = db.Column(db.String(64), index=True, nullable=False)
l_name = db.Column(db.String(64), index=True, nullable=False)
username = db.Column(db.String(128), index=True, nullable=False)
dob = db.Column(db.DateTime, nullable=True, index=True)
gender = db.Column(db.String(6), index=True, nullable=True)
email = db.Column(db.String(120), index=True, unique=True, nullable=False)
password_hash = db.Column(db.String(128))
cover_image = db.Column(db.String(20), default='default.jpeg')
profile_image = db.Column(db.String(20), default='default.jpg')
story_image = db.Column(db.String(20))
about_me = db.Column(db.String(140), default='No Bio')
last_seen = db.Column(db.DateTime, default=datetime.utcnow)
school = db.Column(db.String(140), default="--add data--")
hometown = db.Column(db.String(140), default="--add data--")
location = db.Column(db.String(140), default="--add data--")
relationship = db.Column(db.String(140), default="--add data--")
posts = db.relationship('Post', backref='author', lazy='dynamic')
stories = db.relationship('Story', backref='author', lazy='dynamic')
comments = db.relationship('Comment', backref='author', lazy='dynamic')
likes = db.relationship('Like', backref='author', lazy='dynamic')
followed = db.relationship('User',
secondary=followers,
primaryjoin=(followers.c.follower_id == id),
secondaryjoin=(followers.c.followed_id == id),
backref=db.backref('followers', lazy='dynamic'),
lazy='dynamic')
messages_sent = db.relationship('Message',
foreign_keys='Message.sender_id',
backref='author',
lazy='dynamic')
messages_received = db.relationship('Message',
foreign_keys='Message.recipient_id',
backref='recipient',
lazy='dynamic')
last_message_read_time = db.Column(db.DateTime)
def new_messages(self):
last_read_time = self.last_message_read_time or datetime(1900, 1, 1)
return Message.query.filter_by(recipient=self).filter(
Message.timestamp > last_read_time).count()
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def set_username(self, f_name, l_name):
self.username = f_name + '.' + l_name
def check_password(self, password):
return check_password_hash(self.password_hash, password)
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 like_post(self, post):
if not self.has_liked_post(post):
like = Like(author_id=self.id, post_id=post.id)
db.session.add(like)
def unlike_post(self, post):
if self.has_liked_post(post):
Like.query.filter_by(author_id=self.id, post_id=post.id).delete()
def has_liked_post(self, post):
return Like.query.filter(Like.author_id == self.id, Like.post_id
== post.id).count() > 0
def get_reset_password_token(self, expires_in=600):
return jwt.encode(
{
'reset_password': self.id,
'exp': time() + expires_in
},
app.config['SECRET_KEY'],
algorithm='HS256')
@staticmethod
def verify_reset_password_token(token):
try:
id = jwt.decode(token,
app.config['SECRET_KEY'],
algorithms=['HS256'])['reset_password']
except:
return
return User.query.get(id)
def __repr__(self):
return '<User {}>'.format(self.username)
from facebook import db, login
from datetime import datetime
from flask_login import UserMixin
from werkzeug.security import generate_password_hash, check_password_hash
import jwt
from time import time
from facebook import app
followers = db.Table(
'followers', db.Column('follower_id', db.Integer,
db.ForeignKey('user.id')),
db.Column('followed_id', db.Integer, db.ForeignKey('user.id')))
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
f_name = db.Column(db.String(64), index=True, nullable=False)
l_name = db.Column(db.String(64), index=True, nullable=False)
username = db.Column(db.String(128), index=True, nullable=False)
dob = db.Column(db.DateTime, nullable=True, index=True)
gender = db.Column(db.String(6), index=True, nullable=True)
email = db.Column(db.String(120), index=True, unique=True, nullable=False)
password_hash = db.Column(db.String(128))
cover_image = db.Column(db.String(20), default='default.jpeg')
profile_image = db.Column(db.String(20), default='default.jpg')
story_image = db.Column(db.String(20))
about_me = db.Column(db.String(140), default='No Bio')
last_seen = db.Column(db.DateTime, default=datetime.utcnow)
school = db.Column(db.String(140), default="--add data--")
hometown = db.Column(db.String(140), default="--add data--")
location = db.Column(db.String(140), default="--add data--")