forked from UsergeTeam/Userge
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Added Userge antiSpam Api to antispam.py (UsergeTeam#338)
* Added Userge antiSpam Api to antispam.py * added lib to requirements * pep8 * Update antispam.py * fix probability to spam logs * bug fix Co-authored-by: rking32 <[email protected]>
- Loading branch information
1 parent
858a561
commit a02376d
Showing
5 changed files
with
184 additions
and
145 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -38,6 +38,7 @@ stagger | |
| telegraph | ||
| tgcrypto | ||
| urbandict==0.5 | ||
| UsergeAntiSpamApi | ||
| wget | ||
| wikipedia | ||
| youtube_dl>=2021.1.8 | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,177 @@ | ||
| """ setup antispam """ | ||
|
|
||
| # Copyright (C) 2020-2021 by UsergeTeam@Github, < https://github.com/UsergeTeam >. | ||
| # | ||
| # This file is part of < https://github.com/UsergeTeam/Userge > project, | ||
| # and is released under the "GNU v3.0 License Agreement". | ||
| # Please see < https://github.com/UsergeTeam/Userge/blob/master/LICENSE > | ||
| # | ||
| # All rights reserved | ||
|
|
||
| import json | ||
| import asyncio | ||
| from typing import Union | ||
|
|
||
| import aiohttp | ||
| import spamwatch | ||
| from spamwatch.types import Ban | ||
| from UsergeAntiSpamApi import Client | ||
| from UsergeAntiSpamApi.errors import InvalidApiToken | ||
|
|
||
| from userge import userge, Message, Config, get_collection, filters, pool | ||
|
|
||
| SAVED_SETTINGS = get_collection("CONFIGS") | ||
| GBAN_USER_BASE = get_collection("GBAN_USER") | ||
| WHITELIST = get_collection("WHITELIST_USER") | ||
| CHANNEL = userge.getCLogger(__name__) | ||
| LOG = userge.getLogger(__name__) | ||
|
|
||
|
|
||
| async def _init() -> None: | ||
| s_o = await SAVED_SETTINGS.find_one({'_id': 'ANTISPAM_ENABLED'}) | ||
| if s_o: | ||
| Config.ANTISPAM_SENTRY = s_o['data'] | ||
|
|
||
|
|
||
| @userge.on_cmd("antispam", about={ | ||
| 'header': "enable / disable antispam", | ||
| 'description': "Toggle API Auto Bans"}, allow_channels=False) | ||
| async def antispam_(message: Message): | ||
| """ enable / disable antispam """ | ||
| if Config.ANTISPAM_SENTRY: | ||
| Config.ANTISPAM_SENTRY = False | ||
| await message.edit("`antispam disabled !`", del_in=3) | ||
| else: | ||
| Config.ANTISPAM_SENTRY = True | ||
| await message.edit("`antispam enabled !`", del_in=3) | ||
| await SAVED_SETTINGS.update_one( | ||
| {'_id': 'ANTISPAM_ENABLED'}, {"$set": {'data': Config.ANTISPAM_SENTRY}}, upsert=True) | ||
|
|
||
|
|
||
| @userge.on_filters(filters.group & filters.new_chat_members, group=1, check_restrict_perm=True) | ||
| async def gban_at_entry(message: Message): | ||
| """ handle gbans """ | ||
| chat_id = message.chat.id | ||
| warned = False | ||
| for user in message.new_chat_members: | ||
| user_id = user.id | ||
| first_name = user.first_name | ||
| if await WHITELIST.find_one({'user_id': user_id}): | ||
| continue | ||
| gbanned = await GBAN_USER_BASE.find_one({'user_id': user_id}) | ||
| if gbanned: | ||
| if 'chat_ids' in gbanned: | ||
| chat_ids = gbanned['chat_ids'] | ||
| chat_ids.append(chat_id) | ||
| else: | ||
| chat_ids = [chat_id] | ||
| await asyncio.gather( | ||
| message.client.kick_chat_member(chat_id, user_id), | ||
| message.reply( | ||
| r"\\**#Userge_Antispam**//" | ||
| "\n\nGlobally Banned User Detected in this Chat.\n\n" | ||
| f"**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{gbanned['reason']}`\n\n" | ||
| "**Quick Action:** Banned", del_in=10), | ||
| CHANNEL.log( | ||
| r"\\**#Antispam_Log**//" | ||
| "\n\n**GBanned User $SPOTTED**\n" | ||
| f"**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** {gbanned['reason']}\n**Quick Action:** " | ||
| f"Banned in {message.chat.title}"), | ||
| GBAN_USER_BASE.update_one( | ||
| {'user_id': user_id, 'firstname': first_name}, | ||
| {"$set": {'chat_ids': chat_ids}}, upsert=True) | ||
| ) | ||
| elif Config.ANTISPAM_SENTRY: | ||
| res = await getData(f'https://api.cas.chat/check?user_id={user_id}') | ||
| if res['ok']: | ||
| reason = ' | '.join( | ||
| res['result']['messages']) if 'result' in res else None | ||
| await asyncio.gather( | ||
| message.client.kick_chat_member(chat_id, user_id), | ||
| message.reply( | ||
| r"\\**#Userge_Antispam**//" | ||
| "\n\nGlobally Banned User Detected in this Chat.\n\n" | ||
| "**$SENTRY CAS Federation Ban**\n" | ||
| f"**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{reason}`\n\n" | ||
| "**Quick Action:** Banned", del_in=10), | ||
| CHANNEL.log( | ||
| r"\\**#Antispam_Log**//" | ||
| "\n\n**GBanned User $SPOTTED**\n" | ||
| "**$SENRTY #CAS BAN**" | ||
| f"\n**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{reason}`\n**Quick Action:**" | ||
| f" Banned in {message.chat.title}\n\n$AUTOBAN #id{user_id}") | ||
| ) | ||
| continue | ||
| if Config.USERGE_ANTISPAM_API: | ||
| try: | ||
| response = await _get_userge_antispam_data(user_id) | ||
| except InvalidApiToken: | ||
| if not warned: | ||
| LOG.error("Your Userge AntiSpam Api key is Invalid!") | ||
| await CHANNEL.log("Your Userge AntiSpam Api key is Invalid!") | ||
| warned = True | ||
| else: | ||
| if response["success"]: | ||
| await asyncio.gather( | ||
| message.client.kick_chat_member(chat_id, user_id), | ||
| message.reply( | ||
| r"\\**#Userge_Antispam**//" | ||
| "\n\nGlobally Banned User Detected in this Chat.\n\n" | ||
| "**$SENTRY Userge AntiSpam API Ban**\n" | ||
| f"**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{response['reason']}`\n\n" | ||
| "**Quick Action:** Banned", del_in=10), | ||
| CHANNEL.log( | ||
| r"\\**#Antispam_Log**//" | ||
| "\n\n**GBanned User $SPOTTED**\n" | ||
| "**$SENRTY #USERGE_ANTISPAM_API BAN**" | ||
| f"\n**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{response['reason']}`\n" | ||
| f"**Quick Action:** Banned in {message.chat.title}\n\n" | ||
| f"$AUTOBAN #id{user_id}") | ||
| ) | ||
| continue | ||
| if Config.SPAM_WATCH_API: | ||
| intruder = await _get_spamwatch_data(user_id) | ||
| if intruder: | ||
| await asyncio.gather( | ||
| message.client.kick_chat_member(chat_id, user_id), | ||
| message.reply( | ||
| r"\\**#Userge_Antispam**//" | ||
| "\n\nGlobally Banned User Detected in this Chat.\n\n" | ||
| "**$SENTRY SpamWatch Federation Ban**\n" | ||
| f"**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{intruder.reason}`\n\n" | ||
| "**Quick Action:** Banned", del_in=10), | ||
| CHANNEL.log( | ||
| r"\\**#Antispam_Log**//" | ||
| "\n\n**GBanned User $SPOTTED**\n" | ||
| "**$SENRTY #SPAMWATCH_API BAN**" | ||
| f"\n**User:** [{first_name}](tg://user?id={user_id})\n" | ||
| f"**ID:** `{user_id}`\n**Reason:** `{intruder.reason}`\n" | ||
| f"**Quick Action:** Banned in {message.chat.title}\n\n" | ||
| f"$AUTOBAN #id{user_id}") | ||
| ) | ||
| message.continue_propagation() | ||
|
|
||
|
|
||
| async def getData(link: str): | ||
| async with aiohttp.ClientSession() as ses, ses.get(link) as resp: | ||
| try: | ||
| return json.loads(await resp.text()) | ||
| except json.decoder.JSONDecodeError: | ||
| return dict(ok=False, success=False) | ||
|
|
||
|
|
||
| @pool.run_in_thread | ||
| def _get_userge_antispam_data(user_id: int): | ||
| return Client(Config.USERGE_ANTISPAM_API).get_ban(user_id) | ||
|
|
||
|
|
||
| @pool.run_in_thread | ||
| def _get_spamwatch_data(user_id: int) -> Union[Ban, bool]: | ||
| return spamwatch.Client(Config.SPAM_WATCH_API).get_ban(user_id) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters