class RobobrowserSession:
def __init__(self, login=False, cookies=None):
self.driver = Spacelab(
parser='html.parser', history=True,
allow_redirects=True, tries=5, multiplier=1
)
if login:
self.login()
if cookies:
self.set_cookies(cookies)
def login(self):
self.driver.visit(login_url)
self.driver.get_form_and_submit(username=credentials['username'],
password=credentials['password'])
assert self.status_code == 200
assert self.driver.url == memberships_url
@property
def cookies(self):
return self.driver.session.cookies
def get_cookies(self):
return self.cookies
def set_cookies(self, cookies):
print(len(cookies))
for cookie in cookies:
print(cookie)
# if not isinstance(cookies, CookieJar):
cookies = CookieConverter.to_requests(cookies)
for cookie in cookies:
print(cookie)
self.cookies.update(cookie)
print(len(self.cookies))
# for cookie in cookies:
# self.driver.session.cookies.update(merge_cookies(self.get_cookies(),
# cookies))
@property
def status_code(self):
return self.driver.response.status_code
@property
def url(self):
return self.driver.url
def __init__(self, login=False, cookies=None):
self.driver = Spacelab(
parser='html.parser', history=True,
allow_redirects=True, tries=5, multiplier=1
)
if login:
self.login()
if cookies:
self.set_cookies(cookies)
