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)