Merge pull request #4 from willyw0nka/to-json

feat: add to_json decorator to all dataclasses

pygrype/core/decorators/to_json.py

@@ -0,0 +1,11 @@
+import json
+from dataclasses import asdict
+from typing import Type, TypeVar
+
+T = TypeVar('T')
+
+def to_json(cls: Type[T]) -> Type[T]:
+    def to_json_func(self: T):
+        return json.dumps(asdict(self))
+    cls.to_json = to_json_func
+    return cls

pygrype/core/grype_version.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class GrypeVersion:
     version: str
     syft_version: str

pygrype/core/list/db_meta_data.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class DBMetaData:
     built: str
     version: int

pygrype/core/scan/artifact.py

@@ -1,11 +1,13 @@
 from dataclasses import dataclass
 from typing import List
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.location import Location
 from pygrype.core.scan.upstream import Upstream
 
 
 @dataclass
+@to_json
 class Artifact:
     id: str
     name: str

pygrype/core/scan/cvss.py

@@ -1,10 +1,12 @@
 from dataclasses import dataclass
 from typing import Optional
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.cvss_metrics import CVSSMetrics
 
 
 @dataclass
+@to_json
 class CVSS:
     source: Optional[str]
     type: Optional[str]

pygrype/core/scan/cvss_metrics.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class CVSSMetrics:
     baseScore: float
     exploitabilityScore: float

pygrype/core/scan/distro.py

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class Distro:
     name: str
     version: str

pygrype/core/scan/layer.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class Layer:
     mediaType: str
     digest: str

pygrype/core/scan/location.py

@@ -1,7 +1,10 @@
 from dataclasses import dataclass
 from typing import Optional
 
+from pygrype.core.decorators.to_json import to_json
+
 @dataclass
+@to_json
 class Location:
     path: str
     layerID: Optional[str] = None

pygrype/core/scan/match.py

@@ -1,12 +1,14 @@
 from dataclasses import dataclass
 from typing import List
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.artifact import Artifact
 from pygrype.core.scan.match_details import MatchDetails
 from pygrype.core.scan.vulnerability import Vulnerability
 
 
 @dataclass
+@to_json
 class Match:
     vulnerability: Vulnerability
     relatedVulnerabilities: List[Vulnerability]

pygrype/core/scan/match_details.py

@@ -1,10 +1,12 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.match_details_found import MatchDetailsFound
 from pygrype.core.scan.match_details_searched_by import MatchDetailsSearchedBy
 
 
 @dataclass
+@to_json
 class MatchDetails:
     type: str
     matcher: str

pygrype/core/scan/match_details_found.py

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class MatchDetailsFound:
     vulnerabilityID: str
     versionConstraint: str

pygrype/core/scan/match_details_searched_by.py

@@ -1,11 +1,12 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.package import Package
 from pygrype.core.scan.searched_by_distro import Distro
 
-
 @dataclass
+@to_json
 class MatchDetailsSearchedBy:
     namespace: str
     cpes: Optional[List[str]]

pygrype/core/scan/package.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class Package:
     name: str
     version: str

pygrype/core/scan/scan.py

@@ -1,12 +1,13 @@
 from dataclasses import dataclass
 from typing import List
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.distro import Distro
 from pygrype.core.scan.match import Match
 from pygrype.core.scan.scan_source import ScanSource
 
-
 @dataclass
+@to_json
 class Scan:
     matches: List[Match]
     source: ScanSource

pygrype/core/scan/scan_source.py

@@ -1,10 +1,12 @@
 from dataclasses import dataclass
 from typing import Union
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.target import Target
 
 
 @dataclass
+@to_json
 class ScanSource:
     type: str
     target: Union[Target, str]

pygrype/core/scan/searched_by_distro.py

@@ -1,7 +1,9 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class Distro:
     type: str
     version: str

pygrype/core/scan/target.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass
 from typing import List
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.layer import Layer
 
-
 @dataclass
+@to_json
 class Target:
     userInput: str
     imageID: str

pygrype/core/scan/upstream.py

@@ -1,6 +1,8 @@
 from dataclasses import dataclass
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class Upstream:
     name: str

pygrype/core/scan/vulnerability.py

@@ -1,11 +1,13 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from pygrype.core.decorators.to_json import to_json
 from pygrype.core.scan.cvss import CVSS
 from pygrype.core.scan.vulnerability_fix import VulnerabilityFix
 
 
 @dataclass
+@to_json
 class Vulnerability:
     id: str
     description: Optional[str]

pygrype/core/scan/vulnerability_fix.py

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
 from typing import List
 
+from pygrype.core.decorators.to_json import to_json
 
 @dataclass
+@to_json
 class VulnerabilityFix:
     versions: List[str]
     state: str