瀏覽代碼

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

HuangYiwei 5 年之前
父節點
當前提交
cecb717c4e
共有 1 個文件被更改,包括 5 次插入4 次删除
  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: