[FIX] 修复了因为路径和数据格式问题导致的无法登录,及无法获取笔记列表的问题 (#39)

* [FIX] 修复了因为路径和数据格式问题导致的无法登录,及无法获取笔记列表的问题

Author: HugoFeng

main.py

@@ -126,7 +126,7 @@ def save_cookie(self, cookie):
         logger.info(self.cookies)
 
     def dump_cookies(self):
-        with open(self.cache_path + "./Cookies", "w") as f:
+        with open(os.path.join(self.cache_path, "Cookies"), "w") as f:
             f.write(";".join([f"{k}:{v}" for k, v in self.cookies.items()]))
 
     def load_cookies(self):

requirements.txt

@@ -1,2 +1,3 @@
 pyside6~=6.5.0
 requests~=2.26.0
+loguru

wereader.py

@@ -132,8 +132,10 @@ def get_bookshelf(cookies):
         data = r.json()
     else:
         raise Exception(r.text)
+    finishReadBooks = [b for b in data["finishReadBooks"] if 'bookId' in b]
+    recentBooks = [b for b in data["recentBooks"] if 'bookId' in b]
     books = set()
-    for book in chain(data["finishReadBooks"], data["recentBooks"]):
+    for book in chain(finishReadBooks, recentBooks):
         if not book["bookId"].isdigit():  # 过滤公众号
             continue
         try: