import asyncio from pathlib import Path from typing import Any import filetype from arclet.alconna import ( Alconna, Args, CommandMeta, MultiVar, Option, Subcommand, TextFormatter, ) from arclet.alconna.exceptions import SpecialOptionTriggered from arclet.alconna.tools import RichConsoleFormatter from meme_generator.app import run_server from meme_generator.config import meme_config from meme_generator.download import check_resources from meme_generator.exception import MemeGeneratorException, NoSuchMeme from meme_generator.log import setup_logger from meme_generator.manager import get_meme, get_memes def construct_parser() -> Alconna: sub_commands: list[Subcommand] = [] for meme in get_memes(): options: list[Option] = [] if args_type := meme.params_type.args_type: for option in args_type.parser_options: options.append(option.option()) sub_command = Subcommand( meme.key, *options, Option( "--images", Args["images", MultiVar(str, "+")], help_text="输入图片路径", ), Option("--texts", Args["texts", MultiVar(str, "+")], help_text="输入文字"), help_text="/".join(meme.keywords), ) sub_commands.append(sub_command) parser = Alconna( "meme", Subcommand("list", alias=["ls"], help_text="查看表情列表"), Subcommand( "info", Args["key#表情名", str], alias=["show"], help_text="查看表情详情" ), Subcommand("preview", Args["key#表情名", str], help_text="生成表情预览"), Subcommand("generate", *sub_commands, alias=["make"], help_text="制作表情"), Subcommand("run", alias=["start"], help_text="启动 web server"), Subcommand( "download", Option("--url", Args["url", str], help_text="指定资源链接"), help_text="下载内置表情图片", ), meta=CommandMeta( description="表情包生成器", example="meme generate petpet --images /path/to/image/file", ), formatter_type=RichConsoleFormatter, ) return parser def list_memes() -> str: memes = sorted(get_memes(), key=lambda meme: meme.key) return "\n".join( f"{i}. {meme.key} ({'/'.join(meme.keywords)})" for i, meme in enumerate(memes, start=1) ) def meme_info(key: str) -> str: try: meme = get_meme(key) except NoSuchMeme: return f'表情 "{key}" 不存在！' keywords = "、".join([f'"{keyword}"' for keyword in meme.keywords]) shortcuts = "、".join( [f'"{shortcut.humanized or shortcut.key}"' for shortcut in meme.shortcuts] ) tags = "、".join([f'"{tag}"' for tag in sorted(meme.tags)]) image_num = f"{meme.params_type.min_images}" if meme.params_type.max_images > meme.params_type.min_images: image_num += f" ~ {meme.params_type.max_images}" text_num = f"{meme.params_type.min_texts}" if meme.params_type.max_texts > meme.params_type.min_texts: text_num += f" ~ {meme.params_type.max_texts}" default_texts = ", ".join([f'"{text}"' for text in meme.params_type.default_texts]) args_info = "" if args_type := meme.params_type.args_type: formater = TextFormatter() for option in args_type.parser_options: opt = option.option() alias_text = ( " ".join(opt.requires) + (" " if opt.requires else "") + "│".join(sorted(opt.aliases, key=len)) ) args_info += ( f"\n * {alias_text}{opt.separators[0]}" f"{formater.parameters(opt.args)} {opt.help_text}" ) return ( f"表情名：{meme.key}\n" + f"关键词：{keywords}\n" + (f"快捷指令：{shortcuts}\n" if shortcuts else "") + (f"标签：{tags}\n" if tags else "") + f"需要图片数目：{image_num}\n" + f"需要文字数目：{text_num}\n" + (f"默认文字：[{default_texts}]\n" if default_texts else "") + (f"其他参数：{args_info}" if args_info else "") ) def generate_meme_preview(key: str) -> str: try: meme = get_meme(key) except NoSuchMeme: return f'表情 "{key}" 不存在！' try: result = meme.generate_preview() content = result.getvalue() ext = filetype.guess_extension(content) filename = f"result.{ext}" with open(filename, "wb") as f: f.write(content) return f'表情制作成功！生成的表情文件为 "{filename}"' except MemeGeneratorException as e: return str(e) def generate_meme( key: str, images: list[str], texts: list[str], args: dict[str, Any] ) -> str: try: meme = get_meme(key) except NoSuchMeme: return f'表情 "{key}" 不存在！' for image in images: if not Path(image).exists(): return f'图片路径 "{image}" 不存在！' try: result = meme(images=images, texts=texts, args=args) content = result.getvalue() ext = filetype.guess_extension(content) filename = f"result.{ext}" with open(filename, "wb") as f: f.write(content) return f'表情制作成功！生成的表情文件为 "{filename}"' except MemeGeneratorException as e: return str(e) def main(): setup_logger() parser = construct_parser() result = parser() if not result.matched: if not isinstance(result.error_info, SpecialOptionTriggered): print(result.error_info) # noqa: T201 return if not result.subcommands: print(parser.get_help()) # noqa: T201 for subcommand, sub_result in result.subcommands.items(): if subcommand == "list": print(list_memes()) # noqa: T201 elif subcommand == "info": key = str(sub_result.args["key"]) print(meme_info(key)) # noqa: T201 elif subcommand == "preview": key = str(sub_result.args["key"]) print(generate_meme_preview(key)) # noqa: T201 elif subcommand == "generate": for key, subsub_result in sub_result.subcommands.items(): images = ( list(subsub_result.options["images"].args["images"]) if "images" in subsub_result.options else [] ) texts = ( list(subsub_result.options["texts"].args["texts"]) if "texts" in subsub_result.options else [] ) options = subsub_result.options options.pop("images", None) options.pop("texts", None) args = {} for option, option_result in options.items(): if option_result.value is None: args.update(option_result.args) else: args[option] = option_result.value print(generate_meme(key, images, texts, args)) # noqa: T201 elif subcommand == "run": run_server() elif subcommand == "download": if "url" in sub_result.options: url = sub_result.options["url"].args["url"] meme_config.resource.resource_url = url loop = asyncio.new_event_loop() loop.run_until_complete(check_resources()) if __name__ == "__main__": main()