manyi/script/supermanager.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python


from base import init_env, md5

import click

init_env(interactive = True)

from apps.web.superadmin.models import SuperManager

@click.group()
def cli():
    click.echo(u"welcome")


@cli.command()
@click.option('--username', prompt = 'please input user name(mobile number)')
@click.option('--password', prompt = 'please input password')
@click.option('--nickname', prompt = 'please input your name',
              help = u'create super manager')

def create(username, password, nickname):
    """创建超级系统管理员"""
    SuperManager.create_user(username = username, password = md5(password), nickname = nickname)
    click.echo('create success')


@cli.command()
@click.option('--username', prompt = 'please input user name(mobile number)')
@click.option('--password', prompt = 'please input new password')
def reset_pwd(username, password):
    """重置超级管理员密码"""
    manager = SuperManager.objects.get(username = username)
    manager.set_password(md5(password))
    click.echo('reset success!')


if __name__ == '__main__':
    cli()