PamGram/modules/wiki/material.py

import re
from typing import List, Optional, Tuple, Union

from bs4 import BeautifulSoup
from httpx import URL

from modules.wiki.base import HONEY_HOST, WikiModel

__all__ = ["Material"]

WEEKDAYS = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"]


class Material(WikiModel):
    # noinspection PyUnresolvedReferences
    """武器、角色培养素材

    Attributes:
        type: 类型
        weekdays: 每周开放的时间
        source: 获取方式
        description: 描述
    """
    type: str
    source: Optional[List[str]] = None
    weekdays: Optional[List[int]] = None
    description: str

    @staticmethod
    def scrape_urls() -> List[URL]:
        weapon = [HONEY_HOST.join(f"fam_wep_{i}/?lang=CHS") for i in ["primary", "secondary", "common"]]
        talent = [HONEY_HOST.join(f"fam_talent_{i}/?lang=CHS") for i in ["book", "boss", "common", "reward"]]
        return weapon + talent

    @classmethod
    async def get_name_list(cls, *, with_url: bool = False) -> List[Union[str, Tuple[str, URL]]]:
        return list(sorted(set(await super(Material, cls).get_name_list(with_url=with_url)), key=lambda x: x[0]))

    @classmethod
    async def _parse_soup(cls, soup: BeautifulSoup) -> "Material":
        """解析突破素材页"""
        soup = soup.select(".wp-block-post-content")[0]
        tables = soup.find_all("table")
        table_rows = tables[0].find_all("tr")

        def get_table_row(target: str):
            """一个便捷函数，用于返回对应表格头的对应行的最后一个单元格中的文本"""
            for row in table_rows:
                if target in row.find("td").text:
                    return row.find_all("td")[-1]

        def get_table_text(row_num: int) -> str:
            """一个便捷函数，用于返回表格对应行的最后一个单元格中的文本"""
            return table_rows[row_num].find_all("td")[-1].text.replace("\xa0", "")

        id_ = re.findall(r"/img/(.*?)\.webp", str(table_rows[0]))[0]
        name = get_table_text(0)
        rarity = len(table_rows[3].find_all("img"))
        type_ = get_table_text(1)
        if (item_source := get_table_row("Item Source")) is not None:
            item_source = list(
                # filter 在这里的作用是过滤掉为空的数据
                filter(lambda x: x, item_source.encode_contents().decode().split("<br/>"))
            )
        if (alter_source := get_table_row("Alternative Item")) is not None:
            alter_source = list(
                # filter 在这里的作用是过滤掉为空的数据
                filter(lambda x: x, alter_source.encode_contents().decode().split("<br/>"))
            )
        source = list(sorted(set((item_source or []) + (alter_source or []))))
        if (weekdays := get_table_row("Weekday")) is not None:
            weekdays = [*(WEEKDAYS.index(weekdays.text.replace("\xa0", "").split(",")[0]) + 3 * i for i in range(2)), 6]
        description = get_table_text(-1)
        return Material(
            id=id_, name=name, rarity=rarity, type=type_, description=description, source=source, weekdays=weekdays
        )

    @property
    def icon(self) -> str:
        return str(HONEY_HOST.join(f"/img/{self.id}.webp"))