manyi/script/manager.py

# -*- 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()