Python flask_login.current_user 模块，confirmed() 实例源码

def unconfirmed() :
    if current_user.is_anonymous() or current_user.confirmed :
        return redirect(url_for('auth.unconfirmed')
    return render_template('auth/unconfirmed.html')


#??????????
@auth.route('/confirm')
@login_required
def resend_confirmation():
    token = current_user.generate_confirmation_token()
    send_email(current_user.email,'Confim Your Accout' , 'auth/email/confirm',user=current_user,token=token)
    flash('A new confirmation email has been sent to you by email.')
    return redirect(url_for('main.index'))


# ???????
@main.route('/user/<username>')
def user(username) :
    user = User.query.filter_by(username=username).first()
    if user is None :
        abort(404) 
    return render_template('user.html',user=user)

def confirm_email(token):
    try:
        # try confirm_email
        email = confirm_token(token)
        user = User.query.filter_by(email=email).first_or_404()
        if user.confirmed:
            flash('Account already confirmed Please login.', 'success')
        else:
            user.confirmed = True
            # user follow himself to show his posts in the main page
            user.follow(user)
            db.session.add(user)
            db.session.commit()
            login_user(user)
            flash('You have confirmed your account Thanks', 'success')
    except:
        flash('The confirmation link is invalid or has expired.', 'danger')
    return redirect(url_for('users.discover_users'))

def setting_users(id):
    user = User.query.get_or_404(id)
    form = EditProfileAdminForm(user=user)
    if form.validate_on_submit():
        user.email = form.email.data
        user.username = form.username.data
        user.confirmed = form.confirmed.data
        user.role = Role.query.get(form.role.data)
        user.name = form.name.data
        db.session.add(user)
        db.session.commit()
        flash(constant.PROFILE_UPDATE)
        return redirect(url_for('auth.setting_users', id=id))
    form.email.data = user.email
    form.username.data = user.username
    form.confirmed.data = user.confirmed
    form.role.data = user.role_id
    form.name.data = user.name
    return render_template('auth/admin_edit_profile.html', form=form)

def register():
    form = RegisterForm(request.form)
    if form.validate_on_submit():
        user = User(
            first_name=form.first_name.data,
            last_name=form.last_name.data,
            email=form.email.data,
            password=form.password.data,
            confirmed=False,
        )
        db.session.add(user)
        db.session.commit()

        token = generate_confirmation_token(user.email)
        confirm_url = url_for('user.confirm_email', token=token, _external=True)
        html = render_template('user/activate.html', confirm_url=confirm_url)
        subject = "Please confirm your email"
        send_email(user.email, subject, html)

        login_user(user)
        flash('You registered and are now logged in. Welcome!', 'success')

        return redirect(url_for('user.unconfirmed'))

    return render_template('user/register.html', form=form)

def confirm_email(token):
    try:
        email = confirm_token(token)
        user = User.query.filter_by(email=email).first_or_404()
        if user.confirmed:
            flash('Account already confirmed. Please login', 'success')
        else:
            user.confirmed = True
            user.confirmed_on = datetime.datetime.now()
            db.session.add(user)
            db.session.commit()
            if g.user is None or g.user.id != user.id:
                login_user(user)
            flash('You have confirmed your account. Thank You!', 'success')
    except:
        flash('The confirmation link is invalid or has expired.', 'danger')
    return redirect(url_for('main.home'))

def confirm(token):

    if current_user.confirmed:
        # print current_user
        # print type(current_user)
        # print current_user.id
        # print current_app.config
        #?????????????????
        return redirect(url_for('main.index'))
    #??token
    if current_user.confirm(token):
        flash('You have confirmed your account.Thanks!')
    else:
        flash('The confirmation link is invalid or has expried')

    return redirect(url_for('main.index'))


#??????????

def confirm_email(token):
    try:
        email = confirm_token(token)
    except:
        flash('The confirmation link is invalid or has expired.', 'danger')
        return redirect(url_for('main.index'))

    # If the registration code is valid and unused, modify User table ----------
    user = User.query.filter_by(email = email).first()
    siteMetrics = SiteMetrics.query.filter_by(id = 1).first()
    if user.confirmed:
        flash('Account is already confirmed. Please login.', 'success')
        return redirect(url_for('main.login'))
    elif user:
        user.confirmed = True
        user.confirmed_on = datetime.datetime.now()
        siteMetrics.num_registered = user.id
        db.session.add(user)
        db.session.add(siteMetrics)
        db.session.commit()
        flash('Your account has been confirmed.', 'succes')
        return redirect(url_for('main.login'))
    # If invalid input is provided, go back to the index page ------------------
    else:
        return redirect(url_for('main.index'))


# ------------------------------------------------------------------------------
# Unconfirmed Email
# ------------------------------------------------------------------------------

def unconfirmed():
    if current_user.confirmed:
        return redirect(url_for('main.user_profile'))
    else:
        return render_template('waiting.html')


# ------------------------------------------------------------------------------
# Login
# ------------------------------------------------------------------------------

def confirm(token) :
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    if current_user.confirm(token) :
        flash('Your have confirmed your account . Thanks!')
    else :
        flash('The confirmation link is invalid or has exired.')
    return redirect(url_for('main.index'))

#???????? , ????????

def before_request() :
    if current_user.is_authenticated() \
            current_user.ping()  # ????????????
            if  current_user.confirmed \
                and request.endpoint[:5] != 'auth.' :
                and request.endpoint != 'static' :
            return redirect(url_for('auth.unconfirmed'))

def signup():
    form = SignUpForm()
    if form.validate_on_submit():
        # POST request
        check_u = User.query.filter_by(email=form.email.data).first()
        if check_u:
            if check_u.email == form.email.data:
                flash('email already exsist', "danger")
                return redirect(url_for('auth.signup'))
        check_u2 = User.query.filter_by(name=form.username.data).first()
        if check_u2:
            if check_u2.name == form.username.data:
                flash('username already exsist', "danger")
                return redirect(url_for('auth.signup'))
        # create new User instance
        u = User(
            name=form.username.data,
            email=form.email.data,
            password=form.password.data,
            confirmed=False
        )
        # add new user to the database
        db.session.add(u)
        db.session.commit()
        token = generate_confirmation_token(u.email)
        confirm_url = url_for('confirm_email', token=token, _external=True)
        html = render_template('email_activate.html', confirm_url=confirm_url)
        subject = "email confirmation for minitweet"
        send_email(u.email, subject, html)
        login_user(u)
        flash('A confirmation email has been sent via email', 'success')
        return redirect(url_for("unconfirmed"))

    # GET request
    return render_template("signup.html", form=form, title="signup")

def unconfirmed():
    if current_user.confirmed or not current_user.is_authenticated:
        return redirect(url_for('main.home', page=1))
    flash('Please confirm your account', 'warning')
    return render_template('unconfirmed.html', title="unconfirmed")

def home(page=1):
    """
    Main Page
    if the user is logged in show the posts from followed people
    else show all posts

    param: page -> for pagination
    example urls:
        /   (page=1)     # default
        /posts  (page=1)    # default
        /posts/newest/<page>    (page=page)
    """
    if current_user.is_authenticated and current_user.confirmed:
        # get posts from followed people
        posts = current_user.get_posts_from_followed_users()
    else:
        # query all posts in desceding order
        posts = Post.query.order_by(Post.id.desc())
    per_page = app.config["POSTS_PER_PAGE"]
    paginated_posts = posts.paginate(page, per_page)
    next_url = url_for("main.home", page=page + 1)
    prev_url = url_for("main.home", page=page - 1)
    return render_template("index.html",
                           posts=paginated_posts,
                           title="newest",
                           next_url=next_url,
                           prev_url=prev_url
                           )

def like_post(post_id):
    """
    like a post
    the request will sent with ajax
    example returned json:
        status: either "good" or "error"
        like: if user like the post like = True else like = None
        msg: if there is alert msg="some alert" else msg=None
        category: category of the msg (warning, primary, success)
    """
    msg = None
    like = None
    category = None
    post = Post.query.get(post_id)
    if not current_user.is_authenticated:
        status = "error"
        msg = "Please Login or signup first"
        category = "warning"
    elif not current_user.confirmed:
        status = "error"
        msg = "Please confirm your email first"
        category = "warning"
    else:
        if current_user.is_liking(post):
            current_user.unlike(post)
            like = False
        else:
            current_user.like(post)
            like = True
        status = "good"
        db.session.add(current_user)
        db.session.commit()
    return jsonify({"status": status,
                    "msg": msg,
                    "category": category,
                    "like": like,
                    "likes_counting": post.likers.count()
                    })

def before_request():
    if current_user.is_authenticated() and current_user.is_anonymous() is False:
        current_user.ping()
        if not current_user.confirmed and request.endpoint and request.endpoint[:4] == 'view' and request.endpoint != 'static':
            return redirect(url_for('unconfirmed'))

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('view_rents'))
    if current_user.confirm(token):
        flash(u'??????????')
    else:
        flash(u'?????')
    return redirect(url_for('view_rents'))

def unconfirmed():
    if current_user.is_anonymous() is True or current_user.confirmed:
        return redirect(url_for('view_rents'))
    form = PageInfo()
    logic.LoadBasePageInfo('?????', form)
    return render_template('auth/unconfirmed.html', form=form)

def before_request():
    if current_user.is_authenticated:
        if not current_user.confirmed \
                and request.endpoint \
                and request.endpoint[:5] != 'user.' \
                and request.endpoint != 'static':
            return redirect(url_for('user.unconfirmed'))

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('home.index'))
    if current_user.confirm(token):
        flash('You have confirmed your account. Thanks!')
    else:
        flash('The confirmation link is invalid or has expired.')
    return redirect(url_for('home.index'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for('home.index'))
    return render_template('user/unconfirmed.html')

def before_request():
    if current_user.is_authenticated:
        if not current_user.confirmed \
                and request.endpoint[:5] != 'auth.'\
                and request.endpoint != 'static':
            return redirect(url_for('auth.unconfirmed'))

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    if current_user.confirm(token):
        flash(constant.VERIFI_SUCCESS)
    else:
        flash(constant.LINK_FAIL)
    return redirect(url_for('main.index'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for('main.index'))
    return render_template('auth/unconfirmed.html', user=current_user)

def before_request():
    if current_user.is_authenticated:
        current_user.ping()
        if not current_user.confirmed and request.endpoint[:5] != 'auth.' and request.endpoint != 'static':
            return redirect(url_for('auth.unconfirmed'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for('main.index'))
    return render_template('auth/unconfirmed.html')

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    if current_user.confirm(token):
        flash('You have confirmed your accont. Thanks!')
    else:
        flash('The confirmation link is invalid or has expired.')
    return redirect(url_for('main.index'))

def check_confirmed(func):
    @wraps(func)
    def decorated_function(*args, **kwargs):
        if current_user.confirmed is False:
            flash('Please confirm your account!', 'warning')
            return redirect(url_for('user.unconfirmed'))
        return func(*args, **kwargs)

    return decorated_function

def unconfirmed():
    if current_user.confirmed:
        return redirect('main.home')
    flash('Please confirm your account!', 'warning')
    return render_template('user/unconfirmed.html')

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    if current_user.confirm(token):
        flash('You have confirmed your account. Thanks!')
    else:
        flash('The confirmation link is invalid or has expired.')
    return redirect(url_for('main.index'))

def before_request():
    if current_user.is_authenticated:
        current_user.ping()
        if not current_user.confirmed and request.endpoint \
            and request.endpoint \
            and request.endpoint[:5] != 'auth.' \
            and request.endpoint != 'static':
            return redirect(url_for('auth.unconfirmed'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for('main.index'))
    return render_template('auth/unconfirmed.html')

def before_request():
    if current_user.is_authenticated:
        current_user.ping()     # ??????????
        if not current_user.confirmed \
                and request.endpoint \
                and request.endpoint[:5] != 'auth.' \
                and request.endpoint != 'static':   #?????????????????????auth??????
            return redirect(url_for('auth.unconfirmed'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:     #???????????????
        return redirect(url_for('main.index'))
    return render_template('auth/unconfirmed.html')

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    elif current_user.confirm(token):
        flash('you have confirmed your token')


    else:
        flash('The confirmation link is invalid or expired')
    return redirect(url_for('main.index'))

def before_request():
    if current_user.is_authenticated:
        current_user.ping()
        if not current_user.confirmed \
                and request.endpoint \
                and request.endpoint[:5] != 'auth.' \
                and request.endpoint != 'static':
            return redirect(url_for('auth.unconfirmed'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for ('main.index'))
    return render_template('auth/unconfirmed.html')

def confirm(token):
    if current_user.confirmed:
        return redirect(url_for('main.index'))
    if current_user.confirm(token):
        flash(u'?????????. ??!')
    else:
        flash(u'?????????')
    return redirect(url_for('main.index'))

def before_request():
    if current_user.is_authenticated:
        current_user.ping()
        if not current_user.confirmed and request.endpoint and request.endpoint[:5] != 'auth.' and request.endpoint != 'static':
            return redirect(url_for('auth.unconfirmed'))

def unconfirmed():
    if current_user.is_anonymous or current_user.confirmed:
        return redirect(url_for('main.index'))
    return render_template('auth/unconfirmed.html')



#????

def register():
    form = RegisterForm()
    if form.validate_on_submit():

        user_email = form.email_address.data
        user_email = user_email.lower()
        new_user = User.query.filter_by(email = user_email).first()

        if new_user is not None:
            token = generate_confirmation_token(new_user.email)
            confirm_url = url_for('main.confirm_reset', token = token, _external = True)
            html = render_template('reset_request.html', confirm_url = confirm_url)
            subject = 'InfoSec Mentors Project - Password Reset'
            send_email(app, new_user.email, subject, html)
            return render_template('waiting.html')

        # Add the user to the User table, instantiate other columns --------
        user = User(email = user_email,
                    password = form.password.data,
                    registered_on = datetime.datetime.now(),
                    confirmed = False)
        mentor = Mentor(email = user_email)
        suggested = Suggestion()
        userMetrics = UserMetrics()

        # Use a single row for the SiteMetrics table -----------------------
        siteMetrics = db.session.query(SiteMetrics.id).filter_by(id = 1).scalar()
        if siteMetrics is None:
            siteMetrics = SiteMetrics()
            db.session.add(siteMetrics)

        db.session.add(user)
        db.session.add(mentor)
        db.session.add(suggested)
        db.session.add(userMetrics)
        db.session.commit()

        # Generate the confirmation token, validate email ------------------
        token = generate_confirmation_token(user.email)
        confirm_url = url_for('main.confirm_email', token = token, _external = True)
        html = render_template('activate.html', confirm_url = confirm_url)
        subject = "InfoSec Mentors Project - Email Confirmation"
        send_email(app, user.email, subject, html)
        return render_template('waiting.html')

    # Bad Registration Code attempted --------------------------------------
    return render_template("register.html", form = form)

# ------------------------------------------------------------------------------
# Email Validation
# ------------------------------------------------------------------------------

def user_preferences(email):
    form = UserPreferences()
    user = User.query.filter_by(email = email).first()
    profile = Profile.query.filter_by(credentials_id = user.id).first()

    # Handle unconfirmed users -------------------------------------------------
    if user.confirmed_on is None:
        logout_user()
        return redirect(url_for('main.unconfirmed'))
    # Handle confirmed users with profiles -------------------------------------
    elif profile is not None:
        return redirect(url_for('main.user_profile', id = user.id))
    # Setup a profile ----------------------------------------------------------
    else:
        id = current_user.id
        if ( id == user.id ):
            if form.validate_on_submit():
                gravatar = form.gravatar.data
                gravatar_md5 = md5calc(gravatar)
                new_user = User.query.filter_by(email = email).first_or_404()
                new_profile = Profile(user = current_user,
                                      credentials_id = new_user.id,
                                      available = 1,
                                      gravatar = gravatar_md5,
                                      alias = form.alias.data,
                                      timezone = form.timezone.data,
                                      public_email = form.public_email.data,
                                      twitter = form.twitter.data,
                                      github = form.github.data,
                                      biography = '',
                                      experience = form.experience.data,
                                      mentoring_experience = form.mentoring_experience.data,
                                      mentored_experience = form.mentored_experience.data,
                                      contact_preference = form.contact_preference.data,
                                      mentoring_skill_0 = form.mentoring_skill_0.data,
                                      mentoring_skill_1 = form.mentoring_skill_1.data,
                                      mentoring_skill_2 = form.mentoring_skill_2.data,
                                      mentored_skill_0 = form.mentored_skill_0.data,
                                      mentored_skill_1 = form.mentored_skill_1.data,
                                      mentored_skill_2 = form.mentored_skill_2.data,
                                      mentoring_hours = form.mentoring_hours.data,
                                      mentoring_timeframe = form.mentoring_timeframe.data,
                                      mentored_hours = form.mentored_hours.data,
                                      mentored_timeframe = form.mentored_timeframe.data)
                db.session.add(new_profile)
                db.session.commit()
                return redirect(url_for('main.user_profile', id = new_profile.id))
            return render_template('user_preferences.html', email = email, form = form)

        # If a user tries to modify another person's preferences ---------------
        else:
            #TODO Problematic if Profile 1 is deleted
            return redirect (url_for('main.user_profile', id = 1))

# ------------------------------------------------------------------------------
# User Settings
# ------------------------------------------------------------------------------