1#!/usr/bin/env python
  2# -*- coding: utf-8 -*-
  3import re
  4
  5# From: https://github.com/Rongronggg9/RSS-to-Telegram-Bot/blob/84ce41fd31819d83e146507d50a95e28a5b58afe/src/parsing/weibo_emojify_map.py
  6COMMON = {
  7    "[微笑]": "🙂",
  8    "[可爱]": "😊",
  9    "[太开心]": "😆",
 10    "[鼓掌]": "👏",
 11    "[嘻嘻]": "😁",
 12    "[哈哈]": "😄",
 13    "[笑cry]": "😂",
 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    "[good]": "👍",
 64    "[弱]": "👎",
 65    "[NO]": "✋",
 66    "[耶]": "✌",
 67    "[拳头]": "✊",
 68    "[ok]": "👌",
 69    "[加油]": "💪",
 70    "[haha]": "🤟",
 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    "[点亮橙色]": "🖐",
102    "[看涨]": "📈",
103    "[看跌]": "📉",
104    "[星星]": "⭐",
105    "[空星]": "★",
106    "[全家福]": "👪",
107    "[圆月]": "🌕",
108    # customization
109    "[大笑]": "😆",
110}
111
112XHS = {
113    "[零R]": "0️⃣",
114    "[一R]": "1️⃣",
115    "[二R]": "2️⃣",
116    "[三R]": "3️⃣",
117    "[四R]": "4️⃣",
118    "[五R]": "5️⃣",
119    "[六R]": "6️⃣",
120    "[七R]": "7️⃣",
121    "[八R]": "8️⃣",
122    "[九R]": "9️⃣",
123    "[背包R]": "😏",
124    "[满月R]": "😏",
125    "[生气R]": "😡",
126    "[失望R]": "😞",
127    "[抓狂R]": "😖",
128    "[萌萌哒R]": "😊",
129    "[红色心形R]": "❤",
130    "[爆炸R]": "💥",
131    "[炸弹R]": "💣",
132}
133
134DOUYIN = {
135    "[苦涩]": "😭",
136    "[doge]": "🐶",
137    "[挖鼻]": "👃",
138    "[泪]": "😭",
139    "[眼含热泪]": "🥺",
140    "[呲牙]": "😁",
141    "[玫瑰]": "🌹",
142    "[一起加油]": "💪",
143    "[比心]": "💖",
144    "[看]": "🐶",
145    "[宕机]": "😧",
146    "[捂脸]": "🤦",
147    "[泪奔]": "😭",
148    "[尬笑]": "😅",
149    "[抱一抱]": "🤗",
150    "[打call]": "👏",
151    "[哇]": "🥰",
152}
153BILIBILI = {"[笑哭]": "🤦"}
154
155CURRENCY = {
156    "AED": "🇦🇪",
157    "AFN": "🇦🇫",
158    "ALL": "🇦🇱",
159    "AMD": "🇦🇲",
160    "ANG": "🇳🇱",
161    "AOA": "🇦🇴",
162    "ARS": "🇦🇷",
163    "AUD": "🇦🇺",
164    "AWG": "🇦🇼",
165    "AZN": "🇦🇿",
166    "BAM": "🇧🇦",
167    "BBD": "🇧🇧",
168    "BDT": "🇧🇩",
169    "BGN": "🇧🇬",
170    "BHD": "🇧🇭",
171    "BIF": "🇧🇮",
172    "BMD": "🇧🇲",
173    "BND": "🇧🇳",
174    "BOB": "🇧🇴",
175    "BRL": "🇧🇷",
176    "BSD": "🇧🇸",
177    "BTN": "🇧🇹",
178    "BWP": "🇧🇼",
179    "BYN": "🇧🇾",
180    "BYR": "🇧🇾",
181    "BZD": "🇧🇿",
182    "CAD": "🇨🇦",
183    "CDF": "🇨🇩",
184    "CHF": "🇨🇭",
185    "CLF": "🇨🇱",
186    "CLP": "🇨🇱",
187    "CNY": "🇨🇳",
188    "COP": "🇨🇴",
189    "CRC": "🇨🇷",
190    "CUC": "🇨🇺",
191    "CUP": "🇨🇺",
192    "CVE": "🇨🇻",
193    "CZK": "🇨🇿",
194    "DJF": "🇩🇯",
195    "DKK": "🇩🇰",
196    "DOP": "🇩🇴",
197    "DZD": "🇩🇿",
198    "EGP": "🇪🇬",
199    "ERN": "🇪🇷",
200    "ETB": "🇪🇹",
201    "EUR": "🇪🇺",
202    "FJD": "🇫🇯",
203    "FKP": "🇫🇰",
204    "GBP": "🇬🇧",
205    "GEL": "🇬🇪",
206    "GHS": "🇬🇭",
207    "GIP": "🇬🇮",
208    "GMD": "🇬🇲",
209    "GNF": "🇬🇳",
210    "GTQ": "🇬🇹",
211    "GYD": "🇬🇾",
212    "HKD": "🇭🇰",
213    "HNL": "🇭🇳",
214    "HRK": "🇭🇷",
215    "HTG": "🇭🇹",
216    "HUF": "🇭🇺",
217    "IDR": "🇮🇩",
218    "ILS": "🇮🇱",
219    "INR": "🇮🇳",
220    "IQD": "🇮🇶",
221    "IRR": "🇮🇷",
222    "ISK": "🇮🇸",
223    "JMD": "🇯🇲",
224    "JOD": "🇯🇴",
225    "JPY": "🇯🇵",
226    "KES": "🇰🇪",
227    "KGS": "🇰🇬",
228    "KHR": "🇰🇭",
229    "KMF": "🇰🇲",
230    "KPW": "🇰🇵",
231    "KRW": "🇰🇷",
232    "KWD": "🇰🇼",
233    "KYD": "🇰🇾",
234    "KZT": "🇰🇿",
235    "LAK": "🇱🇦",
236    "LBP": "🇱🇧",
237    "LKR": "🇱🇰",
238    "LRD": "🇱🇷",
239    "LSL": "🇱🇸",
240    "LTL": "🇱🇹",
241    "LVL": "🇱🇻",
242    "LYD": "🇱🇾",
243    "MAD": "🇲🇦",
244    "MDL": "🇲🇩",
245    "MGA": "🇲🇬",
246    "MKD": "🇲🇰",
247    "MMK": "🇲🇲",
248    "MNT": "🇲🇳",
249    "MOP": "🇲🇴",
250    "MRO": "🇲🇷",
251    "MUR": "🇲🇺",
252    "MVR": "🇲🇻",
253    "MWK": "🇲🇼",
254    "MXN": "🇲🇽",
255    "MYR": "🇲🇾",
256    "MZN": "🇲🇿",
257    "NAD": "🇳🇦",
258    "NGN": "🇳🇬",
259    "NIO": "🇳🇮",
260    "NOK": "🇳🇴",
261    "NPR": "🇳🇵",
262    "NZD": "🇳🇿",
263    "OMR": "🇴🇲",
264    "PAB": "🇵🇦",
265    "PEN": "🇵🇪",
266    "PGK": "🇵🇬",
267    "PHP": "🇵🇭",
268    "PKR": "🇵🇰",
269    "PLN": "🇵🇱",
270    "PYG": "🇵🇾",
271    "QAR": "🇶🇦",
272    "RON": "🇷🇴",
273    "RSD": "🇷🇸",
274    "RUB": "🇷🇺",
275    "RWF": "🇷🇼",
276    "SAR": "🇸🇦",
277    "SBD": "🇸🇧",
278    "SCR": "🇸🇨",
279    "SDG": "🇸🇩",
280    "SEK": "🇸🇪",
281    "SGD": "🇸🇬",
282    "SHP": "🇸🇭",
283    "SLE": "🇸🇱",
284    "SLL": "🇸🇱",
285    "SOS": "🇸🇴",
286    "SRD": "🇸🇷",
287    "STD": "🇸🇹",
288    "SYP": "🇸🇾",
289    "SZL": "🇸🇿",
290    "THB": "🇹🇭",
291    "TJS": "🇹🇯",
292    "TMT": "🇹🇲",
293    "TND": "🇹🇳",
294    "TOP": "🇹🇴",
295    "TRY": "🇹🇷",
296    "TTD": "🇹🇹",
297    "TWD": "🇹🇼",
298    "TZS": "🇹🇿",
299    "UAH": "🇺🇦",
300    "UGX": "🇺🇬",
301    "USD": "🇺🇸",
302    "UYU": "🇺🇾",
303    "UZS": "🇺🇿",
304    "VEF": "🇻🇪",
305    "VES": "🇻🇪",
306    "VND": "🇻🇳",
307    "VUV": "🇻🇺",
308    "WST": "🇼🇸",
309    "XAF": "🇨🇲",
310    "XCD": "🇦🇮",
311    "XOF": "🇧🇯",
312    "XPF": "🇵🇫",
313    "YER": "🇾🇪",
314    "ZAR": "🇿🇦",
315    "ZMW": "🇿🇲",
316    "ZWL": "🇿🇼",
317}
318
319CTYPE_EMOJI = {
320    "BOT": "🤖",
321    "GROUP": "👥",
322    "SUPERGROUP": "👥",
323    "CHANNEL": "📢",
324    "PRIVATE": "👤",
325}
326MTYPE_EMOJI = {
327    "text": "🔤",
328    "audio": "🎧",
329    "document": "📔",
330    "photo": "🏞",
331    "sticker": "🎨",
332    "video": "🎬",
333    "video_note": "🎥",
334    "animation": "✨",
335    "voice": "🎤",
336    "web_page": "🌐",
337}
338
339
340def emojify(text: str, platform: str = "all") -> str:
341    """Replace the text emojis with actual emojis.
342
343    [微笑]Hello[爱你] -> 🙂Hello🥰
344    """
345    if platform == "all":
346        EMOJI_MAP = COMMON | BILIBILI | XHS | DOUYIN
347    elif platform == "xhs":
348        EMOJI_MAP = COMMON | XHS
349    elif platform == "douyin":
350        EMOJI_MAP = COMMON | DOUYIN
351    elif platform == "bilibili":
352        EMOJI_MAP = COMMON | BILIBILI
353    else:
354        EMOJI_MAP = COMMON
355    pattern = re.compile("|".join(re.escape(k) for k in EMOJI_MAP))
356
357    return pattern.sub(lambda match: EMOJI_MAP[match.group(0)], text)
358
359
360if __name__ == "__main__":
361    text = "[微笑]Hello[爱你]"
362    print(emojify(text))