fix some import errors, step_before/after now single value no list

src/paveit/datamodels/__init__.py

@@ -3,6 +3,7 @@ from .citt import *
 from .components import *
 from .data import *
 from .enumeration import *
+from .labworks import *
 from .machines import *
 from .material import *
 from .messages import *

src/paveit/datamodels/labworks.py

@@ -1,6 +1,7 @@
-from .tasks import TaskManagerBase
 from mongoengine import *
 
+from .taskmanager import TaskManagerBase
+
 
 # Vorbereitungen Performance Untersuchungen
 class LabworksDrillRoad(TaskManagerBase):

src/paveit/datamodels/taskmanager.py

@@ -34,8 +34,8 @@ class TaskManagerBase(Document):
     
     series = StringField(default='Serie 01')
     
-    step_before = ListField(LazyReferenceField('self'), required=False)
-    step_after = ListField(LazyReferenceField('self'), required=False)
+    step_before = LazyReferenceField('self', required=False)
+    step_after = LazyReferenceField('self', required=False)
     
     meta = {
         'allow_inheritance': True,
@@ -52,6 +52,10 @@ class TaskCITTStiffness(TaskManagerBase):
     
     material = LazyReferenceField(Material, required=True)
     
+class TaskCITTFatigue(TaskManagerBase):
+    
+    material = LazyReferenceField(Material, required=True)
+    
     
 class TaskDynShearStiffness(TaskManagerBase):
     

src/paveit/labtest/citt.py

@@ -7,7 +7,7 @@ import pandas as pd
 from bson import ObjectId
 
 from paveit import calc_nu, fit_cos
-from paveit.datamodels import CITTSiffness, CITTSiffnessResults
+from paveit.datamodels.citt import CITTSiffnessResults
 from paveit.io import read_geosys
 from paveit.labtest import DataSineLoad
 from paveit.labtest.citt_fatigue import CittAnalyseFatigue

src/paveit/labtest/dsv.py

@@ -7,7 +7,6 @@ import pandas as pd
 from bson import ObjectId
 
 from paveit import calc_nu, fit_cos
-from paveit.datamodels import CITTSiffness, CITTSiffnessResults
 from paveit.io import read_geosys
 from paveit.labtest import DataSineLoad