feat: turn dump anonymization into optional

bd_api/apps/core/management/commands/dumpfixture.py

@@ -3,6 +3,7 @@
 from datetime import datetime
 from json import dump, load
 
+from django.core.management.base import CommandParser
 from django.core.management.commands.dumpdata import Command as DumpDataCommand
 from faker import Faker
 from loguru import logger
@@ -46,19 +47,29 @@ def empty():
 class Command(DumpDataCommand):
  """Dump data, avoiding profiles"""
 
+ def add_arguments(self, parser: CommandParser):
+ super().add_arguments(parser)
+ parser.add_argument(
+ "--anonymize",
+ dest="anonymize",
+ action="store_true",
+ help="Anonymize account fixtures with fake data",
+ )
+
  def handle(self, *args, **options) -> str | None:
  logger.info("Dump fixtures")
  response = super().handle(*args, **options)
 
- logger.info("Filter fixtures")
- output = options["output"]
- with open(output, "r") as file:
- data = load(file)
- for datum in data:
- if datum["model"] == "account.account":
- if datum["fields"]["profile"] in (2, 3):
- datum["fields"] = {**datum["fields"], **empty()}
- with open(output, "w") as file:
- dump(data, file)
+ if options["anonymize"]:
+ logger.info("Filter fixtures")
+ output = options["output"]
+ with open(output, "r") as file:
+ data = load(file)
+ for datum in data:
+ if datum["model"] == "account.account":
+ if datum["fields"]["profile"] in (2, 3):
+ datum["fields"] = {**datum["fields"], **empty()}
+ with open(output, "w") as file:
+ dump(data, file)
 
  return response

bd_api/apps/core/management/commands/loadfixture.py

@@ -3,6 +3,7 @@
 
 from django.conf import settings
 from django.core.management import call_command
+from django.core.management.base import CommandParser
 from django.db import connection
 from loguru import logger
 from modeltranslation.management.commands.loaddata import Command as LoadDataCommand
@@ -33,7 +34,7 @@ class Command(LoadDataCommand):
  4. Build index
  """
 
- def add_arguments(self, parser):
+ def add_arguments(self, parser: CommandParser):
  super().add_arguments(parser)
  parser.add_argument(
  "--build-index",