manage.py 文件源码

def create_admin(username):
    from eventit.eventit import db
    from eventit.models import User, Role
    import getpass
    from sys import stdout

    password = None
    password2 = None
    while not password or password != password2:
        password = getpass.getpass()
        password2 = getpass.getpass('Please, repeat your password: ')
        if not password or password != password2:
            stdout.write('Passwords do not match')

    user = User(username=username, email='', is_active=True, password=password)
    role_admin = Role.get_role_obj('admin')
    user.role = role_admin

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