manyi/script/tester.py

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