manyi/script/lige.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

import os

from base import init_env
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "configs.production")

init_env(interactive = False)

from apps.web.device.models import Group
from apps.web.agent.models import Agent
from apps.web.dealer.models import Dealer
from apps.web.management.models import Manager
from apps.web.user.models import MyUser, Card

agent = Agent.objects(username = '15801750954').get()

manager = Manager.objects(primeAgentId = str(agent.id)).get()

agent_id_list = [str(agent.id) for agent in Agent.objects(managerId = str(manager.id)).all()]
print('has {} agents.'.format(len(agent_id_list)))

dealer_id_list = [str(dealer.id) for dealer in Dealer.objects(agentId__in = agent_id_list)]
print('has {} dealers.'.format(len(dealer_id_list)))

card_id_list1 = [str(card.id) for card in Card.objects(dealerId__in = dealer_id_list)]
card_id_list2 = [str(card.id) for card in Card.objects(agentId__in = agent_id_list)]

card_id_list = set(card_id_list1) | set(card_id_list2)

print('has {} cards'.format(len(card_id_list)))

group_id_list = [str(group.id) for group in Group.objects(ownerId__in = dealer_id_list).all()]
print('has {} groups'.format(len(group_id_list)))

users = [user for user in MyUser.objects(groupId__in = group_id_list, balance__gt = 0).all()]
print('has {} users'.format(len(users)))

for user in users:  # type: MyUser
    print('{} has {}'.format(repr(user), user.balance))