Skip to content

Commit

Permalink
fix: add absolute config path
Browse files Browse the repository at this point in the history
  • Loading branch information
ihmily committed May 15, 2024
1 parent d6e65ea commit dbfa8d6
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions tieba_sign.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,19 @@
import asyncio
import datetime
import json
import os
import re
import sys

import httpx
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']

Expand All @@ -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('<a href="/f/like/mylike\?&pn=(.*?)\">', html_str, re.S)
pn_list = re.findall(r'<a href="/f/like/mylike\?&pn=(.*?)\">', html_str, re.S)
page_num = pn_list[-1]
except IndexError:
logger.error("cookie为空或者已过期,请在cookie.json文件中填写正确的cookie后重试")
Expand Down Expand Up @@ -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())
loop.run_until_complete(TiebaSigner().get_like_list())

0 comments on commit dbfa8d6

Please sign in to comment.