From 5d2bff2c8de37698fdb5b1d7bbe2d04e43d7fd2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Markus=20Clau=C3=9F?= <markus@linsenbruch.de>
Date: Sun, 2 Jul 2023 22:37:33 +0200
Subject: [PATCH] add first labwork tasks for flow

---
 src/paveit/datamodels/labworks.py    | 27 +++++++++++++++++++++++++++
 src/paveit/datamodels/taskmanager.py |  7 +++++--
 2 files changed, 32 insertions(+), 2 deletions(-)
 create mode 100644 src/paveit/datamodels/labworks.py

diff --git a/src/paveit/datamodels/labworks.py b/src/paveit/datamodels/labworks.py
new file mode 100644
index 0000000..63917b6
--- /dev/null
+++ b/src/paveit/datamodels/labworks.py
@@ -0,0 +1,27 @@
+from .tasks import TaskManagerBase
+from mongoengine import *
+
+
+# Vorbereitungen Performance Untersuchungen
+class LabworksDrillRoad(TaskManagerBase):
+    pass
+
+class LabworksMakingAsphaltSlabs(TaskManagerBase):
+    pass
+
+class LabworksDrillAsphaltSlabs(TaskManagerBase):
+    pass
+
+class LabworksSawDrillCores(TaskManagerBase):
+    pass
+
+class LabworksGrindingAsphaltSamples(TaskManagerBase):
+    pass
+
+class LabworksDetermineDensity(TaskManagerBase):
+    pass
+
+class LabworksDetermineGeometry(TaskManagerBase):
+    pass
+
+
diff --git a/src/paveit/datamodels/taskmanager.py b/src/paveit/datamodels/taskmanager.py
index 808599f..1da1d26 100755
--- a/src/paveit/datamodels/taskmanager.py
+++ b/src/paveit/datamodels/taskmanager.py
@@ -3,13 +3,13 @@ import datetime
 
 from mongoengine import *
 
+from .client import Client
+from .enumeration import ProcessStatusEnum
 from .machines import MachineBase
 from .material import Material
 from .project import Project
 from .usermanagement import Organisation, User
 from .workpackage import Workpackage
-from .enumeration import ProcessStatusEnum
-from .client import Client
 
 
 class TaskManagerBase(Document):
@@ -34,6 +34,9 @@ class TaskManagerBase(Document):
     
     series = StringField(default='Serie 01')
     
+    step_before = ListField(LazyReferenceField('self'), required=False)
+    step_after = ListField(LazyReferenceField('self'), required=False)
+    
     meta = {
         'allow_inheritance': True,
         'index_opts': {},