class User(db.Model, UserMixin):
""" Create User instance for db"""
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), 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.png')
password = db.Column(db.String(60), nullable=False)
posts = db.relationship('Post', backref='author', lazy=True)
github_username = db.Column(db.String(30), unique=True, nullable=False)
def get_reset_token(self, expires_sec=1800):
"""Function to generate timed token for password reset"""
s = TJS(current_app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
"""FUnction to verify token"""
s = TJS(current_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 "User '{}', '{}', '{}', '{}' ".format(self.username, self.email,
self.image_file,
self.github_username)
class Technicalskills(db.Model):
"""DB class for Technical skills"""
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(50))
def __repr__(self):
return "{}".format(self.name)
class GithubUser(db.Model, UserMixin):
"""DB class for storing Github accesstoken temporarily"""
id = db.Column(db.Integer, primary_key=True)
github_access_token = db.Column(db.String(200), db.ForeignKey('user.id'))
#user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
def __repr__(self):
return "Github User '{}' ".format(self.github_access_token)
class Post(db.Model):
"""DB class for posts"""
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
content = db.Column(db.Text, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return "Post {} {}".format(self.title, self.date_posted)
class User(db.Model, UserMixin):
""" Create User instance for db"""
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), 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.png')
password = db.Column(db.String(60), nullable=False)
user_designation = db.Column(db.String(50),
nullable=False,
default='engineer')
short_description = db.Column(
db.Text,
nullable=False,
default='I love writing codes and building stuff')
posts = db.relationship('Post', backref='author', lazy=True)
techskills = db.relationship('Technicalskills',
secondary=skills,
lazy='subquery',
backref=db.backref('user', lazy=True))
#techskills = db.relationship('Technicalskills', backref='user', lazy=True)
github_username = db.Column(db.String(50), unique=True, nullable=False)
github_access_token = db.relationship('GithubUser',
backref='user',
lazy=True)
user_weblink = db.Column(db.String(200),
nullable=False,
default='http://www.iamsacha.nl/')
def get_reset_token(self, expires_sec=1800):
"""Function to generate timed token for password reset"""
s = TJS(current_app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
"""FUnction to verify token"""
s = TJS(current_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 "User '{}', '{}', '{}', '{}', '{}', '{}', '{}' ".format(
self.username, self.email, self.image_file, self.github_username,
self.techskills, self.github_access_token, self.user_weblink)