From cbb98870203c3f4885f6e417fecdd4eb425a62cd Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Tue, 8 Dec 2020 20:59:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9Bcookies=5Fto=5Fdriver()?=
 =?UTF-8?q?=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91=E3=80=82=E5=9F=9F=E4=B8=BA?=
 =?UTF-8?q?=E7=A9=BA=E7=9A=84cookie=E4=BC=A0=E7=BB=99driver=E6=97=B6?=
 =?UTF-8?q?=E8=87=AA=E5=8A=A8=E6=B7=BB=E5=8A=A0=E5=9F=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/drission.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/DrissionPage/drission.py b/DrissionPage/drission.py
index e305a54..dca4931 100644
--- a/DrissionPage/drission.py
+++ b/DrissionPage/drission.py
@@ -245,9 +245,21 @@ class Drission(object):
         :param url: 作用域
         :return: None
         """
-        url = extract(url)
-        domain = f'{url.domain}.{url.suffix}'
-        cookies = tuple(x for x in self.session.cookies if domain in x.domain)
+        browser_domain = extract(self.driver.current_url).fqdn
+        ex_url = extract(url)
+
+        if ex_url.fqdn not in browser_domain:
+            self.driver.get(url)
+
+        domain = f'{ex_url.domain}.{ex_url.suffix}'
+
+        cookies = []
+        for cookie in self.session.cookies:
+            if cookie.domain == '':
+                cookie.domain = domain
+
+            if domain in cookie.domain:
+                cookies.append(cookie)
 
         self.set_cookies(cookies, set_driver=True)