From dbfa8d61f660da160e7a82730af244eb2a87bdb8 Mon Sep 17 00:00:00 2001 From: ihmily <961532186@qq.com> Date: Wed, 15 May 2024 21:42:30 +0800 Subject: [PATCH] fix: add absolute config path --- tieba_sign.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tieba_sign.py b/tieba_sign.py index 30e3094..cb970ce 100644 --- a/tieba_sign.py +++ b/tieba_sign.py @@ -1,6 +1,7 @@ import asyncio import datetime import json +import os import re import sys @@ -8,9 +9,11 @@ from bs4 import BeautifulSoup from loguru import logger +script_path = os.path.split(os.path.realpath(sys.argv[0]))[0] + def load_cookie_from_config(): - with open('cookie.json', 'r') as f: + with open(f'{script_path}/cookie.json', 'r') as f: content = f.read() return json.loads(content)['baidu_cookie'] @@ -33,7 +36,7 @@ async def get_like_list(self): url = "https://tieba.baidu.com/f/like/mylike" try: html_str = await self.fetch_url(url) - pn_list = re.findall('', html_str, re.S) + pn_list = re.findall(r'', html_str, re.S) page_num = pn_list[-1] except IndexError: logger.error("cookie为空或者已过期,请在cookie.json文件中填写正确的cookie后重试") @@ -79,4 +82,4 @@ async def sign(self, kw, tbs): asyncio.run(TiebaSigner().get_like_list()) else: loop = asyncio.get_event_loop() - loop.run_until_complete(TiebaSigner().get_like_list()) \ No newline at end of file + loop.run_until_complete(TiebaSigner().get_like_list())