Remove stale disk usage stats
This commit is contained in:
parent
4b74f5d04b
commit
3b6d50a62a
2 changed files with 19 additions and 4 deletions
|
@ -1,6 +1,8 @@
|
|||
Changelog
|
||||
=========
|
||||
|
||||
* :bug:`-` remove stale disk usage stats older than 30 minutes
|
||||
|
||||
* :release:`0.15.0 <2023-07-23>
|
||||
* :feature:`10` add disk usage details for mail and web
|
||||
|
||||
|
|
|
@ -6,6 +6,7 @@ from __future__ import absolute_import
|
|||
|
||||
import http
|
||||
import logging
|
||||
from datetime import timedelta
|
||||
|
||||
from django.conf import settings
|
||||
from django.contrib import messages
|
||||
|
@ -13,6 +14,7 @@ from django.contrib.auth import get_user_model
|
|||
from django.contrib.auth.mixins import PermissionRequiredMixin, UserPassesTestMixin
|
||||
from django.http import Http404
|
||||
from django.shortcuts import get_object_or_404, redirect
|
||||
from django.utils import timezone
|
||||
from django.utils.translation import gettext as _
|
||||
from django.views.generic import DetailView, ListView
|
||||
from django.views.generic.edit import CreateView, FormView
|
||||
|
@ -292,12 +294,15 @@ class UploadCustomerPackageDiskUsage(APIView):
|
|||
if request.content_type != "application/json":
|
||||
return Response("Unacceptable", status=http.HTTPStatus.BAD_REQUEST)
|
||||
|
||||
submitted_sources = set()
|
||||
|
||||
for row in request.data:
|
||||
user = row["user"]
|
||||
for key in row:
|
||||
if key == "user":
|
||||
continue
|
||||
else:
|
||||
submitted_sources.add(key)
|
||||
for item, size in row[key].items():
|
||||
try:
|
||||
package = CustomerHostingPackage.objects.get(
|
||||
|
@ -315,7 +320,9 @@ class UploadCustomerPackageDiskUsage(APIView):
|
|||
|
||||
if key == "mail":
|
||||
try:
|
||||
ma_mb = package.mailboxes.get(username=item).mailaddressmailbox_set.first()
|
||||
ma_mb = package.mailboxes.get(
|
||||
username=item
|
||||
).mailaddressmailbox_set.first()
|
||||
if ma_mb:
|
||||
metric.email_address_id = ma_mb.mailaddress_id
|
||||
except Mailbox.DoesNotExist:
|
||||
|
@ -327,6 +334,12 @@ class UploadCustomerPackageDiskUsage(APIView):
|
|||
"hosting package for user %s does not exist", user
|
||||
)
|
||||
|
||||
if submitted_sources:
|
||||
CustomerPackageDiskUsage.objects.filter(
|
||||
source__in=submitted_sources,
|
||||
modified__lt=timezone.now() - timedelta(minutes=30),
|
||||
).delete()
|
||||
|
||||
logger.info("usage data submitted by %s", request.user)
|
||||
|
||||
return Response("Accepted", status=http.HTTPStatus.ACCEPTED)
|
||||
|
|
Loading…
Reference in a new issue