Clean up updateService

Author: lsetiawan

odm2api/ODM2/services/updateService.py

@@ -5,47 +5,47 @@
 from datetime import datetime
 
 from odm2api.ODM2 import serviceBase
-from odm2api.ODM2.models import *
+from odm2api.ODM2.models import (Actions, Results)
 
 
 # ################################################################################
 # Annotations
 # ################################################################################
 
 class UpdateODM2(serviceBase):
-
     def update(self, value):
         self._session.add(value)
         self._session.commit()
         return value
 
-# ################################################################################
-# Core
-# ################################################################################
+    # ################################################################################
+    # Core
+    # ################################################################################
     def updateResultValidDateTime(self, resultId, dateTime):
 
-        #check type of "validdatetime'
-        #if not datetime do this:
+        # check type of "validdatetime'
+        # if not datetime do this:
         # dt = dateTime.to_datetime()
-        #else dt = dateTime
+        # else dt = dateTime
         if (type(dateTime) != datetime):
             dt = dateTime.to_datetime()
         else:
             dt = dateTime
-            q = self._session.query(Results).filter(Results.ResultID == int(resultId)).update({'ValidDateTime': dt})
+            q = self._session.query(Results).filter(Results.ResultID == int(resultId))
+            q.update({'ValidDateTime': dt})
+
             self._session.commit()
 
     def updateResult(self, resultID=None, valuecount=None, result=None):
         if resultID:
             q = self._session.query(Results).filter(Results.ResultID == int(resultID))
             if valuecount:
-                q.update({"ValueCount": valuecount})
+                q.update({'ValueCount': valuecount})
         if result:
             self._session.add(result)
         self._session.commit()
 
-
-    def updateAction(self, actionID=None, begin=None, end=None, action = None):
+    def updateAction(self, actionID=None, begin=None, end=None, action=None):
         if actionID:
             q = self._session.query(Actions).filter(Actions.ActionID == int(actionID))
             # if (type(begin) != datetime):
@@ -54,77 +54,40 @@ def updateAction(self, actionID=None, begin=None, end=None, action = None):
             #     end = end.to_datetime()
 
             if begin:
-                q.update({"BeginDateTime": begin})
+                q.update({'BeginDateTime': begin})
             if end:
-                q.update({"EndDateTime": end})
+                q.update({'EndDateTime': end})
         elif action:
             self._session.add(action)
         self._session.commit()
 
-
 # ################################################################################
 # Data Quality
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Equipment
 # ################################################################################
-
-
-
-
-
 # ################################################################################
 # Extension Properties
 # ################################################################################
-
-
-
-
 # ################################################################################
 # External Identifiers
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Lab Analyses
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Provenance
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Results
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Sampling Features
 # ################################################################################
-
-
-
-
 # ################################################################################
 # Sensors
 # ################################################################################
-
-
-
-# ################################################################################
+################################################################################
 # ODM2
 # ################################################################################