Xtao-Labs/iShotaBot
3
0
Fork 0
mirror of https://github.com/Xtao-Labs/iShotaBot.git synced 2024-11-16 04:35:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

♻️ 重构提取用户信息

Browse Source
This commit is contained in:
xtaodada 2022-10-03 18:04:55 +08:00
parent 288501dbc5
commit 1da600f9d7
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
defs/lofter.py
View File

@ -1,3 +1,4 @@
import contextlib
import os import os
import re import re
@ -90,17 +91,22 @@ async def get_loft(url: str) -> List[LofterItem]:
def parse_loft_user(url: str, content: str): def parse_loft_user(url: str, content: str):
username = urlparse(url).hostname.split(".")[0] username = urlparse(url).hostname.split(".")[0]
soup = BeautifulSoup(content, "lxml") soup = BeautifulSoup(content, "lxml")
name, bio, avatar = "未知用户", "", None # noqa
with contextlib.suppress(Exception):
name = soup.find("title").getText().split("-")[-1].strip()
with contextlib.suppress(Exception):
bio = " - ".join(soup.find("meta", {"name": "Description"}).get("content").split(" - ")[1:])
with contextlib.suppress(Exception):
avatar = soup.find("link", {"rel": "shortcut icon"}).get("href").split("?")[0]
if user := soup.find("div", {"class": "selfinfo"}): if user := soup.find("div", {"class": "selfinfo"}):
avatar = user.find("img").get("src").split("?")[0]
name = user.find("h1").getText() name = user.find("h1").getText()
bio = user.find("div", {"class": "text"}).getText() bio = user.find("div", {"class": "text"}).getText()
return username, avatar, name, bio, soup return username, avatar, name, bio, soup
if user := soup.find("div", {"class": "g-hdc box"}): if user := soup.find("div", {"class": "g-hdc box"}):
avatar = user.find("img").get("src").split("?")[0]
name = user.find("h1").getText() name = user.find("h1").getText()
bio = user.find("p", {"class": "cont"}).getText() bio = user.find("p", {"class": "cont"}).getText()
return username, avatar, name, bio, soup return username, avatar, name, bio, soup
return username, None, "未知用户", "", soup return username, avatar, name, bio, soup
async def get_loft_user(url: str): async def get_loft_user(url: str):