1234567891011121314151617181920212223242526272829303132333435363738394041 |
- # -*- coding: utf-8 -*-
- # !/usr/bin/env python
- from base import init_env, md5
- import click
- init_env(interactive = True)
- from apps.web.test.models import Tester
- @click.group()
- def cli():
- click.echo(u"welcome")
- @cli.command()
- @click.option('--username', prompt = 'please input user name(mobile number)')
- @click.option('--password', prompt = 'please input password')
- @click.option('--nickname', prompt = 'please input your name',
- help = u'create super manager')
- def create(username, password, nickname):
- """创建测试账号"""
- Tester.create_user(username = username, password = md5(password), nickname = nickname)
- click.echo('create success')
- @cli.command()
- @click.option('--username', prompt = 'please input user name(mobile number)')
- @click.option('--password', prompt = 'please input new password')
- def reset_pwd(username, password):
- """重置测试账号密码"""
- tester = Tester.objects.get(username = username)
- tester.set_password(md5(password))
- click.echo('reset success!')
- if __name__ == '__main__':
- cli()
|