BiliBiliRankingScripts/unicode.py at main · Neutralization/BiliBiliRankingScripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# -*- coding: utf-8 -*-

from os import remove
from os.path import abspath

from PIL import Image
from selenium.webdriver import Chrome
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service as ChromeService
from webdriver_manager.chrome import ChromeDriverManager

from constant import SEGOE_UI_EMOJI, STYUAN


def text2img(browser, name, text, font, emoji, color, size):
    html_content = f"""<html>
        <head>
            <style type="text/css">
                @font-face {{
                    font-family: "Custom";
                    src: url("{font}");
                }}
                @font-face {{
                    font-family: "Emoji";
                    src: url("{emoji}");
                }}
                body {{
                    font-family: Custom, Emoji, Segoe UI, Segoe UI Historic, Segoe UI Emoji, sans-serif;
                    font-size: {size}px;
                    overflow: hidden;
                    line-height: 500px;
                    white-space: nowrap;
                    text-overflow: ellipsis;
                    text-align: left;
                    color: {color};
                    margin: 0px;
                    padding: 10px;
                    display: block;
                }}
            </style>
        </head>
        <body>
            <p>{text}</p>
        </body>
    </html>"""

    with open("TEXT.html", "w", encoding="utf-8-sig") as f:
        f.write(html_content)

    browser.get(f'file://{abspath("TEXT.html")}')
    print(f"./{name}.png")
    browser.save_screenshot(f"./{name}.png")


def crop(name):
    img = Image.open(f"./{name}.png")
    x, y = img.size
    i = 0
    while i <= x:
        if sum([img.getpixel((i, k))[-1] for k in range(y)]) != 0:
            break
        i += 1
    j = x - 1
    while j >= 0:
        if sum([img.getpixel((j, k))[-1] for k in range(y)]) != 0:
            break
        j -= 1
    img = img.crop((i, 0) + (j + 1, y))
    img.save(f"./{name}.png")


def main(filename, content, font, color, size):
    browser_options = Options()
    browser_options.add_argument("--headless")
    browser_options.add_argument("--window-size=2560,500")
    browser_options.add_argument("--window-position=-2400,-2400")
    browser = Chrome(
        service=ChromeService(ChromeDriverManager().install()), options=browser_options
    )
    browser.execute_cdp_cmd(
        "Emulation.setDefaultBackgroundColorOverride",
        {"color": {"r": 0, "g": 0, "b": 0, "a": 0}},
    )

    emoji_font = abspath(SEGOE_UI_EMOJI).replace("\\", "/")
    text_font = abspath(font).replace("\\", "/")
    text2img(browser, filename, content, text_font, emoji_font, color, size)
    crop(filename)

    remove("./TEXT.html")
    browser.quit()


if __name__ == "__main__":
    main(
        "test",
        "【𝟒𝐊 𝟔𝟎𝐅𝐏𝐒】这首《𝑭𝒂𝒍𝒍𝒊𝒏𝒈 𝑨𝒈𝒂𝒊𝒏》如今治愈了多少人！！! ℳ₯㎕-沉 沦",
        STYUAN,
        "#6D4B2B",
        54,
    )