app/user/decorators.py

from functools import wraps
from flask import flash, redirect, url_for
from flask_login import current_user

def check_confirmed(func):
  @wraps(func)
  def decorated_function(*args, **kwargs):
    if not  current_user.confirmed:
      return redirect(url_for('user.unconfirmed'))
    return func(*args, **kwargs)

  return decorated_function

def check_sshkey(func):
  @wraps(func)
  def decorated_function(*args, **kwargs):
    if current_user.ssh_public_key is None:
      return redirect(url_for('user.profile'))
    if not current_user.ssh_public_key:
      return redirect(url_for('user.profile'))
    return func(*args, **kwargs)

  return decorated_function