def user():
if current_user.is_authenticated:
username = current_user.username
user = UserService.get_by_username(username)
if user:
posts = [
{'author': user, 'body': 'Test post #1'},
{'author': user, 'body': 'Test post #2'}
]
return render_template('user.html', user=user, posts=posts)
return 'Not found user'
def index():
# print(app.config['SECRET_KEY'])
if not current_user:
flash("Please login!!")
return redirect(url_for("login"))
user = UserService.get_by_username(current_user.username)
posts = [
{
'author': {'username': '******'},
'body': 'Beautiful day in Portland!'
},
{
'author': {'username': '******'},
'body': 'The Avengers movie was so cool!'
}
]
return render_template('index.html', title="Home", posts=posts, user = user)
def authen():
# if current_user.is_authenticated:
if session.get('username'):
username = session.get('username')
# username = current_user.username
user = UserService.get_by_username(username)
session['base32_key'] = user.base32_key
if request.method == "POST":
totp = pyotp.TOTP(session.get('base32_key'))
otp = request.json['otp']
if totp.verify(otp):
login_user(user)
return {'status': 'Ok',
'next_page': session.get('next_page')
}
return {'status': 'Wrong code'}
return render_template('2authentication.html', user=user)
return "Not found user"
def validate_username(self, username):
user = UserService.get_by_username(username.data)
if user is not None:
raise ValidationError('Please use a different username.')
def qrcode():
if not current_user.is_authenticated:
return redirect(url_for('login'))
user = UserService.get_by_username(current_user.username)
return render_template('qrcode.html', user = user)