Exemplo n.º 1
0
def userRegist():
    if current_user.is_authenticated:
        flash('Please logout before registe a new account')
        next = request.args.get('next')
        if next:
            return redirect(next)
        return redirect(url_for('index.indexPage'))
    else:
        form = RegisterForm()
        if form.validate_on_submit():
            user = User()
            user.username = form.username.data
            user.email = form.email.data
            user.password = generate_password_hash(form.password.data)
            user.sex = form.sex.data
            user.age = form.age.data
            user.school = form.school.data
            user.location = form.location.data
            user.createdTime = date.today()
            user.isactive = True
            user.isauthenticated = False

            print(user.username)
            user.save()

            login_user(user)
            return redirect(url_for('index.indexPage'))
        return render_template('login/registe.html', form=form)
Exemplo n.º 2
0
def userRegistWithCode(code):
    '''
    用户注册,有无邀请码来区分不同的组册情况
    '''
    if current_user.is_authenticated:
        flash('Please logout before registe a new account')
        next = request.args.get('next')
        if next:
            return redirect(next)
        return redirect(url_for('index.indexPage'))
    else:
        form = RegisterForm()
        if code:
            form.inviteCode.data = code
        if form.validate_on_submit():
            user = User()
            user.username = form.username.data
            user.email = form.email.data
            user.password = generate_password_hash(form.password.data)
            user.sex = form.sex.data
            user.age = form.age.data
            user.school = form.school.data
            user.location = form.location.data
            user.createdTime = date.today()
            user.isactive = True
            user.isauthenticated = False
            uniqueCode = form.inviteCode.data
            if uniqueCode:
                if Pair.query(bson.objectid.ObjectId(str(code))):
                    flash('This invite code has been used')
                    return redirect(url_for('login.userRegist'))
                user.save()
                login_user(user)
                if user.sex == 'male':
                    try:
                        pair = Pair()
                        pair.boy = user
                        pair.girl = User.query(
                            bson.objectid.ObjectId(str(code)))
                        pair.save()
                    except Exception as e:
                        print(str(e))
                else:
                    try:
                        pair = Pair()
                        pair.girl = user
                        pair.boy = User.query(
                            bson.objectid.ObjectId(str(code)))
                        pair.save()
                    except Exception as e:
                        print(str(e))
                return redirect(url_for('index.indexPage'))
            print(user.username)
            user.save()
            login_user(user)
            return redirect(url_for('index.indexPage'))
        return render_template('login/registe.html', form=form)