summaryrefslogtreecommitdiff
path: root/ishtar_common/management/commands/gdpr_clean.py
blob: c53d24fcc24a4b1d56f2ed7162a1c3b9b0b1a43d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import datetime
from django.conf import settings
from django.core.management.base import BaseCommand
from django.utils import timezone
import sys

from ishtar_common.models import GDPRLog, GDPRPerson
from ishtar_common.utils import BColors


def get_time():
    return datetime.datetime.now().isoformat().split(".")[0]


def clean_gdpr():
    date = timezone.now() - (
        datetime.timedelta(days=settings.GDPR_RETENTION_PERIOD)
    )
    q = GDPRLog.objects.filter(date__lte=date)
    nb = q.count()
    q.delete()
    q = GDPRPerson.objects.filter(logs=None)
    q.delete()
    return nb


class Command(BaseCommand):
    help = "Cleaning up obsolete gdpr data"

    def add_arguments(self, parser):
        parser.add_argument(
            "--quiet", dest="quiet", action="store_true", help="Quiet output"
        )

    def handle(self, *args, **options):
        quiet = options["quiet"]
        if not quiet:
            msg = BColors.format("HEADER", f"[{get_time()}] Processing DGPR cleansing\n")
            sys.stdout.write(msg)
        nb = clean_gdpr()
        if not quiet:
            sys.stdout.write(f"{nb} dgpr logs cleaned\n")
        sys.exit(1)