示例#1
0
文件: setup.py 项目: phille97/LWPCMS
def render():
    if get_option('initialized') is not None:
        return redirect('/')

    form = SetupForm(csrf_enabled=False)

    if form.validate_on_submit():
        site_name = form.site_name.data
        user_name = form.user_name.data
        password = form.password.data
        demo = form.demo.data

        set_option('site_name', site_name)
        set_option('site_demo', demo)
        set_option('site_filespage_limit', 128)
        set_option('site_postspage_limit', 128)

        register_user(
                    name=user_name,
                    password=password
                )

        set_option('initialized', 'True')

        return redirect('/')

    return render_template('setup.html', form=form)
示例#2
0
def render():
    if get_option('initialized') is not None:
        return redirect('/')

    form = SetupForm(csrf_enabled=False)

    if form.validate_on_submit():
        site_name = form.site_name.data
        user_name = form.user_name.data
        password = form.password.data
        demo = form.demo.data

        set_option('site_name', site_name)
        set_option('site_demo', demo, type='boolean')
        set_option('site_filespage_limit', 128, type='integer')
        set_option('site_postspage_limit', 128, type='integer')
        set_option('site_description', 'Your Description', type='textarea')
        set_option('site_tags', 'cms,site,website,blog', type='tags')

        register_user(
                    name=user_name,
                    password=password
                )

        set_option('initialized', 'True', type='boolean', editable=False)

        return redirect('/')

    return render_template('setup.html', form=form)
示例#3
0
def render_publish(id):
    sidenav = get_sidenav()

    form = UserForm(csrf_enabled=False)

    user = None
    msg = None

    if form.validate_on_submit():
        avatars = request.form.getlist('file_id')
        avatar = None
        if avatars is not None:
            if len(avatars) > 0:
                avatar = avatars[0]


        if avatar is not None:
            avatar_file = db.collections.find_one({'_id': ObjectId(avatar)})
        else:
            avatar_file = None

        if user_exists(form.user_name.data) and id is None:
            msg = 'User already exists'
            return render_template(
                    'editUser.html',
                    sidenav=sidenav,
                    form=form,
                    user=user,
                    id=id,
                    msg=msg
                    )

        new_user = register_user(
                name=form.user_name.data,
                password=form.password.data,
                avatar=avatar_file,
                id=id)

        if not id and new_user is not False:
            return redirect('/admin/edituser/{}'.format(new_user.inserted_id))

    if id:
        user = db.collections.find_one({"_id": ObjectId(id)})
        if user and request.method != 'POST':    
            form.user_name.data = user["nick_name"]
            form.password.data = user["password"]
        
    return render_template('editUser.html', sidenav=sidenav,
            form=form, user=user, id=id, msg=msg)
示例#4
0
def test_registration():
    nick = 'JohnDoe'
    user = register_user(name=nick, password='******')

    assert user != None

    db_user = db.collections.find(
          {'structure': '#User', 'nick_name': nick}
          )

    assert db_user != None

    db.collections.delete_many(
          {'structure': '#User', 'nick_name': nick}
          )