Commit ae25099

@@ -20,6 +20,7 @@ from messages.progress import modify_progress
 from messages.sender import send2tg
 from messages.utils import blockquote, equal_prefix, startswith_prefix
 from networking import hx_req
+from others.emoji import CURRENCY
 from utils import nowdt, number
 
 HELP = f"""📖**查询弹幕记录**
@@ -124,13 +125,13 @@ async def query_danmu(client: Client, message: Message, *, show_name: bool = Tru
 
     profit = ""
     for currency, amount in sorted(super_chats.items()):
-        profit += f"\n**{currency}**: {number(amount)}"
+        profit += f"\n{CURRENCY.get(currency, '❔')}**{currency}**: {number(amount)}"
     final = f"{header}{profit}\n{blockquote(danmu)}"
     if len(final) < TEXT_LENGTH - 10:
         await modify_progress(message=status_msg, text=final, force_update=True, **kwargs)
     else:
         caption += f"\n#️⃣弹幕数: {count}"
-        caption += profit
+        caption += f"\n🎉SuperChat:{profit}" if profit else ""
         with BytesIO(danmu.encode("utf-8")) as f:
             await client.send_document(info["cid"], f, file_name=f"{user}{match_time}{keyword}弹幕记录.txt", caption=caption)
         await modify_progress(message=status_msg, del_status=True, **kwargs)
@@ -138,15 +139,13 @@ async def query_danmu(client: Client, message: Message, *, show_name: bool = Tru
 
 def parse_danmu(data: list[dict], super_chats: defaultdict, *, show_name: bool = True) -> dict:
     msg = ""
-    sc_pattern = re.compile(r"^([A-Z]{2,}) (\d+(?:\.\d+)?)$")
     for danmu in sorted(data, key=lambda x: x["timestamp"], reverse=True):  # new to old
         dt = datetime.fromtimestamp(danmu["timestamp"] / 1000000, tz=UTC).astimezone(ZoneInfo(TZ))
         sc_amount = ""
-        if amt := danmu.get("scAmount"):
-            sc_amount = f" ({amt.replace('USD ', '💵')})"
-            if matched := sc_pattern.fullmatch(amt):
-                currency = matched.group(1)
-                super_chats[currency] += Decimal(matched.group(2))
+        if super_chat := danmu.get("scAmount"):
+            currency, amount = super_chat.split(" ")
+            super_chats[currency] += Decimal(amount)
+            sc_amount = f" ({CURRENCY.get(currency, '❔')}{super_chat})"
         msg += f"\n{dt:%m-%d %H:%M:%S}|{danmu['authorName']}{sc_amount}: {danmu['message']}" if show_name else f"\n{dt:%m-%d %H:%M:%S}{sc_amount}: {danmu['message']}"
     return {"danmu": msg.strip(), "num_messages": len(data)}
 
@@ -208,7 +207,7 @@ async def sync_danmu_to_d1() -> None:
     now = nowdt(TZ)
 
     async def batch_sync(danmu_list: list[dict], saved_items: list[str]) -> int:
-        sc_pattern = re.compile(r"^([A-Z]{2,}) (\d+(?:\.\d+)?)$")
+        sc_pattern = re.compile(r"^([A-Z]{3,}) (\d+(?:\.\d+)?)$")
         db_columns = "id INTEGER PRIMARY KEY, time TEXT, uid INTEGER, user TEXT, text TEXT, sc_amt REAL NULL, sc_ccy TEXT NULL"
         tasks = []
         for danmu in danmu_list:

@@ -152,6 +152,170 @@ DOUYIN = {
 }
 BILIBILI = {"[笑哭]": "🤦"}
 
+CURRENCY = {
+    "AED": "🇦🇪",
+    "AFN": "🇦🇫",
+    "ALL": "🇦🇱",
+    "AMD": "🇦🇲",
+    "ANG": "🇳🇱",
+    "AOA": "🇦🇴",
+    "ARS": "🇦🇷",
+    "AUD": "🇦🇺",
+    "AWG": "🇦🇼",
+    "AZN": "🇦🇿",
+    "BAM": "🇧🇦",
+    "BBD": "🇧🇧",
+    "BDT": "🇧🇩",
+    "BGN": "🇧🇬",
+    "BHD": "🇧🇭",
+    "BIF": "🇧🇮",
+    "BMD": "🇧🇲",
+    "BND": "🇧🇳",
+    "BOB": "🇧🇴",
+    "BRL": "🇧🇷",
+    "BSD": "🇧🇸",
+    "BTN": "🇧🇹",
+    "BWP": "🇧🇼",
+    "BYN": "🇧🇾",
+    "BYR": "🇧🇾",
+    "BZD": "🇧🇿",
+    "CAD": "🇨🇦",
+    "CDF": "🇨🇩",
+    "CHF": "🇨🇭",
+    "CLF": "🇨🇱",
+    "CLP": "🇨🇱",
+    "CNY": "🇨🇳",
+    "COP": "🇨🇴",
+    "CRC": "🇨🇷",
+    "CUC": "🇨🇺",
+    "CUP": "🇨🇺",
+    "CVE": "🇨🇻",
+    "CZK": "🇨🇿",
+    "DJF": "🇩🇯",
+    "DKK": "🇩🇰",
+    "DOP": "🇩🇴",
+    "DZD": "🇩🇿",
+    "EGP": "🇪🇬",
+    "ERN": "🇪🇷",
+    "ETB": "🇪🇹",
+    "EUR": "🇪🇺",
+    "FJD": "🇫🇯",
+    "FKP": "🇫🇰",
+    "GBP": "🇬🇧",
+    "GEL": "🇬🇪",
+    "GHS": "🇬🇭",
+    "GIP": "🇬🇮",
+    "GMD": "🇬🇲",
+    "GNF": "🇬🇳",
+    "GTQ": "🇬🇹",
+    "GYD": "🇬🇾",
+    "HKD": "🇭🇰",
+    "HNL": "🇭🇳",
+    "HRK": "🇭🇷",
+    "HTG": "🇭🇹",
+    "HUF": "🇭🇺",
+    "IDR": "🇮🇩",
+    "ILS": "🇮🇱",
+    "INR": "🇮🇳",
+    "IQD": "🇮🇶",
+    "IRR": "🇮🇷",
+    "ISK": "🇮🇸",
+    "JMD": "🇯🇲",
+    "JOD": "🇯🇴",
+    "JPY": "🇯🇵",
+    "KES": "🇰🇪",
+    "KGS": "🇰🇬",
+    "KHR": "🇰🇭",
+    "KMF": "🇰🇲",
+    "KPW": "🇰🇵",
+    "KRW": "🇰🇷",
+    "KWD": "🇰🇼",
+    "KYD": "🇰🇾",
+    "KZT": "🇰🇿",
+    "LAK": "🇱🇦",
+    "LBP": "🇱🇧",
+    "LKR": "🇱🇰",
+    "LRD": "🇱🇷",
+    "LSL": "🇱🇸",
+    "LTL": "🇱🇹",
+    "LVL": "🇱🇻",
+    "LYD": "🇱🇾",
+    "MAD": "🇲🇦",
+    "MDL": "🇲🇩",
+    "MGA": "🇲🇬",
+    "MKD": "🇲🇰",
+    "MMK": "🇲🇲",
+    "MNT": "🇲🇳",
+    "MOP": "🇲🇴",
+    "MRO": "🇲🇷",
+    "MUR": "🇲🇺",
+    "MVR": "🇲🇻",
+    "MWK": "🇲🇼",
+    "MXN": "🇲🇽",
+    "MYR": "🇲🇾",
+    "MZN": "🇲🇿",
+    "NAD": "🇳🇦",
+    "NGN": "🇳🇬",
+    "NIO": "🇳🇮",
+    "NOK": "🇳🇴",
+    "NPR": "🇳🇵",
+    "NZD": "🇳🇿",
+    "OMR": "🇴🇲",
+    "PAB": "🇵🇦",
+    "PEN": "🇵🇪",
+    "PGK": "🇵🇬",
+    "PHP": "🇵🇭",
+    "PKR": "🇵🇰",
+    "PLN": "🇵🇱",
+    "PYG": "🇵🇾",
+    "QAR": "🇶🇦",
+    "RON": "🇷🇴",
+    "RSD": "🇷🇸",
+    "RUB": "🇷🇺",
+    "RWF": "🇷🇼",
+    "SAR": "🇸🇦",
+    "SBD": "🇸🇧",
+    "SCR": "🇸🇨",
+    "SDG": "🇸🇩",
+    "SEK": "🇸🇪",
+    "SGD": "🇸🇬",
+    "SHP": "🇸🇭",
+    "SLE": "🇸🇱",
+    "SLL": "🇸🇱",
+    "SOS": "🇸🇴",
+    "SRD": "🇸🇷",
+    "STD": "🇸🇹",
+    "SYP": "🇸🇾",
+    "SZL": "🇸🇿",
+    "THB": "🇹🇭",
+    "TJS": "🇹🇯",
+    "TMT": "🇹🇲",
+    "TND": "🇹🇳",
+    "TOP": "🇹🇴",
+    "TRY": "🇹🇷",
+    "TTD": "🇹🇹",
+    "TWD": "🇹🇼",
+    "TZS": "🇹🇿",
+    "UAH": "🇺🇦",
+    "UGX": "🇺🇬",
+    "USD": "🇺🇸",
+    "UYU": "🇺🇾",
+    "UZS": "🇺🇿",
+    "VEF": "🇻🇪",
+    "VES": "🇻🇪",
+    "VND": "🇻🇳",
+    "VUV": "🇻🇺",
+    "WST": "🇼🇸",
+    "XAF": "🇨🇲",
+    "XCD": "🇦🇮",
+    "XOF": "🇧🇯",
+    "XPF": "🇵🇫",
+    "YER": "🇾🇪",
+    "ZAR": "🇿🇦",
+    "ZMW": "🇿🇲",
+    "ZWL": "🇿🇼",
+}
+
 
 def emojify(text: str, platform: str = "all") -> str:
     """Replace the text emojis with actual emojis.