ws_sso_content_reader Namespace Reference

Classes
class	HTTPSClientAuthHandler

Functions
def	_getResponse (opener, url, post_data=None, debug=False)
def	checkRequiredArguments (opts, parser)
def	getContent (target_url, cert_path, key_path, post_data=None, debug=False, adfslogin=None)
def	getFile (path)
def	getParentURL (url)
def	getResponseContent (opener, url, post_data=None, debug=False)
def	getResponseURL (opener, url, post_data=None, debug=False)
def	getSSOCookie (opener, target_url, cookie, debug=False)

Variables
	action
	args
	content
	default
	dest
	help
	opts
	parser
	usage

Function Documentation

_getResponse()

def ws_sso_content_reader._getResponse (   opener,   url,   post_data = None,   debug = False  )

private

Definition at line 27 of file ws_sso_content_reader.py.

Referenced by getContent(), getResponseContent(), and getResponseURL().

def _getResponse(opener, url, post_data=None, debug=False):
  response = opener.open(url, post_data)
  if debug:
    sys.stderr.write("Code: %s\n" % response.code)
    sys.stderr.write("Headers: %s\n" % response.headers)
    sys.stderr.write("Msg: %s\n" % response.msg)
    sys.stderr.write("Url: %s\n" % response.url)
  return response

checkRequiredArguments()

def ws_sso_content_reader.checkRequiredArguments	(		opts,
			parser
	)

Definition at line 101 of file ws_sso_content_reader.py.

References str.

def checkRequiredArguments(opts, parser):
  missing_options = []
  for option in parser.option_list:
    if re.match(r'^\[REQUIRED\]', option.help) and eval('opts. %s' % option.dest) == None:
      missing_options.extend(option._long_opts)
    if len(missing_options) > 0:
      parser.error('Missing REQUIRED parameters: %s' % str(missing_options))    

getContent()

def ws_sso_content_reader.getContent	(		target_url,
			cert_path,
			key_path,
			post_data = None,
			debug = False,
			adfslogin = None
	)

Definition at line 65 of file ws_sso_content_reader.py.

References _getResponse(), getParentURL(), getResponseContent(), getSSOCookie(), print(), fileinputsource_cfi.read, and str.

def getContent(target_url, cert_path, key_path, post_data=None, debug=False, adfslogin=None):
  opener = urllib2.build_opener(urllib2.HTTPSHandler())
  if adfslogin:
    opener.addheaders = [('Adfs-Login', adfslogin)] #local version of tc test
  
  #try to access the url first
  try:
    content = getResponseContent(opener, target_url, post_data, debug)
    if not 'Sign in with your CERN account' in content:
      return content
  except Exception:
    if debug:
      sys.stderr.write("The request has an error, will try to create a new cookie\n")

  cookie = cookielib.CookieJar()
  opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookie), HTTPSClientAuthHandler(key_path, cert_path))  #will use private key and ceritifcate
  if debug:
    sys.stderr.write("The return page is sso login page, will request cookie.")
  hasCookie = False
  # if the access gave an exception, try to get a cookie
  try:
    getSSOCookie(opener, target_url, cookie, debug)
    hasCookie = True 
    result = getResponseContent(opener, target_url, post_data, debug)
  except Exception as e:
    result = ""
    print(sys.stderr.write("ERROR:"+str(e)))
  if hasCookie:
    burl = getParentURL(target_url)
    try:
      _getResponse(opener, burl+"signOut").read()
      _getResponse(opener, "https://login.cern.ch/adfs/ls/?wa=wsignout1.0").read()
    except:
      sys.stderr.write("Error, could not logout correctly from server") 
  return result

getFile()

def ws_sso_content_reader.getFile

(

path

)

Definition at line 7 of file ws_sso_content_reader.py.

def getFile(path):
  npath = os.path.expanduser(path)
  while os.path.islink(npath):
    path = os.readlink(npath)
    if path[0] != "/": path = os.path.join(os.path.dirname(npath),path)
    npath = path
  return npath

getParentURL()

def ws_sso_content_reader.getParentURL

(

url

)

Definition at line 42 of file ws_sso_content_reader.py.

Referenced by getContent(), and getSSOCookie().

def getParentURL(url):
  items = url.split("/")
  return '%s//%s/%s/' % (items[0],items[2],items[3])

getResponseContent()

def ws_sso_content_reader.getResponseContent	(		opener,
			url,
			post_data = None,
			debug = False
	)

Definition at line 36 of file ws_sso_content_reader.py.

References _getResponse(), and fileinputsource_cfi.read.

Referenced by getContent(), and getSSOCookie().

def getResponseContent(opener, url, post_data=None, debug=False):
  return _getResponse(opener, url, post_data, debug).read()

getResponseURL()

def ws_sso_content_reader.getResponseURL	(		opener,
			url,
			post_data = None,
			debug = False
	)

Definition at line 39 of file ws_sso_content_reader.py.

References _getResponse().

Referenced by getSSOCookie().

def getResponseURL(opener, url, post_data=None, debug=False):
  return urllib2.unquote(_getResponse(opener, url, post_data, debug).url)

getSSOCookie()

def ws_sso_content_reader.getSSOCookie	(		opener,
			target_url,
			cookie,
			debug = False
	)

Definition at line 46 of file ws_sso_content_reader.py.

References getParentURL(), getResponseContent(), and getResponseURL().

Referenced by getContent().

def getSSOCookie(opener, target_url, cookie, debug=False):
  opener.addheaders = [('User-agent', 'curl-sso-certificate/0.0.2')] #in sync with cern-get-sso-cookie tool
  url = getResponseURL(opener, getParentURL(target_url), debug=debug)
  content = getResponseContent(opener, url, debug=debug)
  ret = re.search('<form .+? action="(.+?)">', content)
  if ret == None:
    raise Exception("error: The page doesn't have the form with adfs url, check 'User-agent' header")
  url = urllib2.unquote(ret.group(1))
  h = HTMLParser.HTMLParser()
  post_data_local = ''
  for match in re.finditer('input type="hidden" name="([^"]*)" value="([^"]*)"', content):
    post_data_local += "&%s=%s" % (match.group(1), urllib.quote(h.unescape(match.group(2))))
    is_link_found = True
  
  if not is_link_found:
    raise Exception("error: The page doesn't have the form with security attributes, check 'User-agent' header")
  post_data_local = post_data_local[1:] #remove first &
  getResponseContent(opener, url, post_data_local, debug)

Variable Documentation

action

ws_sso_content_reader.action

Definition at line 111 of file ws_sso_content_reader.py.

args

ws_sso_content_reader.args

Definition at line 116 of file ws_sso_content_reader.py.

content

ws_sso_content_reader.content

Definition at line 118 of file ws_sso_content_reader.py.

default

ws_sso_content_reader.default

Definition at line 111 of file ws_sso_content_reader.py.

dest

ws_sso_content_reader.dest

Definition at line 111 of file ws_sso_content_reader.py.

help

ws_sso_content_reader.help

Definition at line 111 of file ws_sso_content_reader.py.

opts

ws_sso_content_reader.opts

Definition at line 116 of file ws_sso_content_reader.py.

parser

ws_sso_content_reader.parser

Definition at line 110 of file ws_sso_content_reader.py.

usage

ws_sso_content_reader.usage

Definition at line 110 of file ws_sso_content_reader.py.