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