panjd123 / ruclogin Goto Github PK

class RUC_LOGIN:
...
    def get_cookies(self, domain="v"):
        ...
        elif domain.startswith("jw"):
            [
                self.driver.wait_for_request(
                    url,
                    timeout=10,
                )
                for url in [
                    "/Njw2017/index.html*",
                    "/secService/oauthlogin*",
                ]
            ]
            cookies = {}
            for request in self.driver.requests:
                if request.response:
                    cookie_header = request.response.headers["Set-Cookie"]
                    if cookie_header:
                        for cookie in cookie_header.split(", "):
                            if "=" in cookie and ";" in cookie:
                                cookies.update(
                                    {
                                        cookie.split("=")[0]: cookie.split("=")[1].split(";")[0]
                                    }
                            )
            return cookies