From 458e951d42bca4ed33f30e3b2d475521773a2a05 Mon Sep 17 00:00:00 2001
From: yoshiko2 <wenead99@gmail.com>
Date: Tue, 6 Jul 2021 16:47:41 +0800
Subject: [PATCH] Update source javdb #2

---
 WebCrawler/javdb.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/WebCrawler/javdb.py b/WebCrawler/javdb.py
index c1a72fd..1a81c97 100755
--- a/WebCrawler/javdb.py
+++ b/WebCrawler/javdb.py
@@ -208,18 +208,17 @@ def getSeries(a):
     return str(result1 + result2).strip('+').replace("', '", '').replace('"', '')
 
 def main(number):
-    javdb_site = ["javdb9", "javdb30"]
+    javdb_site = random.choice(["javdb9", "javdb30"])
     try:
         # if re.search(r'[a-zA-Z]+\.\d{2}\.\d{2}\.\d{2}', number).group():
         #     pass
         # else:
         #     number = number.upper()
         number = number.upper()
-        cookie_json = './javdb.json'
+        cookie_json = './' + javdb_site + '.json'
         javdb_cookies = None
         # 不加载过期的cookie，javdb登录界面显示为7天免登录，故假定cookie有效期为7天
         cdays = file_modification_days(cookie_json)
-        javdb_site = random.choice(javdb_site)
         if cdays < 7:
             javdb_cookies = load_cookies(cookie_json)
         elif cdays != 9999: