# -*- 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 Dealer @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('--username', prompt = u'请输入用户名(手机号)') @click.option('--password', prompt = u'请输入将要重置的密码') def reset_dealer_pwd(username, password): """重置代理商密码""" dealer = Dealer.objects.get(username = username) dealer.set_password(md5(password)) click.echo('success!') @cli.command() @click.option('--username', prompt = u'请输入用户名(手机号)') @click.option('--mid', prompt = u'请输入代理商ID', help = u'代理商ID') def forbit_user(username, mid): try: dealer = Dealer.objects(username = username, agentId = mid).get() # type: Dealer setattr(dealer, 'status', 'forbidden') dealer.save() click.echo('success!') except Exception as e: click.echo('dealer is not exist') if __name__ == '__main__': cli()