manyi/script/subaccount.py

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