Changeset 16:bb9f1fb34a32 for trac_captcha/ticket.py

Timestamp:

06/29/10 23:38:00 (23 months ago)

Author:

fs

Branch:

default

Message:

moved code for captcha injection in the controller so it can be shared

Files:

• trac_captcha/ticket.py (modified) (4 diffs)

trac_captcha/ticket.py

@@ -23 +23 @@
 # THE SOFTWARE.
 
-from genshi.builder import tag
 from genshi.filters.transform import Transformer
 from trac.core import Component, implements
@@ -29 +28 @@
 from trac.web.api import ITemplateStreamFilter
 
-from trac_captcha.controller import initialize_captcha_data, TracCaptchaController
+from trac_captcha.controller import TracCaptchaController
 
 __all__ = ['TicketCaptcha']
@@ -41 +40 @@
         if filename != 'ticket.html':
             return stream
-        initialize_captcha_data(req)
-        controller = TracCaptchaController(self.env)
-        if 'token' in req.captcha_data:
-            return stream | self.captcha_token_tag(req)
-        if controller.should_skip_captcha(req):
-            return stream
-        
-        captcha = controller.genshi_stream(req)
-        return stream | Transformer('//div[@class="buttons"]').before(captcha)
+        transformer = Transformer('//div[@class="buttons"]')
+        return TracCaptchaController(self.env).inject_captcha_into_stream(req, stream, transformer)
     
     # --- ITicketManipulator ---------------------------------------------------
@@ -61 +53 @@
         return ((None, error_message),)
     
-    # --- private API ----------------------------------------------------------
-    
-    def captcha_token_tag(self, req):
-        token = req.captcha_data['token']
-        input_tag = tag.input(type='hidden', name='__captcha_token', value=token)
-        return Transformer('//div[@class="buttons"]').before(input_tag)