You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.5 KiB
113 lines
3.5 KiB
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 NO_NAG, Profile
|
|
from hc.api.models import Check
|
|
from hc.lib.date import choose_next_report_date
|
|
|
|
|
|
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 and nags"
|
|
tmpl = "Sent monthly report to %s"
|
|
|
|
def pause(self):
|
|
time.sleep(1)
|
|
|
|
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_monthly_report(self):
|
|
report_due = Q(next_report_date__lt=timezone.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)
|
|
profile = q.first()
|
|
|
|
if profile is None:
|
|
# No matching profiles found – nothing to do right now.
|
|
return False
|
|
|
|
# A sort of optimistic lock. Will try to update next_report_date,
|
|
# and if does get modified, we're in drivers seat:
|
|
qq = Profile.objects.filter(
|
|
id=profile.id, next_report_date=profile.next_report_date
|
|
)
|
|
|
|
# Next report date is currently not scheduled: schedule it and move on.
|
|
if profile.next_report_date is None:
|
|
qq.update(next_report_date=choose_next_report_date())
|
|
return True
|
|
|
|
num_updated = qq.update(next_report_date=choose_next_report_date())
|
|
if num_updated != 1:
|
|
# next_report_date was already updated elsewhere, skipping
|
|
return True
|
|
|
|
if profile.send_report():
|
|
self.stdout.write(self.tmpl % profile.user.email)
|
|
# Pause before next report to avoid hitting sending quota
|
|
self.pause()
|
|
|
|
return True
|
|
|
|
def handle_one_nag(self):
|
|
now = timezone.now()
|
|
q = Profile.objects.filter(next_nag_date__lt=now)
|
|
q = q.exclude(nag_period=NO_NAG)
|
|
profile = q.first()
|
|
|
|
if profile is None:
|
|
return False
|
|
|
|
qq = Profile.objects.filter(id=profile.id, next_nag_date=profile.next_nag_date)
|
|
|
|
num_updated = qq.update(next_nag_date=now + profile.nag_period)
|
|
if num_updated != 1:
|
|
# next_rag_date was already updated elsewhere, skipping
|
|
return True
|
|
|
|
if profile.send_report(nag=True):
|
|
self.stdout.write("Sent nag to %s" % profile.user.email)
|
|
# Pause before next report to avoid hitting sending quota
|
|
self.pause()
|
|
else:
|
|
profile.next_nag_date = None
|
|
profile.save()
|
|
|
|
return True
|
|
|
|
def handle(self, *args, **options):
|
|
self.stdout.write("sendreports is now running")
|
|
while True:
|
|
# Monthly reports
|
|
while self.handle_one_monthly_report():
|
|
pass
|
|
|
|
# Daily and hourly nags
|
|
while self.handle_one_nag():
|
|
pass
|
|
|
|
if not options["loop"]:
|
|
break
|
|
|
|
formatted = timezone.now().isoformat()
|
|
self.stdout.write("-- MARK %s --" % formatted)
|
|
|
|
# Sleep for 1 minute before looking for more work
|
|
time.sleep(60)
|