manyi/script/dealer.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 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()