1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- # -*- coding: utf-8 -*-
- # !/usr/bin/env python
- from base import init_env, md5
- import click
- init_env(interactive = True)
- from apps.web.dealer.models import SubAccount
- @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_dealer(username, password, nickname, mid):
- # """创建经销商"""
- # Dealer.create_user(username = username, password = md5(password), nickname = nickname, **{'agentId': mid})
- # click.echo('success!')
- @cli.command()
- @click.option('--bossid', prompt = u'请输入经销商ID')
- @click.option('--username', prompt = u'请输入用户名(手机号)')
- @click.option('--password', prompt = u'请输入将要重置的密码')
- def reset_pwd(bossid, username, password):
- sub_account = SubAccount.objects(username = username, bossId = bossid).get()
- sub_account.set_password(md5(password))
- click.echo('success!')
- @cli.command()
- @click.option('--bossid', prompt = u'请输入经销商ID')
- @click.option('--username', prompt = u'请输入用户名(手机号)')
- def unlock_login(bossid, username):
- sub_account = SubAccount.objects(username = username, bossId = bossid).get()
- sub_account.unlock_login()
- click.echo('success!')
- if __name__ == '__main__':
- cli()
|