1#!/usr/bin/env python
 2# -*- coding: utf-8 -*-
 3
 4import zhconv
 5from pyrogram.client import Client
 6from pyrogram.types import Message
 7
 8from config import PREFIX
 9from messages.sender import send2tg
10from messages.utils import equal_prefix, startswith_prefix
11
12HELP = f"""🔤**简繁转换**
13`{PREFIX.CONVERT_TO_SC}` + 文字 或 消息: 转换为简体
14`{PREFIX.CONVERT_TO_TC}` + 文字 或 消息: 转换为繁体
15"""
16
17
18async def chinese_conversion(client: Client, message: Message, **kwargs):
19    """Traditional Chinese <-> Simplified Chinese."""
20    # send docs, without reply
21    if equal_prefix(message.content, prefix=[PREFIX.CONVERT_TO_SC, PREFIX.CONVERT_TO_TC]) and not message.reply_to_message:
22        await send2tg(client, message, texts=HELP, **kwargs)
23        return
24    if not startswith_prefix(message.content, prefix=[PREFIX.CONVERT_TO_SC, PREFIX.CONVERT_TO_TC]):
25        return
26    if startswith_prefix(message.content, prefix=PREFIX.CONVERT_TO_SC):
27        texts = message.reply_to_message.content if message.reply_to_message else message.content.removeprefix(PREFIX.CONVERT_TO_SC).lstrip()
28        simplified_chinese = zhconv.convert(texts, "zh-cn")
29        await message.reply(f"🔄繁体转简体:\n{simplified_chinese}", quote=True)
30    if startswith_prefix(message.content, prefix=PREFIX.CONVERT_TO_TC):
31        texts = message.reply_to_message.content if message.reply_to_message else message.content.removeprefix(PREFIX.CONVERT_TO_TC).lstrip()
32        traditional_chinese = zhconv.convert(texts, "zh-tw")
33        await message.reply(f"🔄简体转繁体:\n{traditional_chinese}", quote=True)