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)
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)
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"))