示例#1
0
文件: views.py 项目: xiaoabiang/MyWeb
def login():
    if g.user is not None and g.user.is_authenticated:
        return redirect(url_for('index'))
    form = LoginForm()
    if form.validate_on_submit():
        user = User()
        user.name = form.name.data
        user.passwd = form.password.data.strip()
        users = User.find_all('name=?', [user.name])
        if users is None or len(users) == 0:
            log(level=1, msg="用户名不存在!")
            return redirect(url_for('login'))
        elif not check_password_hash(users[0].passwd, user.passwd):
            log(level=1, msg="密码错误!")
            return redirect(url_for('login'))
        login_user(users[0], remember=True)
        return redirect(request.args.get('next') or url_for('index'))
    return render_template('login.html', title='Sign in', form=form)
示例#2
0
文件: views.py 项目: xiaoabiang/MyWeb
def user(id):
    user = User.find(id)
    form = UserInfoEditForm()
    if len(user) == 0:
        flash('User not found')
        return redirect(url_for('index'))
    if form.validate_on_submit():
        user.name = form.name.data
        user.email = form.name.email
        user.update()
        return redirect(url_for('index'))
    return render_template('user.html',
                           user=user,
                           form=form)
示例#3
0
def create_user(user_to_create):
    session = Session()

    try:
        session.query(User).filter_by(email=user_to_create.get('email')).one()
    except NoResultFound:
        password = generate_password_hash(user_to_create.get('password'))
        user = User(email=user_to_create.get('email'), username=user_to_create.get('username'), password=password)
        session.add(user)
        session.commit()
        session.refresh(user)

        return UserData().dump(user)
    else:
        raise ValidationError("Email must be unique")
示例#4
0
文件: views.py 项目: xiaoabiang/MyWeb
def confirm(id):
    if g.user is not None and g.user.is_authenticated:
        return redirect(url_for('index'))
    user = User.find(id)
    if not user:
        return redirect(url_for('register'))
    elif int(time.time()) - user.created_at > 3600:
        user.remove()
        return redirect(url_for('register'))
    elif user.id != id:
        return redirect(url_for('register'))
    # 验证正确,将数据保存到数据库
    user.isconfirmed = True
    user.update()

    login_user(user, remember=True)
    return redirect(url_for('index'))
示例#5
0
    def test_models(self):
        from flask_bcrypt import generate_password_hash
        user = User(username='******',
                    firstName='admin',
                    lastName='admin',
                    email='*****@*****.**',
                    phone='+38099',
                    userAuthStatus='notSignedIn',
                    password=generate_password_hash('admin'))
        self.assertEqual(
            str(user),
            'User None, terminator2000 (admin admin) [email protected] +38099 status: notSignedIn'
        )

        wallet = Wallet(name='MyFirstWallet',
                        balance=10000,
                        currency='USD',
                        owner=user)
        self.assertEqual(
            str(wallet),
            f'Wallet {wallet.id} {wallet.name}, owner: {wallet.user_id}, balance: {wallet.balance} {wallet.currency}'
        )
示例#6
0
文件: views.py 项目: xiaoabiang/MyWeb
def register():
    if g.user is not None and g.user.is_authenticated:
        return redirect(url_for('index'))
    form = RegisterForm()
    if form.validate_on_submit():
        user = User()
        user.name = form.name.data
        user.email = form.email.data
        user.isconfirmed = False
        user.image = "/static/img/user.png"
        user.passwd = generate_password_hash(form.password.data.strip())
        rows = user.save()
        if rows != 1:
            return redirect(url_for('register'))
        else:
            send_email(to=user.email, subject='确认注册信息!', template='/email/confirm', user=user, id=user.id)
            # login_user(user, remember=True)
            return redirect(url_for('index'))
    return render_template('register.html',
                           title='Sign in',
                           form=form,
                           providers=app.config['OPENID_PROVIDERS'])
示例#7
0
def create_objects():
    session = Session()

    user = User(username='******',
                firstName='admin',
                lastName='admin',
                email='*****@*****.**',
                phone='+38099',
                userAuthStatus=UserStatus.notSignedIn,
                password=generate_password_hash('admin'))

    user2 = User(username='******',
                 firstName='admin',
                 lastName='admin',
                 email='*****@*****.**',
                 phone='+380990',
                 userAuthStatus=UserStatus.notSignedIn,
                 password=generate_password_hash('admin'))

    user3 = User(username='******',
                 firstName='admin',
                 lastName='admin',
                 email='*****@*****.**',
                 phone='+380940',
                 userAuthStatus=UserStatus.notSignedIn,
                 password=generate_password_hash('admin'))

    wallet13 = Wallet(
        name='MyFirstWallet',
        balance=10000,
        currency=Currency.USD,
        owner=user3,
    )

    wallet23 = Wallet(
        name='MySecondWallet',
        balance=100,
        currency=Currency.UAH,
        owner=user3,
    )

    wallet1 = Wallet(
        name='MyWallet',
        balance=10000,
        currency=Currency.USD,
        owner=user,
    )

    wallet2 = Wallet(
        name='MySecoWallet',
        balance=100,
        currency=Currency.UAH,
        owner=user,
    )

    session.add(user)
    session.add(user2)
    session.add(wallet1)
    session.add(wallet2)

    session.add(user3)
    session.add(wallet13)
    session.add(wallet23)

    session.commit()
示例#8
0
文件: views.py 项目: xiaoabiang/MyWeb
def load_user(id):
    return User.find(id)
示例#9
0
文件: views.py 项目: xiaoabiang/MyWeb
def userview(id):
    user = User.find(id)
    if g.user is not None and g.user.is_authenticated:
        return render_template('userview.html', userview=user, user=g.user)
    return render_template('userview.html', userview=user)