mirror of
https://0xacab.org/johnxina/rat.git
synced 2024-12-23 13:09:08 +00:00
34 lines
918 B
Python
34 lines
918 B
Python
|
'''Extra APIs'''
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import bs4
|
||
|
|
import re
|
||
|
|
|
||
|
|
async def find_tieba_info(tname):
|
||
|
|
"""Get the tiebat avatar for the forum name.
|
||
|
|
|
||
|
|
:param tname: the name of the target forum.
|
||
|
|
:returns: the internal ID of the corresponding avatar.
|
||
|
|
|
||
|
|
"""
|
||
|
|
info = { 'name': tname }
|
||
|
|
|
||
|
|
res = requests.get('https://tieba.baidu.com/f', params={'kw': tname})
|
||
|
|
soup = bs4.BeautifulSoup(res.text, 'html.parser')
|
||
|
|
|
||
|
|
elems = soup.select('#forum-card-head')
|
||
|
|
match = re.search(r'/(\w+)\.jpg', elems[0]['src'])
|
||
|
|
|
||
|
|
info['avatar'] = match.group(1) + '.jpg'
|
||
|
|
|
||
|
|
footer = soup.select('.th_footer_l')[0]
|
||
|
|
stat_elems = footer.findAll('span', {'class': 'red_text'}, recursive=False)
|
||
|
|
stats = list(map(lambda x: int(x.text), stat_elems))
|
||
|
|
|
||
|
|
info |= { 'topic': stats[0], 'thread': stats[1], 'member': stats[2] }
|
||
|
|
|
||
|
|
slogan = soup.select('.card_slogan')[0]
|
||
|
|
info['desc'] = slogan.text
|
||
|
|
|
||
|
|
return info
|