manyi/script/agent.py

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

from base import init_env, md5

import click

init_env(interactive=True)

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'请输入名字',
              help=u'代理商')
@click.option('--mid', prompt=u'请输入厂商ID',
              help=u'厂商ID')
def create_agent(username, password, nickname, mid):
    """创建代理商"""
    Agent.create_user(username = username, password = md5(password), nickname = nickname, **{'managerId': mid})
    click.echo('success!')

@cli.command()
@click.option('--username', prompt=u'请输入用户名(手机号)')
@click.option('--password', prompt=u'请输入将要重置的密码')
def reset_agent_pwd(username, password):
    """重置代理商密码"""
    agent = Agent.objects.get(username=username)
    agent.set_password(md5(password))
    click.echo('success!')

if __name__ == '__main__':
    cli()