0001-build-cookies-from-page.cookies-9702.patch

Josué Kouka, 20 janvier 2016 09:32

Voir les différences: en ligne côte à côte

Subject: [PATCH] build cookies from page.cookies (#9702)

 mandayejs/do_login.js      | 35 ++++++++++++-----------------------
 mandayejs/mandaye/utils.py | 22 ++++++++++++++--------
 mandayejs/mandaye/views.py |  4 ++--
 3 files changed, 28 insertions(+), 33 deletions(-)

     var input = JSON.parse(system.stdin.read(2000)); // no .readAll()...
     headers_list = []
     page.onResourceReceived = function(response){
         for (var i=0; i < response.headers.length; i++){
             var c_header = response.headers[i];
             if (c_header['name'] === 'Set-Cookie'){
                 headers_list.push(c_header);
+            }
+        }
+    }
     page.open(input.address, function() {
       page.onLoadFinished = function() {
          if (page.injectJs(input.auth_checker)){
              input.auth_success = page.evaluate(function(){
                  return auth_success();
              });
+         }
           if (page.injectJs(input.auth_checker)){
               input.auth_success = page.evaluate(function(){
                   return auth_success();
               });
+          }
         if (!input.auth_success){
             console.log(JSON.stringify({'result': 'failure', 'reason': 'authentication failed'}));
             phantom.exit();
+        }
         page.render('login.png');
         console.log(JSON.stringify({'result': 'ok', 'cookies': page.cookies, 'headers': headers_list, 'url': page.frameUrl}));
         phantom.exit();
           if (!input.auth_success){
               console.log(JSON.stringify({'result': 'failure', 'reason': 'authentication failed'}));
               phantom.exit();
+          }
           page.render('login.png');
           console.log(JSON.stringify({'result': 'ok', 'cookies': page.cookies, 'url': page.frameUrl}));
           phantom.exit();
+      }
       page.evaluate(function(input) {
           var locators = input.locators;

         result = json.loads(stdout)
         return result
     def cookie_builder(headers):
         """Build Cookies from list of headers
         """
         cookie = SimpleCookie()
         for header in headers:
             cookie.load('; '.join(header.values()).encode('ascii'))
         return cookie
     def cookie_builder(cookies):
         ck = SimpleCookie()
         for cookie in cookies:
             raw_cookie = ';'.join([
                 '{}={}'.format(k,v) for k,v in cookie.items()
                 if k not in ('name','value')
             ])
             raw_cookie = '{}={};{}'.format(
                 cookie['name'],
                 cookie['value'],
                 raw_cookie
+            )
             ck.load(raw_cookie)
         return ck
     def get_password_field():
         """Return name of the password field

                     window.top.location = "{{url}}";</script>')
         context = RequestContext(request, {'url': url})
         response = HttpResponse(template.render(context))
         if result.get('headers',None):
             response.cookies = cookie_builder(result.get('headers'))
         if result.get('cookies',None):
             response.cookies = cookie_builder(result.get('cookies'))
         return response
+    -

Projet

Général

Profil

Produits Entr'ouvert » Mandaye

0001-build-cookies-from-page.cookies-9702.patch