manyi/script/user_operation/close_charge_switch.py

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

import os,sys

PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
sys.path.insert(0, PROJECT_ROOT)

os.environ.update({"DJANGO_SETTINGS_MODULE": "configs.production"})
import django
django.setup()

from apps.web.dealer.models import Dealer
from apps.web.device.models import Device
from apps.web.device.models import Group
from apps.web.agent.models import Agent

agentBlackList = ['15888766630','13966711825','13812791529']
dealerBlackList = ['13827102928','18674568321','13930448219','15854515805','13837145142','13838521870','13812791529','13053339762','13968093002','19916092108','19916092108','13837145142','15915611166','15359713511','13122516396','13705625096','15896595915']

#黑名单中的关闭掉
# agentIds = [str(agent.id) for agent in Agent.objects.filter(username__in = agentBlackList)]
# dealers = [dealer for dealer in Dealer.objects.filter(agentId__in = agentIds).only('id')]
# dealers2 = [dealer for dealer in Dealer.objects.filter(username__in = dealerBlackList).only('id')]
# dealers.extend(dealers2)

# agentIds = [str(agent.id) for agent in Agent.objects.filter(username__in = ['13930448219'])]
# dealers = [dealer for dealer in Dealer.objects.filter(agentId__in = agentIds).only('id')]
dealers = [dealer for dealer in Dealer.objects.filter(username__in = ['13837145142'])]
count = 0
dealerIds = []
for dealer in dealers:
    devCount = Device.objects.filter(ownerId = str(dealer.id)).count()
    if devCount > 10:
        continue
    dealerIds.append(str(dealer.id))

print 'dealer id num',len(dealerIds)
    
groupIds = [str(_.id) for _ in Group.objects.filter(ownerId__in = dealerIds)]
 
for groupId in groupIds:
    Group.update_group(groupId,beforeChargeUnpay = False)
    
    
print 'finished'