subaccount.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. # -*- coding: utf-8 -*-
  2. # !/usr/bin/env python
  3. from base import init_env, md5
  4. import click
  5. init_env(interactive = True)
  6. from apps.web.dealer.models import SubAccount
  7. @click.group()
  8. def cli():
  9. click.echo(u"欢迎进入创建经销商脚本")
  10. # @cli.command()
  11. # @click.option('--username', prompt = u'请输入用户名(手机号)')
  12. # @click.option('--password', prompt = u'请输入密码')
  13. # @click.option('--nickname', prompt = u'请输入名字',
  14. # help = u'经销商')
  15. # @click.option('--mid', prompt = u'请输入代理商ID',
  16. # help = u'代理商ID')
  17. # def create_dealer(username, password, nickname, mid):
  18. # """创建经销商"""
  19. # Dealer.create_user(username = username, password = md5(password), nickname = nickname, **{'agentId': mid})
  20. # click.echo('success!')
  21. @cli.command()
  22. @click.option('--bossid', prompt = u'请输入经销商ID')
  23. @click.option('--username', prompt = u'请输入用户名(手机号)')
  24. @click.option('--password', prompt = u'请输入将要重置的密码')
  25. def reset_pwd(bossid, username, password):
  26. sub_account = SubAccount.objects(username = username, bossId = bossid).get()
  27. sub_account.set_password(md5(password))
  28. click.echo('success!')
  29. @cli.command()
  30. @click.option('--bossid', prompt = u'请输入经销商ID')
  31. @click.option('--username', prompt = u'请输入用户名(手机号)')
  32. def unlock_login(bossid, username):
  33. sub_account = SubAccount.objects(username = username, bossId = bossid).get()
  34. sub_account.unlock_login()
  35. click.echo('success!')
  36. if __name__ == '__main__':
  37. cli()