forked from GithubBackups/healthchecks
1e3285423f
Check.n_pings now stores the total number of pings the check has ever received. Running "prunepings" command doesn't affect this field. +a new "prunepingsslow" command which works in smaller chunks so is appropriate for initial pruning of a huge api_ping table.
23 lines
740 B
Python
23 lines
740 B
Python
from django.db.models import F
|
|
from django.contrib.auth.models import User
|
|
from django.core.management.base import BaseCommand
|
|
from hc.accounts.models import Profile
|
|
from hc.api.models import Ping
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Prune pings based on limits in user profiles'
|
|
|
|
def handle(self, *args, **options):
|
|
# Create any missing user profiles
|
|
for user in User.objects.filter(profile=None):
|
|
Profile.objects.for_user(user)
|
|
|
|
q = Ping.objects
|
|
q = q.annotate(limit=F("owner__user__profile__ping_log_limit"))
|
|
q = q.filter(n__lt=F("owner__n_pings") - F("limit"))
|
|
q = q.filter(n__gt=0)
|
|
n_pruned, _ = q.delete()
|
|
|
|
return "Done! Pruned %d pings" % n_pruned
|