45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils.timezone import now
|
|
from apiserver.api import models, utils, utils_stats
|
|
|
|
import time
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Tasks to run on the portal hourly.'
|
|
|
|
def generate_stats(self):
|
|
utils_stats.calc_next_events()
|
|
counts = utils_stats.calc_member_counts()
|
|
signup_count = utils_stats.calc_signup_counts()
|
|
|
|
# do this hourly in case an admin causes a change
|
|
models.StatsMemberCount.objects.update_or_create(
|
|
date=utils.today_alberta_tz(),
|
|
defaults=dict(
|
|
member_count=counts['member_count'],
|
|
green_count=counts['green_count'],
|
|
six_month_plus_count=counts['six_month_plus_count'],
|
|
vetted_count=counts['vetted_count'],
|
|
subscriber_count=counts['subscriber_count'],
|
|
),
|
|
)
|
|
|
|
models.StatsSignupCount.objects.update_or_create(
|
|
month=utils.today_alberta_tz().replace(day=1),
|
|
defaults=dict(signup_count=signup_count),
|
|
)
|
|
|
|
utils_stats.calc_card_scans()
|
|
|
|
|
|
def handle(self, *args, **options):
|
|
self.stdout.write('{} - Beginning hourly tasks'.format(str(now())))
|
|
start = time.time()
|
|
|
|
self.generate_stats()
|
|
self.stdout.write('Generated stats')
|
|
|
|
self.stdout.write('Completed tasks in {} s'.format(
|
|
str(time.time() - start)[:4]
|
|
))
|