Move FixQuality and Rating to common codetf module

Author: drdavella

src/codemodder/codetf/common.py

@@ -84,3 +84,14 @@ def validate_description(self):
         if self.description is not None and not self.description:
             raise ValueError("description must not be empty")
         return self
+
+
+class Rating(BaseModel):
+    score: int
+    description: Optional[str] = None
+
+
+class FixQuality(BaseModel):
+    safetyRating: Rating
+    effectivenessRating: Rating
+    cleanlinessRating: Rating

src/codemodder/codetf/v2/codetf.py

@@ -21,6 +21,7 @@
 from ..common import Change as CommonChange
 from ..common import (
     CodeTFWriter,
+    FixQuality,
     Rule,
 )
 
@@ -104,17 +105,6 @@ class Strategy(Enum):
     deterministic = "deterministic"
 
 
-class Rating(BaseModel):
-    score: int
-    description: Optional[str] = None
-
-
-class FixQuality(BaseModel):
-    safetyRating: Rating
-    effectivenessRating: Rating
-    cleanlinessRating: Rating
-
-
 class ChangeSet(BaseModel):
     """A set of changes made to a file at `path`"""
 

src/codemodder/codetf/v3/codetf.py

@@ -5,7 +5,7 @@
 
 from pydantic import BaseModel, model_validator
 
-from ..common import Change, CodeTFWriter, Finding
+from ..common import Change, CodeTFWriter, Finding, FixQuality
 from ..v2.codetf import Finding as V2Finding
 
 
@@ -92,17 +92,6 @@ class FixMetadata(BaseModel):
     generation: GenerationMetadata
 
 
-class Rating(BaseModel):
-    score: int
-    description: Optional[str] = None
-
-
-class FixQuality(BaseModel):
-    safetyRating: Rating
-    effectivenessRating: Rating
-    cleanlinessRating: Rating
-
-
 class FixResult(BaseModel):
     """Result corresponding to a single finding"""