From 4bb06571002fec7f7b0caab50978fbd5debc41cf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bj=C3=B6rn=20Ricks?= Date: Thu, 16 Nov 2023 16:38:42 +0100 Subject: [PATCH] Change: Use StrEnum for CVE Change EventName This enum already used the same pattern as our new StrEnum class. --- pontos/nvd/models/cve_change.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/pontos/nvd/models/cve_change.py b/pontos/nvd/models/cve_change.py index 62b1e54f..bcd0136b 100644 --- a/pontos/nvd/models/cve_change.py +++ b/pontos/nvd/models/cve_change.py @@ -4,14 +4,13 @@ from dataclasses import dataclass from datetime import datetime -from enum import Enum from typing import Optional from uuid import UUID -from pontos.models import Model +from pontos.models import Model, StrEnum -class EventName(str, Enum): +class EventName(StrEnum): CVE_RECEIVED = "CVE Received" INITAL_ANALYSIS = "Initial Analysis" REANALYSIS = "Reanalysis" @@ -25,9 +24,6 @@ class EventName(str, Enum): CVE_REJECTED = "CVE Rejected" CVE_UNREJECTED = "CVE Unrejected" - def __str__(self) -> str: - return self.value - @dataclass class Detail: