jenglamlow · June 1, 2016 02:46
diff --git a/cookie_store.py b/cookie_store.py
 import pickle
 import os

 def save_cookies(requests_cookiejar, filename):
    with open(filename, 'wb') as f:
        pickle.dump(requests_cookiejar, f)

 def load_cookies(filename):
    with open(filename, 'rb') as f:
        return pickle.load(f)

 def auth()
    Check whether cookie file exist
    filename = 'cookie'
    if os.path.isfile(filename):
        # Use existing cookie file as credential
        cookies = load_cookies(filename)
        r = requests.get(self._url.is_auth(),
                         headers=headers,
                         auth=user_pass,
                         cookies=cookies)

        if r.status_code == 401:
            print('Creating Cookie')
            self.auth()
            self._save_cookies(self._cookies, filename)
        elif r.status_code == 200:
            print("Using Existing Cookie")
        else:
            err = "\Is_Auth ALM Session failed: " + \
                self._url.is_auth() + '\n'
            err = err + "Status Code: " + str(r.status_code) + '\n'
            err = err + "Reason: " + r.reason + '\n'
            raise RuntimeError(err)
    else:
        print('Creating Cookie')
        self.auth()
        self._save_cookies(self._cookies, filename)
	import pickle
	import os

	def save_cookies(requests_cookiejar, filename):
	with open(filename, 'wb') as f:
	pickle.dump(requests_cookiejar, f)

	def load_cookies(filename):
	with open(filename, 'rb') as f:
	return pickle.load(f)

	def auth()
	Check whether cookie file exist
	filename = 'cookie'
	if os.path.isfile(filename):
	# Use existing cookie file as credential
	cookies = load_cookies(filename)
	r = requests.get(self._url.is_auth(),
	headers=headers,
	auth=user_pass,
	cookies=cookies)

	if r.status_code == 401:
	print('Creating Cookie')
	self.auth()
	self._save_cookies(self._cookies, filename)
	elif r.status_code == 200:
	print("Using Existing Cookie")
	else:
	err = "\Is_Auth ALM Session failed: " + \
	self._url.is_auth() + '\n'
	err = err + "Status Code: " + str(r.status_code) + '\n'
	err = err + "Reason: " + r.reason + '\n'
	raise RuntimeError(err)
	else:
	print('Creating Cookie')
	self.auth()
	self._save_cookies(self._cookies, filename)
No results found