Skip to content

Commit 637d8cc

Browse files
committed
CodetfV2 will now use more common types
1 parent 1f2788c commit 637d8cc

File tree

1 file changed

+12
-20
lines changed

1 file changed

+12
-20
lines changed

src/codemodder/codetf/v2/codetf.py

Lines changed: 12 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -15,40 +15,32 @@
1515

1616
from codemodder import __version__
1717

18-
from ..common import (
19-
CaseInsensitiveEnum,
20-
)
18+
from ..common import Action as CommonAction
2119
from ..common import Change as CommonChange
2220
from ..common import (
2321
CodeTFWriter,
22+
)
23+
from ..common import DiffSide as CommonDiffSide
24+
from ..common import (
2425
FixQuality,
26+
)
27+
from ..common import PackageAction as CommonPackageAction
28+
from ..common import PackageResult as CommonPackageResult
29+
from ..common import (
2530
Rule,
2631
)
2732

2833
if TYPE_CHECKING:
2934
from codemodder.context import CodemodExecutionContext
3035

3136

32-
class Action(CaseInsensitiveEnum):
33-
ADD = "add"
34-
REMOVE = "remove"
35-
36-
37-
class PackageResult(CaseInsensitiveEnum):
38-
COMPLETED = "completed"
39-
FAILED = "failed"
40-
SKIPPED = "skipped"
41-
37+
Action = CommonAction
4238

43-
class DiffSide(CaseInsensitiveEnum):
44-
LEFT = "left"
45-
RIGHT = "right"
39+
PackageResult = CommonPackageResult
4640

41+
DiffSide = CommonDiffSide
4742

48-
class PackageAction(BaseModel):
49-
action: Action
50-
result: PackageResult
51-
package: str
43+
PackageAction = CommonPackageAction
5244

5345

5446
class Change(BaseModel):

0 commit comments

Comments
 (0)