feat(spamblock): add SpamBot spamblock check and language support

This commit is contained in:
2026-08-02 02:14:19 +05:00
parent c96a335352
commit 7e3b31f32a
4 changed files with 69 additions and 15 deletions
+42 -5
View File
@@ -27,6 +27,7 @@ class Result:
me: Optional[User] = None
stars: Optional[int] = None
country: Optional[str] = None
spamblock: Optional[bool] = None
contacts: Optional[List[User]] = None
chats: Optional[List[Chat]] = None
@@ -35,9 +36,10 @@ class Result:
class Checker:
def __init__(self, path: Path, temp_path: Path):
def __init__(self, path: Path, temp_path: Path, lang_code: str = "en"):
self.path = path
self.temp_path = temp_path
self.lang_code = lang_code
self.client = None
logger.debug(f"Создание экземпляра Checker для пути: {path}")
try:
@@ -53,7 +55,7 @@ class Checker:
logger.error("TDATA не инициализирована")
raise SessionInvalid()
try:
await self.__ensure_client()
await self.__ensure_client(lang_code=self.lang_code)
except (OpenTeleException, Exception) as e:
logger.error(f"Не удалось инициализировать клиент: {e}")
raise SessionInvalid()
@@ -68,10 +70,10 @@ class Checker:
except Exception as e:
logger.warning(f"Ошибка при отключении клиента: {e}")
async def __ensure_client(self):
async def __ensure_client(self, lang_code: str = "en"):
if self.client is None:
logger.debug("Конвертация TDATA в Telethon клиент")
self.client = await self.desktop.convert()
logger.debug(f"Конвертация TDATA в Telethon клиент с языком '{lang_code}'")
self.client = await self.desktop.convert(lang_code=lang_code)
logger.debug("Telethon клиент успешно создан")
async def check_stars(self, me: User) -> int:
@@ -160,6 +162,39 @@ class Checker:
logger.warning(f"Не удалось проверить админские права: {e}")
return []
async def check_spamblock(self) -> bool:
logger.debug("Проверка spamblock через @SpamBot")
try:
await self.client.send_message("@SpamBot", "/start")
logger.debug("Отправлено сообщение /start в @SpamBot")
# Ждем ответ от SpamBot
import asyncio
await asyncio.sleep(2)
# Получаем последние сообщения от SpamBot
messages = await self.client.get_messages("@SpamBot", limit=1)
if messages and len(messages) > 0:
response_text = messages[0].message
logger.debug(f"Ответ от SpamBot: {response_text[:50]}...")
# Проверяем, содержит ли ответ текст об отсутствии ограничений
if "свободен от каких-либо ограничений" in response_text.lower():
logger.debug("Аккаунт не имеет spamblock")
return False
else:
logger.debug("Аккаунт имеет spamblock")
return True
else:
logger.warning("Не удалось получить ответ от SpamBot")
return None
except Exception as e:
logger.warning(f"Не удалось проверить spamblock: {e}")
return None
async def check_all(self) -> Result:
logger.debug("Начало полной проверки аккаунта")
try:
@@ -172,6 +207,7 @@ class Checker:
stars = await self.check_stars(me=me)
country = await self.check_country(me=me)
spamblock = await self.check_spamblock()
contacts = await self.check_contacts()
chats = await self.check_chats()
channels = await self.check_channels(chats=chats) if chats else []
@@ -184,6 +220,7 @@ class Checker:
me=me,
stars=stars,
country=country,
spamblock=spamblock,
contacts=contacts,
chats=chats,
channels=channels,
+3 -3
View File
@@ -107,7 +107,7 @@ class TDATA:
logger.debug("Отключение клиента после валидации")
await client.disconnect()
async def convert(self) -> TelegramClient:
async def convert(self, lang_code: str = "en") -> TelegramClient:
logger.debug("Начало конвертации TDATA в Telethon сессию")
if not await self.validate():
logger.error("TDATA невалидна, конвертация невозможна")
@@ -116,9 +116,9 @@ class TDATA:
tempPath = self.__gen_session_path()
try:
logger.debug(f"Создание Telethon сессии по пути '{tempPath}'")
logger.debug(f"Создание Telethon сессии по пути '{tempPath}' с языком '{lang_code}'")
client = await self.desktop.ToTelethon(
session=tempPath, flag=UseCurrentSession, api=self.api
session=tempPath, flag=UseCurrentSession, api=self.api, lang_code=lang_code
)
logger.debug("Валидация созданной сессии")
+23 -6
View File
@@ -16,17 +16,23 @@ class Worker:
input_path: str = "tdata",
output_path: str = "data",
temp_path: str = "temp",
lang_code: str = "en",
):
logger.debug(f"Инициализация Worker с путями: input={input_path}, output={output_path}, temp={temp_path}")
logger.debug(
f"Инициализация Worker с путями: input={input_path}, output={output_path}, temp={temp_path}, lang={lang_code}"
)
self.input_path = Path(input_path).absolute()
self.output_path = Path(output_path).absolute()
self.temp_path = Path(temp_path).absolute()
self.lang_code = lang_code
self.valid_path = self.output_path.joinpath("valid")
self.invalid_path = self.output_path.joinpath("invalid")
self.all_path = self.valid_path.joinpath("all.txt")
logger.debug(f"Пути настроены: valid={self.valid_path}, invalid={self.invalid_path}")
logger.debug(
f"Пути настроены: valid={self.valid_path}, invalid={self.invalid_path}"
)
async def __aenter__(self):
logger.debug("Вход в контекстный менеджер Worker")
@@ -105,10 +111,15 @@ class Worker:
logger.debug(f"Не удалось получить флаг для страны '{result.country}': {e}")
flag = "🌐"
spamblock_status = "unknown"
if result.spamblock is not None:
spamblock_status = "true" if result.spamblock else "false"
return (
f"👤 [{flag} {result.country}] {result.me.phone}{result.me.id}\n"
f"👤 [{flag} {result.country}] +{result.me.phone}{result.me.id}\n"
f"⭐️ {result.stars}\n"
f"💎 Premium: {'true' if result.me.premium else 'false'}\n"
f"🚫 Spamblock: {spamblock_status}\n"
f"👥 {len(result.chats)}/{len(result.contacts)}\n"
f"🍀 {len(result.admin_channels)}"
)
@@ -162,11 +173,17 @@ class Worker:
logger.info(f"Проверка TDATA: {tdata.name}")
try:
async with Checker(path=tdata_absolute, temp_path=self.temp_path) as checker:
async with Checker(
path=tdata_absolute, temp_path=self.temp_path, lang_code=self.lang_code
) as checker:
result = await checker.check_all()
logger.info(f"Проверка завершена: {result.me.first_name} (+{result.me.phone})")
self.__save_valid(user_id=result.me.id, result=result, tdata=tdata_absolute)
logger.info(
f"Проверка завершена: {result.me.first_name} (+{result.me.phone})"
)
self.__save_valid(
user_id=result.me.id, result=result, tdata=tdata_absolute
)
except SessionInvalid:
logger.warning(f"TDATA невалидна: {tdata.name}")