Example #1
0
def register():
    """Registers the user."""
    form = RegisterForm(request.form)
    if not (current_user.is_anonymous()):
        flash("you are logined")
        return render_template("logout.html")
    if request.method == "POST" and form.validate_on_submit():
        username = form.username.data
        password = form.password.data
        password2 = form.password2.data
        email = form.email.data
        if User.query.filter_by(username=username).first():
            flash("The username is already taken")
        elif cmp(password, password2) != 0:
            flash("password mismatched")
        else:
            user = User(username, None, email)
            user.set_password(password)
            try:
                user.store_to_db()
                return render_template("account/registerok.html")
            except:
                flash("You were register failed, pls contact %s for help." % app.config["ADMIN"][1])
    else:
        return render_template("account/register.html", form=form)
    return render_template("account/register.html", form=form)
Example #2
0
def adduser():
    add_form = AddUserForm(request.form)
    if request.method == "POST" and add_form.validate_on_submit():
        username = add_form.username.data
        password = add_form.password.data
        email = add_form.email.data
        if User.query.filter_by(username=username).first():
            flash("The username is already taken")
        hash_pass = generate_password_hash(password, method="sha1", salt_length=8)
        user = User(username, hash_pass, email)
        try:
            user.store_to_db()
            flash("add user %s done." % user.username)
        except:
            flash("add user failed %s" % user.username, "error")
    return render_template("admin/adduser.html", add_form=add_form)
Example #3
0
def initdb():
    try:
        db.create_all()
    except:
        db.drop_all()
    # add some records
    try:
        admin_pass = generate_password_hash("admin", method="sha1", salt_length=8)
        guest_pass = generate_password_hash("guest", method="sha1", salt_length=8)
        admin = User("admin", admin_pass, "*****@*****.**")
        guest = User("guest", guest_pass, "*****@*****.**")
        # save them into database
        admin.store_to_db()
        flash("user admin added")
        guest.store_to_db()
        flash("user guest added")
    except:
        flash("create User table failed.", "error")

    try:
        todo1 = Todo("check in code", "modify reset")
        todo2 = Todo("talk with somebody", "about cloud")
        todo1.store_to_db()
        flash("todo #1 added")
        todo2.store_to_db()
        flash("todo #2 added")
    except:
        flash("create Todo table failed.", "error")

    try:
        wiki1 = Wiki("MainPage", "wikitest")
        wiki2 = Wiki("Sandbox", "wikitest")
        wiki1.store_to_db()
        flash("wiki #1 added")
        wiki2.store_to_db()
        flash("wiki #2 added")
    except:
        flash("create Wiki table failed.", "error")
    return redirect(url_for("index"))