add datamodel for project and workpackages; mod other modules

src/pytestpavement/models/__init__.py

@@ -1,3 +1,5 @@
 from .citt import *
 from .material import *
+from .project import *
 from .sheartest import *
+from .workpackage import *

src/pytestpavement/models/data.py

@@ -1,5 +1,4 @@
 import datetime
-from xml.dom.minidom import Document
 
 from mongoengine import *
 

src/pytestpavement/models/material.py

@@ -11,6 +11,8 @@ class Material(Document):
                          }})
 
     tags = ListField(StringField())
+    project = StringField(required=False)
+    norm = StringField(required=True, default='TP Asphalt Teil 24')
 
     meta = {
         'allow_inheritance': True,
@@ -26,8 +28,10 @@ class Asphalt(Material):
 
     name = StringField()
     material = StringField()
+    bitumen = StringField(required=False)
 
     young_modulus = DictField()
+    fatigue = DictField()
 
 
 class Bitumen(Material):

src/pytestpavement/models/project.py

@@ -0,0 +1,27 @@
+import datetime
+
+from mongoengine import *
+
+
+class Project(Document):
+    
+    name = StringField(required=True)
+    name_short = StringField(required=False)
+    project_id = StringField(required=True)
+
+    date = DateTimeField(default=datetime.datetime.now,
+                         wtf_options={"render_kw": {
+                             "step": "60"
+                         }})
+
+    tags = ListField(StringField())
+
+    meta = {
+        'allow_inheritance': True,
+        'index_opts': {},
+        'index_background': True,
+        'index_cls': False,
+        'auto_create_index': True,
+        'collection': 'projects'
+    }
+

src/pytestpavement/models/sheartest.py

@@ -4,6 +4,8 @@ from mongoengine import *
 
 from .data import DataSheartest
 from .material import Material
+from .project import Project
+from .workpackage import Workpackage
 
 
 class DynamicShearTest(Document):
@@ -24,8 +26,8 @@ class DynamicShearTest(Document):
     filehash = StringField(required=True)
     speciment_name = StringField()
 
-    project = StringField(required=True)
-    workpackage = StringField()
+    project = LazyReferenceField(Project, required=True)
+    workpackage = LazyReferenceField(Workpackage, required=True)
 
     material1 = LazyReferenceField(Material, required=True)
     material2 = LazyReferenceField(Material, required=True)

src/pytestpavement/models/workpackage.py

@@ -0,0 +1,28 @@
+import datetime
+
+from mongoengine import *
+
+from .project import Project
+
+
+class Workpackage(Document):
+    
+    name = StringField(required=True)
+    name_short = StringField(required=False)
+    wp_id = StringField(required=True)
+    
+    project = LazyReferenceField(Project, required=True)
+    
+    date = DateTimeField(default=datetime.datetime.now,
+                        wtf_options={"render_kw": {
+                            "step": "60"
+                        }})
+    
+    meta = {
+        'allow_inheritance': True,
+        'index_opts': {},
+        'index_background': True,
+        'index_cls': False,
+        'auto_create_index': True,
+        'collection': 'workpackages'
+    }