class Follow(db.Model):
__tablename__ = 'follows'
follower_id = db.Column(db.Integer, db.ForeignKey('user.id'),
primary_key=True)
followed_id = db.Column(db.Integer, db.ForeignKey('user.id'),
primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
class Travel(db.Model):
__tablename__ = 'travel'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
place = db.Column(db.String)
start_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
end_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
class Address(db.Model):
__tablename__ = 'address'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
street = db.Column(db.Text)
city = db.Column(db.Text)
zip_code = db.column(db.text)
country = db.Column(db.Text)
class Role(db.Model):
__tablename__ = 'roles'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
default = db.Column(db.Boolean, default=False, index=True)
permissions = db.Column(db.Integer)
users = db.relationship('User', backref='role', lazy='dynamic')
def __init__(self, **kwargs):
super(Role, self).__init__(**kwargs)
if self.permissions is None:
self.permissions = 0
@staticmethod
def insert_roles():
roles = {
'User': [Permission.FOLLOW, Permission.COMMENT, Permission.WRITE],
'Moderator': [Permission.FOLLOW, Permission.COMMENT,
Permission.WRITE, Permission.MODERATE],
'Administrator': [Permission.FOLLOW, Permission.COMMENT,
Permission.WRITE, Permission.MODERATE,
Permission.ADMIN],
}
default_role = 'User'
for r in roles:
role = Role.query.filter_by(name=r).first()
if role is None:
role = Role(name=r)
role.reset_permissions()
for perm in roles[r]:
role.add_permission(perm)
role.default = (role.name == default_role)
db.session.add(role)
db.session.commit()
def add_permission(self, perm):
if not self.has_permission(perm):
self.permissions += perm
def remove_permission(self, perm):
if self.has_permission(perm):
self.permissions -= perm
def reset_permissions(self):
self.permissions = 0
def has_permission(self, perm):
return self.permissions & perm == perm
def __repr__(self):
return '<Role %r>' % self.name
class Events(db.Model):
__tablename__ = 'events'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
event_name = db.Column(db.String)
event_description = db.Column(db.String)
event_location = db.Column(db.String)
event_start_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
event_end_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
event_status = db.Column(db.Boolean)
class Blog(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
blog_city = db.Column(db.String(50), nullable=False)
blog_category = db.Column(db.String(50), nullable=False)
blog_story_line = db.Column(db.String(500), nullable=False)
blog_story_text = db.Column(db.Text, nullable=False)
blog_youtube_link = db.Column(db.String(500), nullable=True)
blog_date_posted = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
blogcomments = db.relationship('BlogComment', backref='blog', lazy='dynamic')
def __repr__(self):
return f"Blog('{self.blog_author}', '{self.date_posted}')"
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
city = db.Column(db.String(50), nullable=False)
category = db.Column(db.String(50), nullable=False)
story_line = db.Column(db.String(500), nullable=False)
story_text = db.Column(db.Text, nullable=False)
youtube_link = db.Column(db.String(500), nullable=True)
date_posted = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
comments = db.relationship('Comment', backref='post', lazy='dynamic')
def __repr__(self):
return f"Post('{self.Protagonist}', '{self.date_posted}')"
class Sponsership(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
sponser_type = db.Column(db.String)
sponser_name = db.Column(db.String)
sponsership_start_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
sponsership_end_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
class Occupation(db.Model):
__tablename__ = 'occupation'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
occupation_name = db.Column(db.Text)
occupation_company = db.Column(db.Text)
occupation_start_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
occupation_end_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
class Links(db.Model):
__tablename__ = 'links'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
facebook_id = db.Column(db.Text)
twitter_id = db.Column(db.Text)
instagram_id = db.Column(db.Text)
snapchat_id = db.Column(db.Text)
class BlogComment(db.Model):
__tablename__ = 'blogcomments'
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.Text)
body_html = db.Column(db.Text)
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
disabled = db.Column(db.Boolean)
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
blog_id = db.Column(db.Integer, db.ForeignKey('blog.id'))
@staticmethod
def on_changed_body(target, value, oldvalue, initiator):
allowed_tags = ['a', 'abbr', 'acronym', 'b', 'code', 'em', 'i',
'strong']
target.body_html = bleach.linkify(bleach.clean(
markdown(value, output_format='html'),
tags=allowed_tags, strip=True))
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
username = db.Column(db.String(20), index=True, unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
image_file = db.Column(db.String(20), nullable=False, default='default.jpg')
password = db.Column(db.String(60), nullable=False)
about_me = db.Column(db.String(140))
social_id = db.Column(db.String(140))
gender = db.Column(db.String(5))
contact_number = db.Column(db.Integer)
education_level = db.Column(db.String(60))
date_of_birth = db.Column(db.DateTime, default=datetime.utcnow)
last_seen = db.Column(db.DateTime, default=datetime.utcnow)
address = db.relationship('Address', backref='Protagonist', lazy='dynamic')
occumpation = db.relationship('Occupation', backref='Protagonist', lazy='dynamic')
posts = db.relationship('Post', backref='Protagonist', lazy='dynamic')
blog = db.relationship('Blog', backref = 'blog_author', lazy= 'dynamic')
member_since = db.Column(db.DateTime(), default=datetime.utcnow)
followed = db.relationship('Follow',
foreign_keys=[Follow.follower_id],
backref=db.backref('follower', lazy='joined'),
lazy='dynamic',
cascade='all, delete-orphan')
followers = db.relationship('Follow',
foreign_keys=[Follow.followed_id],
backref=db.backref('followed', lazy='joined'),
lazy='dynamic',
cascade='all, delete-orphan')
comments = db.relationship('Comment', backref='Protagonist', lazy='dynamic')
blog_comments = db.relationship('BlogComment', backref='blog_author', lazy='dynamic')
links = db.relationship('Links', backref='Protagonist', lazy='dynamic')
intrest = db.relationship('Intrest', backref='Protagonist', lazy='dynamic')
achievement = db.relationship('Achievement', backref='Protagonist', lazy='dynamic')
events = db.relationship('Events', backref='Protagonist', lazy='dynamic')
travel = db.relationship('Travel', backref='Protagonist', lazy='dynamic')
special_event = db.relationship('Special_event', backref='Protagonist', lazy='dynamic')
media = db.relationship('Media', backref='Protagonist', lazy='dynamic')
language = db.relationship('Language', backref='Protagonist', lazy='dynamic')
sponsership = db.relationship('Sponsership', backref='Protagonist', lazy='dynamic')
def __init__(self, **kwargs):
super(User, self).__init__(**kwargs)
if self.role is None:
if self.email == app.config['FLASKY_ADMIN']:
self.role = Role.query.filter_by(name='Administrator').first()
if self.role is None:
self.role = Role.query.filter_by(default=True).first()
self.follow(self)
def can(self, perm):
return self.role is not None and self.role.has_permission(perm)
def is_administrator(self):
return self.can(Permission.ADMIN)
def follow(self, user):
if not self.is_following(user):
f = Follow(follower=self, followed=user)
db.session.add(f)
def unfollow(self, user):
f = self.followed.filter_by(followed_id=user.id).first()
if f:
db.session.delete(f)
def is_following(self, user):
if user.id is None:
return False
return self.followed.filter_by(
followed_id=user.id).first() is not None
def is_followed_by(self, user):
if user.id is None:
return False
return self.followers.filter_by(
follower_id=user.id).first() is not None
def get_reset_token(self, expires_sec=1800):
s = Serializer(app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
def __repr__(self):
return f"User('{self.username}', '{self.user_id}', '{self.email}', '{self.image_file}', '{self.id}')"
class Language(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
language = db.Column(db.String)
language_accuracy = db.Column(db.String)
class Media(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
media_name = db.Column(db.String)
media_format = db.Column(db.String)
class Special_event(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
life_event = db.Column(db.String)
life_event_start_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
life_event_end_date = db.Column(db.DateTime, index=True, default=datetime.utcnow)
class Achievement(db.Model):
__tablename__ = 'achievements'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
medal_count = db.Column(db.String)
timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
class Intrest(db.Model):
__tablename__ = 'intrest'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
intrest_type = db.Column(db.Text)
