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"))
def dropdb():
db.drop_all()
flash("drop database ok")
return redirect(url_for("index"))
