|
|
- from datetime import timedelta
- import time
-
- from django.core.management.base import BaseCommand
- from django.db.models import Q
- from django.utils import timezone
- from hc.accounts.models import Profile
- from hc.api.models import Check
-
-
- def num_pinged_checks(profile):
- q = Check.objects.filter(user_id=profile.user.id,)
- q = q.filter(last_ping__isnull=False)
- return q.count()
-
-
- class Command(BaseCommand):
- help = 'Send due monthly reports'
- tmpl = "Sending monthly report to %s"
-
- def add_arguments(self, parser):
- parser.add_argument(
- '--loop',
- action='store_true',
- dest='loop',
- default=False,
- help='Keep running indefinitely in a 300 second wait loop',
- )
-
- def handle_one_run(self):
- now = timezone.now()
- month_before = now - timedelta(days=30)
- month_after = now + timedelta(days=30)
-
- report_due = Q(next_report_date__lt=now)
- report_not_scheduled = Q(next_report_date__isnull=True)
-
- q = Profile.objects.filter(report_due | report_not_scheduled)
- q = q.filter(reports_allowed=True)
- q = q.filter(user__date_joined__lt=month_before)
- profiles = list(q)
-
- sent = 0
- for profile in profiles:
- qq = Profile.objects
- qq = qq.filter(id=profile.id,
- next_report_date=profile.next_report_date)
-
- num_updated = qq.update(next_report_date=month_after)
- if num_updated != 1:
- # Was updated elsewhere, skipping
- continue
-
- if num_pinged_checks(profile) == 0:
- continue
-
- self.stdout.write(self.tmpl % profile.user.email)
- profile.send_report()
- sent += 1
-
- return sent
-
- def handle(self, *args, **options):
- if not options["loop"]:
- return "Sent %d reports" % self.handle_one_run()
-
- self.stdout.write("sendreports is now running")
- while True:
- self.handle_one_run()
-
- formatted = timezone.now().isoformat()
- self.stdout.write("-- MARK %s --" % formatted)
-
- time.sleep(300)
|
