dizzylab.py

import html
import os
import requests
import re

from handlers.message import Message


def search(key: str = 'Static World') -> dict:
    url = 'https://www.dizzylab.net/'
    params = {'s': key}
    res = requests.get(url=os.path.join(url, 'search/'), params=params, timeout=15)
    ans = {'l': [], "d": []}
    if res.status_code != 200:
        return ans

    blks = re.findall(r'<a href="(.+?)">.+? <img src="(.+?)".+?/>.+? <h1>(.+?)</h1>.+? <h3.*?>(.*?)</h3>.+? </a>',
                      res.text, re.M | re.S)
    for blk in blks:
        if re.match(r'/l/.+', blk[0]):
            ans['l'].append({'link': os.path.join(url, blk[0][1:] if blk[0][0] == '/' else blk[0]),
                             'cover': blk[1],
                             'title': blk[2],
                             'summary': blk[3]})
        elif re.match(r'/d/.+', blk[0]):
            ans['d'].append({'link': os.path.join(url, blk[0][1:] if blk[0][0] == '/' else blk[0]),
                             'cover': blk[1],
                             'title': blk[2],
                             'summary': blk[3]})
    return ans


def run(message: Message) -> str:
    help_msg = '小白会试着从 dizzylab 搜索 ´ ▽ ` )ﾉ'
    req = list(message.raw_message.split(' ', 1))
    ans = '好像啥也没有找到诶'
    if len(req) == 1:
        return help_msg

    key = req[1].strip()
    res = search(key)
    key = None
    if len(res['l']):
        key = res['l'][0]
    elif len(res['d']):
        key = res['d'][0]
    if not key:
        return ans

    # ans = f'[CQ:share,url={key["link"]},title={key["title"]}]'
    ans = '<?xml version="1.0" encoding="utf-8"?>' \
          '<msg templateID="12345" action="web" brief="[分享] %s" serviceID="1" url="%s">' \
          '<item layout="2"><title>%s</title>' \
          '<summary>%s</summary>' \
          '<picture cover="%s"/></item>' \
          '<source name="dizzylab" icon="https://cdn.dizzylab.net/static/favicon.ico" action="" appid="-1" /></msg>' \
          % (html.escape(key['title']), html.escape(key["link"]), html.escape(key['title']),
             html.escape(key["summary"]), html.escape(key["cover"]))
    ans = ans.replace(',', '&#44;').replace('&', '&amp;').replace('[', '&#91;').replace(']', '&#93;')
    ans = '[CQ:xml,data=%s]' % ans

    return ans


if __name__ == "__main__":
    print(search())