Move BaseEnum to common code

pyplanner/models/__init__.py

@@ -32,6 +32,18 @@ class Base:
         return self.uuid[:8]
 
 
+class BaseEnum:
+    @classmethod
+    def to_yaml(cls, representer, node):
+        return representer.represent_scalar(
+            node.value
+        )
+
+    @classmethod
+    def from_yaml(cls, constructor, node):
+        return cls(node)
+
+
 from .collections import Collection
 from .milestones import Milestone
 from .sprints import Sprint

pyplanner/models/items.py

@@ -1,5 +1,5 @@
 from pyplanner.database import SQLABase
-from pyplanner.models import Base
+from pyplanner.models import Base, BaseEnum
 from sqlalchemy import Column, Integer, String, Text, ForeignKey
 from sqlalchemy.orm import relationship, backref
 from sqlalchemy.ext.hybrid import hybrid_property
@@ -8,18 +8,6 @@ from enum import Enum
 from functools import total_ordering
 
 
-class BaseEnum:
-    @classmethod
-    def to_yaml(cls, representer, node):
-        return representer.represent_scalar(
-            node.value
-        )
-
-    @classmethod
-    def from_yaml(cls, constructor, node):
-        return cls(node)
-
-
 @total_ordering
 class Length(BaseEnum, Enum):
     MINUTES = 'mi'