def setUp(self): melange.database.drop_db() melange.database.init_db() user = User('api') user.password = '******' user.save()
def list_users(): if request.method == 'POST': if 'add-user' in request.form: user_name = request.form['user-name'] user_password = request.form['user-password'] user = User.find(user_name) if not user: user = User(user_name) user.password = user_password user.save() users = User.find_all() return render_template('users.html', users=users)
import os from optparse import OptionParser parser = OptionParser() parser.add_option('-i', '--initdb', default=None, action='store_true', dest='initdb') parser.add_option('-d', '--dropdb', default=None, action='store_true', dest='dropdb') options, args = parser.parse_args() if options.initdb: from melange.database import init_db print('Initializing database') init_db() from melange import User admin = User('admin') admin.password = '******' admin.save() elif options.dropdb: from melange.database import drop_db print('Dropping database') drop_db() else: from melange import app app.run()