manager.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  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.management.models import Manager
  7. from apps.web.agent.models import Agent
  8. @click.group()
  9. def cli():
  10. click.echo(u"欢迎进入创建管理员脚本")
  11. @cli.command()
  12. @click.option('--username', prompt=u'请输入用户名(手机号)')
  13. @click.option('--password', prompt=u'请输入密码')
  14. @click.option('--nickname', prompt=u'请输入名字')
  15. def create_manager(username, password, nickname):
  16. """创建系统管理员|厂商"""
  17. manager = Manager.create_user(username = username, password = md5(password), nickname = nickname)
  18. agent = Agent.create_user(username = username, password = md5(password), nickname = nickname,
  19. managerId = str(manager.id))
  20. updated = manager.update(primeAgentId = str(agent.id))
  21. assert updated, u'更新主代理商ID错误'
  22. click.echo(u'创建管理员|厂商成功,同时已生成并绑定同帐号同密码的厂商首席代理商')
  23. @cli.command()
  24. @click.option('--username', prompt=u'请输入用户名(手机号)')
  25. @click.option('--password', prompt=u'请输入将要重置的密码')
  26. def reset_manager_pwd(username, password):
  27. """重置管理员密码"""
  28. manager = Manager.objects.get(username=username)
  29. manager.set_password(md5(password))
  30. click.echo('success!')
  31. if __name__ == '__main__':
  32. cli()