[Kimchi-devel] [PATCH] [Wok 5/6] Save log entry IDs for requests that generate tasks

[Lucio Correia]

Signed-off-by: Lucio Correia <luciojhc at linux.vnet.ibm.com>
---
 src/wok/control/base.py | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/src/wok/control/base.py b/src/wok/control/base.py
index 686a1aa..786ecb7 100644
--- a/src/wok/control/base.py
+++ b/src/wok/control/base.py
@@ -26,6 +26,7 @@ import urllib2
 
 
 import wok.template
+from wok.asynctask import save_request_log_id
 from wok.auth import USER_GROUPS, USER_NAME, USER_ROLES
 from wok.control.utils import get_class_name, internal_redirect, model_fn
 from wok.control.utils import parse_request, validate_method
@@ -148,6 +149,7 @@ class Resource(object):
                      self.info['persistent'] is True):
                     result = render_fn(self, action_result)
                     status = cherrypy.response.status
+
                     return result
             except WokException, e:
                 details = e
@@ -157,7 +159,9 @@ class Resource(object):
                 # log request
                 code = self.getRequestMessage(method, action_name)
                 reqParams = utf8_dict(self.log_args, request)
-                log_request(code, reqParams, details, method, status)
+                log_id = log_request(code, reqParams, details, method, status)
+                if status == 202:
+                    save_request_log_id(log_id, action_result['id'])
 
         wrapper.__name__ = action_name
         wrapper.exposed = True
@@ -209,7 +213,7 @@ class Resource(object):
             raise
         finally:
             # log request
-            if method not in LOG_DISABLED_METHODS:
+            if method not in LOG_DISABLED_METHODS and status != 202:
                 code = self.getRequestMessage(method)
                 log_request(code, self.log_args, details, method, status)
 
@@ -293,6 +297,15 @@ class AsyncResource(Resource):
             raise cherrypy.HTTPError(405, e.message)
 
         cherrypy.response.status = 202
+
+        # log request
+        method = 'DELETE'
+        code = self.getRequestMessage(method)
+        reqParams = utf8_dict(self.log_args)
+        log_id = log_request(code, reqParams, None, method,
+                             cherrypy.response.status)
+        save_request_log_id(log_id, task['id'])
+
         return wok.template.render("Task", task)
 
 
@@ -440,7 +453,7 @@ class Collection(object):
             status = e.status
             raise
         finally:
-            if method not in LOG_DISABLED_METHODS:
+            if method not in LOG_DISABLED_METHODS and status != 202:
                 # log request
                 code = self.getRequestMessage(method)
                 reqParams = utf8_dict(self.log_args, params)
@@ -466,6 +479,15 @@ class AsyncCollection(Collection):
         args = self.model_args + [params]
         task = create(*args)
         cherrypy.response.status = 202
+
+        # log request
+        method = 'POST'
+        code = self.getRequestMessage(method)
+        reqParams = utf8_dict(self.log_args, params)
+        log_id = log_request(code, reqParams, None, method,
+                             cherrypy.response.status)
+        save_request_log_id(log_id, task['id'])
+
         return wok.template.render("Task", task)
 
 
-- 
1.9.1