Sync/SlashBot
1
0
Fork 0
mirror of https://github.com/Rongronggg9/SlashBot.git synced 2025-02-06 09:13:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: nickname unescaped

Browse Source
This commit is contained in:
Rongrong 2023-10-18 00:31:53 +08:00
parent 002f583ea7
commit 9d1dd01fca
No known key found for this signature in database
GPG Key ID: 1C2D45D45AB7FE94
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
SlashBot.py
View File

@ -3,6 +3,7 @@ from __future__ import annotations
import os import os
import sys import sys
import re import re
import html
import requests import requests
import telegram import telegram
from loguru import logger as _logger from loguru import logger as _logger
@ -171,10 +172,13 @@ class User:
def __get_user_by_username(self): def __get_user_by_username(self):
r = requests.get(f'https://t.me/{self.username}', proxies=REQUEST_PROXIES) r = requests.get(f'https://t.me/{self.username}', proxies=REQUEST_PROXIES)
self.name = re.search(r'(?<=<meta property="og:title" content=").*(?=")', r.text, re.IGNORECASE).group(0) og_t = re.search(r'(?<=<meta property="og:title" content=").*(?=")', r.text, re.IGNORECASE).group(0)
name = html.unescape(og_t) if og_t else None
page_title = re.search(r'(?<=<title>).*(?=</title>)', r.text, re.IGNORECASE).group(0) page_title = re.search(r'(?<=<title>).*(?=</title>)', r.text, re.IGNORECASE).group(0)
if page_title == self.name: # user does not exist if page_title == og_t: # user does not exist
self.name = None self.name = None
elif name:
self.name = name
def mention(self, mention_self: bool = False, pure: bool = False) -> str: def mention(self, mention_self: bool = False, pure: bool = False) -> str:
if not self.name: if not self.name:
@ -184,6 +188,7 @@ class User:
if (self.username and (not self.uid or self.uid < 0)) if (self.username and (not self.uid or self.uid < 0))
else f'tg://user?id={self.uid}') else f'tg://user?id={self.uid}')
name = self.name if not mention_self else "自己" name = self.name if not mention_self else "自己"
name = htmlEscape(name)
return f'<a href="{mention_deep_link}">{name}</a>' if not pure else name return f'<a href="{mention_deep_link}">{name}</a>' if not pure else name
def __eq__(self, other): def __eq__(self, other):