Move clean summary function to AdvisoryData class

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
aboutcode-org · Sep 11, 2023 · 7aa8162 · 7aa8162
1 parent cc644a7
commit 7aa8162
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/vulnerabilities/import_runner.py b/vulnerabilities/import_runner.py
@@ -91,7 +91,6 @@ def process_advisories(
             # https://nvd.nist.gov/vuln/detail/CVE-2013-4314
             # https://github.com/cms-dev/cms/issues/888#issuecomment-516977572
             try:
-                data.summary = data.summary.replace("\x00", "\uFFFD")
                 obj, created = Advisory.objects.get_or_create(
                     aliases=data.aliases,
                     summary=data.summary,

diff --git a/vulnerabilities/importer.py b/vulnerabilities/importer.py
@@ -249,9 +249,14 @@ class AdvisoryData:
     date_published: Optional[datetime.datetime] = None
     weaknesses: List[int] = dataclasses.field(default_factory=list)
 
+
     def __post_init__(self):
         if self.date_published and not self.date_published.tzinfo:
             logger.warning(f"AdvisoryData with no tzinfo: {self!r}")
+        self.clean_summary()
+
+    def clean_summary(self):
+        self.summary = self.summary.replace("\x00", "\uFFFD")
 
     def to_dict(self):
         return {