Merge pull request #1539 from gustavomm19/context-command

add flag --all to generate_asset_context command

breathecode/registry/management/commands/generate_asset_context.py

@@ -11,8 +11,17 @@
 class Command(BaseCommand):
     help = "Generate asset context for all assets."
 
+    def add_arguments(self, parser):
+        parser.add_argument("--all", type=str, help="Allow to update or create context on all assets")
+
     def handle(self, *args, **options):
-        assets = Asset.objects.filter(assetcontext__isnull=True)
+
+        filters = {}
+        assets = Asset.objects.all()
+        if "all" not in options or options["all"] not in ["true", "True", "1"]:
+            filters["assetcontext__isnull"] = True
+        assets = assets.filter(**filters)
+
         for asset in assets:
             try:
                 AssetContext.objects.update_or_create(asset=asset, defaults={"status": "PROCESSING"})