Draft of a gdpr export command
This commit is contained in:
Théophile Bastian
parent
1c880b265e
commit
e9bb3999a0
1 changed files with 208 additions and 0 deletions
208
kfet/management/commands/gdprexport.py
Normal file
208
kfet/management/commands/gdprexport.py
Normal file
|
|
@ -0,0 +1,208 @@
|
||||||
|
import sys
|
||||||
|
import typing as ty
|
||||||
|
|
||||||
|
from django.core.management.base import BaseCommand, CommandError
|
||||||
|
from django.db.models import Prefetch, Q
|
||||||
|
|
||||||
|
from kfet.models import (
|
||||||
|
Account,
|
||||||
|
AccountNegative,
|
||||||
|
Operation,
|
||||||
|
OperationGroup,
|
||||||
|
Transfer,
|
||||||
|
TransferGroup,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
help = "Exports all personal user data for a given trigramme"
|
||||||
|
|
||||||
|
def add_arguments(self, parser):
|
||||||
|
parser.add_argument("trigramme", help="the trigramme to)export")
|
||||||
|
parser.add_argument(
|
||||||
|
"-o",
|
||||||
|
"--output",
|
||||||
|
default="-",
|
||||||
|
help="File in which the results should be written (defaults to stdout)",
|
||||||
|
)
|
||||||
|
|
||||||
|
def _do_handle_history(self, account: Account, stream: ty.TextIO):
|
||||||
|
transfer_queryset_prefetch = Transfer.objects.select_related(
|
||||||
|
"from_acc", "to_acc", "canceled_by"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Le check sur les comptes est dans le prefetch pour les transferts
|
||||||
|
transfer_queryset_prefetch = transfer_queryset_prefetch.filter(
|
||||||
|
Q(from_acc=account) | Q(to_acc=account)
|
||||||
|
)
|
||||||
|
|
||||||
|
transfer_prefetch = Prefetch(
|
||||||
|
"transfers",
|
||||||
|
queryset=transfer_queryset_prefetch,
|
||||||
|
to_attr="filtered_transfers",
|
||||||
|
)
|
||||||
|
|
||||||
|
# Construction de la requête (sur les opérations) pour le prefetch
|
||||||
|
ope_queryset_prefetch = Operation.objects.select_related(
|
||||||
|
"article", "canceled_by", "addcost_for"
|
||||||
|
)
|
||||||
|
ope_prefetch = Prefetch("opes", queryset=ope_queryset_prefetch)
|
||||||
|
|
||||||
|
# Construction de la requête principale
|
||||||
|
opegroups = (
|
||||||
|
OperationGroup.objects.prefetch_related(ope_prefetch)
|
||||||
|
.select_related("on_acc", "valid_by")
|
||||||
|
.filter(on_acc=account)
|
||||||
|
.order_by("at")
|
||||||
|
)
|
||||||
|
transfergroups = (
|
||||||
|
TransferGroup.objects.prefetch_related(transfer_prefetch)
|
||||||
|
.select_related("valid_by")
|
||||||
|
.order_by("at")
|
||||||
|
)
|
||||||
|
|
||||||
|
# Construction de la réponse
|
||||||
|
history_groups = []
|
||||||
|
for opegroup in opegroups:
|
||||||
|
opegroup_dict = {
|
||||||
|
"type": "operation",
|
||||||
|
"id": opegroup.id,
|
||||||
|
"amount": opegroup.amount,
|
||||||
|
"at": opegroup.at,
|
||||||
|
"checkout_id": opegroup.checkout_id,
|
||||||
|
"is_cof": opegroup.is_cof,
|
||||||
|
"comment": opegroup.comment,
|
||||||
|
"entries": [],
|
||||||
|
"on_acc__trigramme": opegroup.on_acc
|
||||||
|
and opegroup.on_acc.trigramme
|
||||||
|
or None,
|
||||||
|
}
|
||||||
|
for ope in opegroup.opes.all():
|
||||||
|
ope_dict = {
|
||||||
|
"id": ope.id,
|
||||||
|
"type": ope.type,
|
||||||
|
"amount": ope.amount,
|
||||||
|
"article_nb": ope.article_nb,
|
||||||
|
"addcost_amount": ope.addcost_amount,
|
||||||
|
"canceled_at": ope.canceled_at,
|
||||||
|
"article__name": ope.article and ope.article.name or None,
|
||||||
|
"addcost_for__trigramme": ope.addcost_for
|
||||||
|
and ope.addcost_for.trigramme
|
||||||
|
or None,
|
||||||
|
}
|
||||||
|
opegroup_dict["entries"].append(ope_dict)
|
||||||
|
history_groups.append(opegroup_dict)
|
||||||
|
for transfergroup in transfergroups:
|
||||||
|
if transfergroup.filtered_transfers:
|
||||||
|
transfergroup_dict = {
|
||||||
|
"type": "transfer",
|
||||||
|
"id": transfergroup.id,
|
||||||
|
"at": transfergroup.at,
|
||||||
|
"comment": transfergroup.comment,
|
||||||
|
"entries": [],
|
||||||
|
}
|
||||||
|
|
||||||
|
for transfer in transfergroup.filtered_transfers:
|
||||||
|
transfer_dict = {
|
||||||
|
"id": transfer.id,
|
||||||
|
"amount": transfer.amount,
|
||||||
|
"canceled_at": transfer.canceled_at,
|
||||||
|
"from_acc": transfer.from_acc.trigramme,
|
||||||
|
"to_acc": transfer.to_acc.trigramme,
|
||||||
|
}
|
||||||
|
transfergroup_dict["entries"].append(transfer_dict)
|
||||||
|
history_groups.append(transfergroup_dict)
|
||||||
|
|
||||||
|
history_groups.sort(key=lambda group: group["at"])
|
||||||
|
|
||||||
|
stream.write("## Historique\n\n")
|
||||||
|
for group in history_groups:
|
||||||
|
if group["type"] == "operation":
|
||||||
|
stream.write(
|
||||||
|
f"* {group['at']} : opération (valeur {group['amount']} EUR)\n"
|
||||||
|
)
|
||||||
|
if group["comment"]:
|
||||||
|
stream.write(f" Commentaire : {group['comment']}\n")
|
||||||
|
for op in group["entries"]:
|
||||||
|
stream.write(f" - {op['type']} : ")
|
||||||
|
if op["type"] == "purchase":
|
||||||
|
stream.write(f"{op['article_nb']}x {op['article__name']}")
|
||||||
|
stream.write(f" ({op['amount']} EUR)")
|
||||||
|
if op["canceled_at"]:
|
||||||
|
stream.write(" [ANNULÉ]")
|
||||||
|
stream.write("\n")
|
||||||
|
elif group["type"] == "transfer":
|
||||||
|
stream.write(f"* {group['at']} : transferts d'UKF\n")
|
||||||
|
if group["comment"]:
|
||||||
|
stream.write(f" Commentaire : {group['comment']}\n")
|
||||||
|
for op in group["entries"]:
|
||||||
|
stream.write(
|
||||||
|
f" - {op['from_acc']} -> {op['to_acc']} : "
|
||||||
|
f"{op['amount']} EUR"
|
||||||
|
)
|
||||||
|
if op["canceled_at"]:
|
||||||
|
stream.wrte(" [ANNULÉ]")
|
||||||
|
stream.write("\n")
|
||||||
|
|
||||||
|
def _do_handle(self, trigramme: str, stream: ty.TextIO):
|
||||||
|
"""Actually write the personal user data to `stream`"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
kfet_account = Account.objects.get(trigramme=trigramme)
|
||||||
|
except Account.DoesNotExist as exn:
|
||||||
|
raise CommandError("No such trigramme: {}", trigramme) from exn
|
||||||
|
|
||||||
|
try:
|
||||||
|
cof_account = kfet_account.cofprofile
|
||||||
|
except Account.DoesNotExist as exn:
|
||||||
|
raise CommandError(
|
||||||
|
"No cof profile with PK={}", kfet_account.cofprofile
|
||||||
|
) from exn
|
||||||
|
|
||||||
|
stream.write(
|
||||||
|
"# Export de données personnelles pour le trigramme {}\n\n".format(
|
||||||
|
trigramme
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# Generic data
|
||||||
|
stream.write("## Informations générales (COF)\n\n")
|
||||||
|
for field in cof_account._meta.get_fields():
|
||||||
|
fname = field.name
|
||||||
|
if not hasattr(field, "attname"):
|
||||||
|
continue
|
||||||
|
if hasattr(field, "verbose_name"):
|
||||||
|
fname = field.verbose_name
|
||||||
|
stream.write(f"{fname}: {getattr(cof_account, field.attname)}\n")
|
||||||
|
stream.write("\n")
|
||||||
|
|
||||||
|
stream.write("## Informations générales (K-Fêt)\n\n")
|
||||||
|
stream.write(
|
||||||
|
f"Balance : {kfet_account.balance}\n"
|
||||||
|
f"Compte figé : {kfet_account.is_frozen}\n"
|
||||||
|
f"Créé le : {kfet_account.created_at}\n"
|
||||||
|
f"Promotion : {kfet_account.promo}\n"
|
||||||
|
f"Pseudo : {kfet_account.nickname}\n"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Negative
|
||||||
|
try:
|
||||||
|
neg_profile = kfet_account.negative
|
||||||
|
stream.write("Balance négative : oui\n")
|
||||||
|
stream.write(f"Depuis : f{neg_profile.start}\n")
|
||||||
|
stream.write(f"Jusqu'à : f{neg_profile.end}\n")
|
||||||
|
stream.write(f"Rappel le : f{neg_profile.last_rappel}\n\n")
|
||||||
|
except AccountNegative.DoesNotExist:
|
||||||
|
stream.write("Balance négative : non\n\n")
|
||||||
|
|
||||||
|
self._do_handle_history(kfet_account, stream)
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
trigramme = options["trigramme"]
|
||||||
|
out_stream_path = options["output"]
|
||||||
|
|
||||||
|
if out_stream_path == "-":
|
||||||
|
self._do_handle(trigramme, sys.stdout)
|
||||||
|
else:
|
||||||
|
with open(out_stream_path, "w") as handle:
|
||||||
|
self._do_handle(trigramme, handle)
|
||||||
Loading…
Reference in a new issue