manyi/script/ad/count_ad_noshow.py

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

import datetime
import os

from script.base import init_env

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "configs.testing")

init_env(interactive = True)

from apps.web.user.models import ConsumeRecord
from apps.web.dealer.models import Dealer
from apps.web.device.models import Group

startTs = datetime.datetime(2021, 12, 8, 0, 0, 0)
endTs = datetime.datetime(2021, 12, 9, 0, 0, 0)

total = 0

dealers = [dealer for dealer in Dealer.objects(adShow = False).only('id')]
for dealer in dealers:
    group_ids = [str(group.id) for group in Group.objects(ownerId = str(dealer.id)).only('id')]

    count = ConsumeRecord.objects(groupId__in = group_ids, dateTimeAdded__gte = startTs, dateTimeAdded__lt = endTs).count()
    print('{} no show {}.'.format(str(dealer.id), count))

    total += count

print 'total: {}'.format(total)