mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2025-06-29 00:35:45 +08:00
Added pruning command
This commit is contained in:
parent
c8fca412b8
commit
13ba0cb12a
3 changed files with 89 additions and 4 deletions
34
stude/accounts/management/commands/clean_old_entries.py
Normal file
34
stude/accounts/management/commands/clean_old_entries.py
Normal file
|
@ -0,0 +1,34 @@
|
|||
from django.core.management.base import BaseCommand
|
||||
from django.utils import timezone
|
||||
from django.contrib.gis.geos import Point
|
||||
from study_groups.models import StudyGroup
|
||||
from student_status.models import StudentStatus
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'Removes old study groups and resets old student statuses'
|
||||
|
||||
def handle(self, *args, **kwargs):
|
||||
# Get the current time
|
||||
now = timezone.now()
|
||||
|
||||
# Get the time 8 hours ago
|
||||
time_threshold = now - timezone.timedelta(hours=8)
|
||||
|
||||
# Delete StudyGroup entries older than 8 hours
|
||||
StudyGroup.objects.filter(timestamp__lt=time_threshold).delete()
|
||||
|
||||
# Get StudentStatus entries older than 8 hours
|
||||
old_statuses = StudentStatus.objects.filter(
|
||||
timestamp__lt=time_threshold)
|
||||
|
||||
# Set the fields of the old statuses to the required values
|
||||
for status in old_statuses:
|
||||
status.location = Point(0, 0)
|
||||
status.subject = None
|
||||
status.landmark = None
|
||||
status.study_group = None
|
||||
status.save()
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(
|
||||
'Successfully cleaned old entries'))
|
Loading…
Add table
Add a link
Reference in a new issue