初始化Cookie时判断cookies目录是否存在

LittleBearBond · Dec 21, 2020 · cecb717 · cecb717
1 parent fbccc95
commit cecb717
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/jd_spider_requests.py b/jd_spider_requests.py
@@ -86,10 +86,11 @@ def __seckill(self):
 
     def _load_cookies(self):
         cookies_file = ''
-        for name in os.listdir('./cookies'):
-            if name.endswith('.cookies'):
-                cookies_file = './cookies/{0}'.format(name)
-                break
+        if os.path.exists('./cookies'):
+            for name in os.listdir('./cookies'):
+                if name.endswith('.cookies'):
+                    cookies_file = './cookies/{0}'.format(name)
+                    break
         if cookies_file == '':
             return
         with open(cookies_file, 'rb') as f: