def get_session_id(cls, communicator): if cls.debug: return 1 headers = parse_request_headers(communicator) cookies = request_headers_get_cookies(headers) if Config.ulteo_session_cookie in cookies: sess_id = cookies[Config.ulteo_session_cookie].strip('"') return sess_id
def pre_process(self, context): """ Get auth_cookies and cookies form session and inject to request headers. """ Logger.debug("CookieFilter pre_process") session = context.session headers = context.communicator.http.headers.split('\r\n')[:-1] cookies = '' if session.get('auth_cookies'): for cookie in session['auth_cookies'].values(): cookie_match = COOKIE_RE.search(cookie) cookies += cookie_match.group(1) + '=' + cookie_match.group( 2) + '; ' if session.get('cookies'): for name, value in session['cookies'].items(): cookies += name + '=' + value + '; ' if cookies: Logger.debug("Injecting following cookies: " + cookies) req_headers = parse_request_headers(context.communicator) if 'Cookie' in req_headers: req_cookies = request_headers_get_cookies(req_headers) modified = bool(cookies) for name, value in req_cookies.items(): if name in self.options['suppressed']: modified = True Logger.debug("Suppressing cookies: " + name) else: cookies += name + '=' + value + '; ' if modified: for header in headers: if header.startswith('Cookie: '): newheader = 'Cookie: ' + cookies headers[headers.index(header)] = newheader elif cookies: headers.append('Cookie: ' + cookies) context.communicator.http.headers = '\r\n'.join(headers)
def pre_process(self, context): """ Get auth_cookies and cookies form session and inject to request headers. """ Logger.debug("CookieFilter pre_process") session = context.session headers = context.communicator.http.headers.split('\r\n')[:-1] cookies = '' if session.get('auth_cookies'): for cookie in session['auth_cookies'].values(): cookie_match = COOKIE_RE.search(cookie) cookies += cookie_match.group(1) + '=' + cookie_match.group(2) + '; ' if session.get('cookies'): for name, value in session['cookies'].items(): cookies += name + '=' + value + '; ' if cookies: Logger.debug("Injecting following cookies: " + cookies) req_headers = parse_request_headers(context.communicator) if 'Cookie' in req_headers: req_cookies = request_headers_get_cookies(req_headers) modified = bool(cookies) for name, value in req_cookies.items(): if name in self.options['suppressed']: modified = True Logger.debug("Suppressing cookies: " + name) else: cookies += name + '=' + value + '; ' if modified: for header in headers: if header.startswith('Cookie: '): newheader = 'Cookie: ' + cookies headers[headers.index(header)] = newheader elif cookies: headers.append('Cookie: ' + cookies) context.communicator.http.headers = '\r\n'.join(headers)