From e1eaffe0618278c3a74878212a1bbbfbddc8761a Mon Sep 17 00:00:00 2001
From: Samuel Conjard <samuel.conjard@univ-grenoble-alpes.fr>
Date: Mon, 27 Jan 2025 16:02:47 +0100
Subject: [PATCH] Secure set body xml management command with catching errors 2

---
 src/ptf_tools/management/commands/set_article_body_xml.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/ptf_tools/management/commands/set_article_body_xml.py b/src/ptf_tools/management/commands/set_article_body_xml.py
index 6827da1d..47097db5 100644
--- a/src/ptf_tools/management/commands/set_article_body_xml.py
+++ b/src/ptf_tools/management/commands/set_article_body_xml.py
@@ -1,5 +1,5 @@
 from django.core.management.base import BaseCommand  # type: ignore
-from ptf import utils
+from ptf import exceptions, models, utils
 from ptf.cmds.xml import xml_utils
 from ptf.models import Article
 
@@ -43,5 +43,9 @@ class Command(BaseCommand):
 
                     article.body_xml = xml_utils.get_xml_from_text("body", text)
                     article.save()
-            except RuntimeError:
+            except (
+                RuntimeError,
+                models.DataStream.DoesNotExist,
+                exceptions.ResourceDoesNotExist,
+            ):
                 continue
-- 
GitLab