scripts.py

from flask.ext.script import Command, Option
from myapp.models import db, Things, User

class AddUser(Command):
    "add a user"

    def get_options(self):
        return [
            Option('-n', '--name', dest='name'),
            Option('-p', '--password', dest='password'),
            Option('-r', '--role', dest='role', default=0)
        ]

    def run(self, name, password, role):
        if not name:
            print "missing name"
            return
        if not password:
            print "missing password"
            return
        user = User(name, password)
        print "username: ", name
        if role:
            role = int(role)
            if role not in [0, 1]:
                print "role must be 0 or 1"
                return
            user.role = role
            print "role: ", role
        if password: print "password: ", password

        db.session.add(user)
        db.session.commit()