Parcourir la source

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

HuangYiwei il y a 5 ans
Parent
commit
cecb717c4e
1 fichiers modifiés avec 5 ajouts et 4 suppressions
  1. 5 4
      jd_spider_requests.py

+ 5 - 4
jd_spider_requests.py

@@ -86,10 +86,11 @@ class JdSeckill(object):
 
     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: