# -*- coding: utf-8 -*- #!/usr/bin/env python from base import init_env, md5 import click init_env(interactive=True) from apps.web.management.models import Manager from apps.web.agent.models import Agent @click.group() def cli(): click.echo(u"欢迎进入创建管理员脚本") @cli.command() @click.option('--username', prompt=u'请输入用户名(手机号)') @click.option('--password', prompt=u'请输入密码') @click.option('--nickname', prompt=u'请输入名字') def create_manager(username, password, nickname): """创建系统管理员|厂商""" manager = Manager.create_user(username = username, password = md5(password), nickname = nickname) agent = Agent.create_user(username = username, password = md5(password), nickname = nickname, managerId = str(manager.id)) updated = manager.update(primeAgentId = str(agent.id)) assert updated, u'更新主代理商ID错误' click.echo(u'创建管理员|厂商成功,同时已生成并绑定同帐号同密码的厂商首席代理商') @cli.command() @click.option('--username', prompt=u'请输入用户名(手机号)') @click.option('--password', prompt=u'请输入将要重置的密码') def reset_manager_pwd(username, password): """重置管理员密码""" manager = Manager.objects.get(username=username) manager.set_password(md5(password)) click.echo('success!') if __name__ == '__main__': cli()