文件:WPVG Newsletter 202201 Feature Graph 1 (zh-cn).png

"""Created by [[User:SuperGrey]] with modifications."""

from collections import OrderedDict
from collections.abc import Mapping
from typing import Literal, NamedTuple, TypedDict

from matplotlib import pyplot as plt
from matplotlib.lines import Line2D

# For drawing figures.
FIG_WIDTH = 10  # hpx
FIG_HEIGHT = 5  # hpx
MAGIC_PERCENTAGE = 0.05  # Don't know how it works.
FONT_MAPPING = {
    "tw": "Source Han Sans TC",
    "hk": "Source Han Sans HC",
    "cn": "Source Han Sans SC",
}

type ChineseVariants = Literal["cn", "hk", "tw"]


class GradeConfig(NamedTuple):
    """Configuration for an article quality grade."""

    hant_name: str
    hans_name: str
    color: str


class GradeInfo(NamedTuple):
    """Information about a specific article quality grade."""

    name: str
    color: str
    count: int


class FigureConfigDict(TypedDict):
    """Configuration for generating a figure."""

    data: Mapping[str, list[str | int]]
    header: str
    path: str
    chinese_font: str


GRADES = OrderedDict(
    [
        ("fa", GradeConfig("典範級", "典范级", "#9CBDFF")),
        ("fl", GradeConfig("特色列表級", "特色列表级", "#9CBDFF")),
        ("ga", GradeConfig("優良級", "优良级", "#66FF66")),
        ("b", GradeConfig("乙級", "乙级", "#B2FF66")),
        ("bl", GradeConfig("乙級列表級", "乙级列表级", "#B2FF66")),
        ("c", GradeConfig("丙級", "丙级", "#FFFF66")),
        ("cl", GradeConfig("丙級列表級", "丙级列表级", "#FFFF66")),
        ("start", GradeConfig("初級", "初级", "#FFAA66")),
        ("list", GradeConfig("列表級", "列表级", "#C7B1FF")),
        ("stub", GradeConfig("小作品級", "小作品级", "#FFA4A4")),
        ("sl", GradeConfig("小列表級", "小列表级", "#FFA4A4")),
        ("unassessed", GradeConfig("未評級", "未评级", "#EEEEEE")),
    ],
)


def handle_data(
    stat: Mapping[str, int | None],
    /,
    variant: ChineseVariants,
) -> dict[str, list[int] | list[str]]:
    """Pre-process statistical data for visualization."""

    counts = []
    colors = []
    names = []
    for code, grade in GRADES.items():
        try:
            count = stat[code]
        except KeyError:
            continue
        counts.append(count)
        name = grade.hans_name if variant == "cn" else grade.hant_name
        names.append(name)
        color = grade.color
        colors.append(color)

    total = sum(counts)
    labels, explodes = [], []
    for count in counts:
        label, explode = "", 0
        if (percentage := count / total) > MAGIC_PERCENTAGE:
            label = f"{percentage * 100:.1f}%"
            explode = MAGIC_PERCENTAGE
        labels.append(label)
        explodes.append(explode)

    return {
        "names": names,
        "counts": counts,
        "labels": labels,
        "colors": colors,
        "explodes": explodes,
    }


def handle_config(
    stat: Mapping[str, int],
    variant: ChineseVariants,
    /,
    path_pattern: str,
    hant_header: str,
    hans_header: str,
) -> FigureConfigDict:
    """Create a figure configuration."""

    header = hans_header if variant == "cn" else hant_header
    data = handle_data(stat, variant)
    path = path_pattern.format(variant)
    chinese_font = FONT_MAPPING[variant]
    return FigureConfigDict(
        data=data, header=header, path=path, chinese_font=chinese_font
    )


def save_figure(config: FigureConfigDict) -> None:
    """Generate and save a pie chart as a png file."""

    data = config["data"]
    plt.rc(
        "font",
        family=(
            "DejaVu Sans",
            config["chinese_font"],
        ),
    )

    _, ax = plt.subplots(figsize=(FIG_WIDTH, FIG_HEIGHT))
    ax.pie(
        data["counts"],
        colors=data["colors"],
        explode=data["explodes"],
        labels=data["labels"],
        labeldistance=0.5,
        startangle=90,
        textprops={"rotation_mode": "anchor", "ha": "center", "va": "center"},
    )

    legend_data = zip(
        data["names"],
        data["colors"],
        data["counts"],
        strict=False,
    )
    legend_handles = [
        Line2D(
            [0],
            [0],
            marker="o",
            color="w",
            label=f"{d[0]}：{d[2]}",  # noqa: RUF001
            markersize=10,
            markerfacecolor=d[1],
        )
        for d in legend_data
    ]
    ax.legend(
        handles=legend_handles,
        loc="upper left",
        bbox_to_anchor=(0.95, 0.8),
        frameon=False,
    )
    plt.figtext(
        x=0.705,
        y=0.76,
        s=config["header"],
        fontsize=12,
        verticalalignment="top",
        fontweight="bold",
        backgroundcolor="white",
    )
    plt.savefig(
        config["path"],
        dpi=300,
        bbox_inches="tight",
    )


def mian() -> None:
    """As you know, the function name is correct."""

    # 修改下面的资料
    info = {
        "hant_header": "2021年年終條目品質概況",
        "hans_header": "2021年年终条目品质概况",
        "path_pattern": "WPVG Newsletter 202201 Feature Graph 1 (zh-{}).png",
    }
    data = {
        "fa": 14,
        "fl": 9,
        "ga": 118,
        "b": 140,
        "bl": 5,
        "c": 770,
        "cl": 25,
        "start": 3_328,
        "list": 319,
        "stub": 5_410,
        "unassessed": 309,
    }
    # 修改上面的资料

    for variant in ("tw", "hk", "cn"):
        config = handle_config(
            data,
            variant,
            path_pattern=info["path_pattern"],
            hant_header=info["hant_header"],
            hans_header=info["hans_header"],
        )
        save_figure(config)


if __name__ == "__main__":
    mian()