import re from os import sep from typing import Optional import qrcode import string from bilibili_api import Credential from bilibili_api.audio import Audio from bilibili_api.video import Video from bilibili_api.user import User from pyrogram import ContinuePropagation from qrcode.image.pil import PilImage from io import BytesIO from PIL import Image, ImageDraw, ImageFont from defs.cookie import get_bili_cookie, get_bili_browser_cookie, set_bili_cookie from defs.browser import get_browser from init import request def from_cookie_get_credential() -> Optional[Credential]: """ 从 cookie 中获取 Credential 对象。 Returns: Credential: Credential 对象。 """ cookie = get_bili_cookie() try: sessdata = cookie["SESSDATA"] bili_jct = cookie["bili_jct"] buvid3 = cookie["buvid3"] dedeuserid = cookie["DedeUserID"] except KeyError: return None try: ac_time_value = cookie["ac_time_value"] except KeyError: ac_time_value = None return Credential(sessdata, bili_jct, buvid3, dedeuserid, ac_time_value) credential = from_cookie_get_credential() def set_cookie_from_cred(new_cred: Credential) -> None: cookie = get_bili_cookie() cookie["SESSDATA"] = new_cred.sessdata cookie["bili_jct"] = new_cred.bili_jct cookie["buvid3"] = new_cred.buvid3 cookie["ac_time_value"] = new_cred.ac_time_value set_bili_cookie(cookie) async def check_and_refresh_credential() -> None: """ 检查并刷新 Credential 对象。 """ if await credential.check_refresh(): await credential.refresh() set_cookie_from_cred(credential) def cut_text(old_str, cut): """ :说明: `get_cut_str` > 长文本自动换行切分 :参数: * `str: [type]`: 文本 * `cut: [type]`: 换行宽度，需要至少大于20 :返回: - `List`: 切分后的文本列表 """ punc = """，,、。.？?）》】“"‘'；;：:！!·`~%^& """ si = 0 i = 0 next_str = old_str str_list = [] for s in next_str: if s in string.printable: si += 1 else: si += 2 i += 1 if next_str[0] == "\n": next_str = next_str[1:] elif s == "\n": str_list.append(next_str[: i - 1]) next_str = next_str[i - 1 :] si = 0 i = 0 continue if si > cut: try: if next_str[i] in punc: i += 1 if next_str[i] in string.ascii_letters: for j in range(i, i - 18, -1): if next_str[j] == " ": i = j + 1 break except IndexError: str_list.append(next_str) return str_list str_list.append(next_str[:i]) next_str = next_str[i:] si = 0 i = 0 str_list.append(next_str) i = 0 non_wrap_str = [] for p in str_list: if not p: continue if p[-1] == "\n": p = p[:-1] non_wrap_str.append(p) i += 1 return non_wrap_str async def b23_extract(text): b23 = re.compile(r"b23.tv\\/(\w+)").search(text) if not b23: b23 = re.compile(r"b23.tv/(\w+)").search(text) try: assert b23 is not None url = f"https://b23.tv/{b23[1]}" except AssertionError: raise ContinuePropagation resp = await request.head(url, follow_redirects=True) r = str(resp.url) return r def create_video(cid) -> Optional[Video]: v = None if cid[:2] == "av": v = Video(aid=int(cid[2:]), credential=credential) elif cid[:2] == "BV": v = Video(bvid=cid, credential=credential) return v def create_audio(aid: str) -> Optional[Audio]: a = None if aid[:2] == "au": a = Audio(auid=int(aid[2:]), credential=credential) return a async def video_info_get(cid): v = create_video(cid) if not v: return None video_info = await v.get_info() return video_info def numf(num: int): if num < 10000: view = str(num) elif num < 100000000: view = ("%.2f" % (num / 10000)) + "万" else: view = ("%.2f" % (num / 100000000)) + "亿" return view async def binfo_image_create(video_info: dict): bg_y = 0 # 封面 pic_url = video_info["pic"] pic_get = (await request.get(pic_url)).content pic_bio = BytesIO(pic_get) pic = Image.open(pic_bio) pic = pic.resize((560, 350)) pic_time_box = Image.new("RGBA", (560, 50), (0, 0, 0, 150)) pic.paste(pic_time_box, (0, 300), pic_time_box) bg_y += 350 + 20 # 时长 minutes, seconds = divmod(video_info["duration"], 60) hours, minutes = divmod(minutes, 60) video_time = f"{hours:02d}:{minutes:02d}:{seconds:02d}" tiem_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-bold.ttf", 30 ) draw = ImageDraw.Draw(pic) draw.text((10, 305), video_time, "white", tiem_font) # 分区 tname = video_info["tname"] _, _, tname_x, _ = tiem_font.getbbox(tname) draw.text((560 - tname_x - 10, 305), tname, "white", tiem_font) # 标题 title = video_info["title"] title_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-bold.ttf", 25 ) title_cut_str = "\n".join(cut_text(title, 40)) _, _, _, title_text_y = draw.multiline_textbbox((0, 0), title_cut_str, title_font) title_bg = Image.new("RGB", (560, title_text_y + 23), "#F5F5F7") draw = ImageDraw.Draw(title_bg) draw.text((15, 10), title_cut_str, "black", title_font) title_bg_y = title_bg.size[1] bg_y += title_bg_y # 简介 dynamic = "该视频没有简介" if video_info["desc"] == "" else video_info["desc"] dynamic_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-semibold.ttf", 18 ) dynamic_cut_str = "\n".join(cut_text(dynamic, 58)) _, _, _, dynamic_text_y = draw.multiline_textbbox( (0, 0), dynamic_cut_str, dynamic_font ) dynamic_bg = Image.new("RGB", (560, dynamic_text_y + 24), "#F5F5F7") draw = ImageDraw.Draw(dynamic_bg) draw.rectangle((0, 0, 580, dynamic_text_y + 24), "#E1E1E5") draw.text((10, 10), dynamic_cut_str, "#474747", dynamic_font) dynamic_bg_y = dynamic_bg.size[1] bg_y += dynamic_bg_y # 视频数据 icon_font = ImageFont.truetype(f"resources{sep}font{sep}vanfont.ttf", 46) icon_color = (247, 145, 185) info_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-bold.ttf", 26 ) view = numf(video_info["stat"]["view"]) # 播放 \uE6E6 danmaku = numf(video_info["stat"]["danmaku"]) # 弹幕 \uE6E7 favorite = numf(video_info["stat"]["favorite"]) # 收藏 \uE6E1 coin = numf(video_info["stat"]["coin"]) # 投币 \uE6E4 like = numf(video_info["stat"]["like"]) # 点赞 \uE6E0 info_bg = Image.new("RGB", (560, 170), "#F5F5F7") draw = ImageDraw.Draw(info_bg) draw.text((5 + 10, 20), "\uE6E0", icon_color, icon_font) draw.text((5 + 64, 27), like, "#474747", info_font) draw.text((5 + 10 + 180, 20), "\uE6E4", icon_color, icon_font) draw.text((5 + 64 + 180, 27), coin, "#474747", info_font) draw.text((5 + 10 + 180 + 180, 20), "\uE6E1", icon_color, icon_font) draw.text((5 + 64 + 180 + 180, 27), favorite, "#474747", info_font) draw.text((5 + 100, 93), "\uE6E6", icon_color, icon_font) draw.text((5 + 154, 100), view, "#474747", info_font) draw.text((5 + 100 + 210, 93), "\uE6E7", icon_color, icon_font) draw.text((5 + 154 + 210, 100), danmaku, "#474747", info_font) info_bg_y = info_bg.size[1] bg_y += info_bg_y # UP主 # 等级 0-4 \uE6CB-F 5-6\uE6D0-1 # UP \uE723 if "staff" in video_info: up_list = [] for up in video_info["staff"]: up_mid = up["mid"] u = User(up_mid, credential=credential) up_data = await u.get_user_info() up_list.append( { "name": up["name"], "up_title": up["title"], "face": up["face"], "color": up_data["vip"]["nickname_color"] if up_data["vip"]["nickname_color"] != "" else "black", "follower": up["follower"], "level": up_data["level"], } ) else: up_mid = video_info["owner"]["mid"] u = User(up_mid, credential=credential) up_data = await u.get_user_info() up_stat = await u.get_relation_info() up_list = [ { "name": up_data["name"], "up_title": "UP主", "face": up_data["face"], "color": up_data["vip"]["nickname_color"] if up_data["vip"]["nickname_color"] != "" else "black", "follower": up_stat["follower"], "level": up_data["level"], } ] up_num = len(up_list) up_bg = Image.new("RGB", (560, 20 + (up_num * 120) + 20), "#F5F5F7") draw = ImageDraw.Draw(up_bg) face_size = (80, 80) mask = Image.new("RGBA", face_size, color=(0, 0, 0, 0)) mask_draw = ImageDraw.Draw(mask) mask_draw.ellipse( (0, 0, face_size[0], face_size[1]), fill=(0, 0, 0, 255) # type: ignore ) name_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-bold.ttf", 24 ) up_title_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-bold.ttf", 20 ) follower_font = ImageFont.truetype( f"resources{sep}font{sep}sarasa-mono-sc-semibold.ttf", 22 ) i = 0 for up in up_list: if up["level"] == 0: up_level = "\uE6CB" level_color = (191, 191, 191) elif up["level"] == 1: up_level = "\uE6CC" level_color = (191, 191, 191) elif up["level"] == 2: up_level = "\uE6CD" level_color = (149, 221, 178) elif up["level"] == 3: up_level = "\uE6CE" level_color = (146, 209, 229) elif up["level"] == 4: up_level = "\uE6CF" level_color = (255, 179, 124) elif up["level"] == 5: up_level = "\uE6D0" level_color = (255, 108, 0) else: up_level = "\uE6D1" level_color = (255, 0, 0) # 头像 face_url = up["face"] face_get = (await request.get(face_url)).content face_bio = BytesIO(face_get) face = Image.open(face_bio) face.convert("RGB") face = face.resize(face_size) up_bg.paste(face, (20, 20 + (i * 120)), mask) # 名字 draw.text((160, 25 + (i * 120)), up["name"], up["color"], name_font) _, _, name_size_x, _ = name_font.getbbox(up["name"]) # 等级 draw.text( (160 + name_size_x + 10, 16 + (i * 120)), up_level, level_color, icon_font ) # 身份 _, _, up_title_size_x, up_title_size_y = up_title_font.getbbox(up["up_title"]) draw.rectangle( ( 60, 10 + (i * 120), 73 + up_title_size_x, 18 + (i * 120) + up_title_size_y, ), "white", icon_color, 3, ) draw.text((67, 13 + (i * 120)), up["up_title"], icon_color, up_title_font) # 粉丝量 draw.text( (162, 66 + (i * 120)), "粉丝 " + numf(up["follower"]), "#474747", follower_font, ) i += 1 up_bg_y = up_bg.size[1] bg_y += up_bg_y # 底部栏 baner_bg = Image.new("RGB", (600, 170), icon_color) draw = ImageDraw.Draw(baner_bg) # 二维码 qr = qrcode.QRCode(border=1) qr.add_data("https://b23.tv/" + video_info["bvid"]) qr_image = qr.make_image(PilImage, fill_color=icon_color, back_color="#F5F5F7") qr_image = qr_image.resize((140, 140)) baner_bg.paste(qr_image, (50, 10)) # Logo # LOGO \uE725 logo_font = ImageFont.truetype(f"resources{sep}font{sep}vanfont.ttf", 100) draw.text((300, 28), "\uE725", "#F5F5F7", logo_font) bg_y += 170 video = Image.new("RGB", (600, bg_y + 40), "#F5F5F7") video.paste(pic, (20, 20)) video.paste(title_bg, (20, 390)) video.paste(dynamic_bg, (20, 390 + title_bg_y + 20)) video.paste(info_bg, (20, 390 + title_bg_y + 20 + dynamic_bg_y + 20)) video.paste(up_bg, (20, 390 + title_bg_y + 20 + dynamic_bg_y + 10 + info_bg_y)) video.paste( baner_bg, (0, 390 + title_bg_y + 20 + dynamic_bg_y + 10 + info_bg_y + up_bg_y) ) picture = BytesIO() video.save(picture, format="PNG") picture.name = "bilibili.png" return picture async def get_dynamic_screenshot_pc(dynamic_id): """电脑端动态截图""" url = f"https://t.bilibili.com/{dynamic_id}" browser = await get_browser() context = await browser.new_context( viewport={"width": 2560, "height": 1080}, device_scale_factor=2, ) await context.add_cookies(get_bili_browser_cookie()) page = await context.new_page() try: await page.goto(url, wait_until="networkidle", timeout=10000) # 动态被删除或者进审核了 if page.url == "https://www.bilibili.com/404": return None card = await page.query_selector(".bili-dyn-item__main") assert card clip = await card.bounding_box() assert clip return await page.screenshot(clip=clip, full_page=True) except Exception: print(f"截取动态时发生错误：{url}") return await page.screenshot(full_page=True) finally: await context.close()