adding in a finaltime in query

sdss · Feb 9, 2018 · c304c7c · c304c7c
1 parent c805353
commit c304c7c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/python/marvin/tools/query.py b/python/marvin/tools/query.py
@@ -798,11 +798,17 @@ def run(self, start=None, end=None, raw=None, orm=None, core=None):
             # clear the session
             self.session.close()
 
-            #memory_usage('7 - after session close, before results dump')
+            # memory_usage('7 - after session close, before results dump')
 
-            return Results(results=res, query=self.query, count=count, mode=self.mode,
-                           returntype=self.returntype, queryobj=self, totalcount=self.totalcount,
-                           chunk=self.limit, runtime=self.runtime, start=start, end=end)
+            final = Results(results=res, query=self.query, count=count, mode=self.mode,
+                            returntype=self.returntype, queryobj=self, totalcount=self.totalcount,
+                            chunk=self.limit, runtime=self.runtime, start=start, end=end)
+
+            # get the final time
+            posttime = datetime.datetime.now()
+            self.finaltime = (posttime - starttime)
+
+            return final
 
         elif self.mode == 'remote':
             # Fail if no route map initialized