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)
