bakatrouble/e621_bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

protect bot

Browse Source
This commit is contained in:
bakatrouble 2023-04-14 11:24:20 +03:00
parent f5cce5b9bf
commit 3f3831bdba
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -7,6 +7,7 @@ from PIL import Image
import httpx import httpx
import redis.asyncio as aioredis import redis.asyncio as aioredis
from aiogram import Bot, Dispatcher from aiogram import Bot, Dispatcher
from aiogram.dispatcher import filters
from aiogram.types import Message, ParseMode from aiogram.types import Message, ParseMode
from aiogram.utils import executor, exceptions from aiogram.utils import executor, exceptions
import dotenv import dotenv
@ -22,6 +23,8 @@ logging.basicConfig(level=logging.INFO)
bot = Bot(token=os.environ['BOT_TOKEN']) bot = Bot(token=os.environ['BOT_TOKEN'])
dp = Dispatcher(bot) dp = Dispatcher(bot)
user_ids = os.environ['USERS'].split(',')
async def check_updates(): async def check_updates():
logging.warning('Waiting for lock...') logging.warning('Waiting for lock...')
@ -99,7 +102,7 @@ async def check_updates():
logging.exception(e) logging.exception(e)
@dp.message_handler(commands=['add']) @dp.message_handler(commands=['add'], filter=filters.IDFilter(chat_id=user_ids))
async def add_tag(msg: Message): async def add_tag(msg: Message):
args = msg.get_args() args = msg.get_args()
if not args: if not args:
@ -110,7 +113,7 @@ async def add_tag(msg: Message):
await msg.reply(f'Tags {args} added') await msg.reply(f'Tags {args} added')
@dp.message_handler(regexp=r'^\/del_\S+$') @dp.message_handler(regexp=r'^\/del_\S+$', filter=filters.IDFilter(chat_id=user_ids))
async def del_tag(msg: Message): async def del_tag(msg: Message):
args = msg.text[5:] args = msg.text[5:]
if not args: if not args:
@ -126,7 +129,7 @@ async def del_tag(msg: Message):
await msg.reply(f'Tag {args} removed') await msg.reply(f'Tag {args} removed')
@dp.message_handler(commands=['del']) @dp.message_handler(commands=['del'], filter=filters.IDFilter(chat_id=user_ids))
async def del_command(msg: Message): async def del_command(msg: Message):
args = msg.get_args() args = msg.get_args()
if not args: if not args:
@ -137,7 +140,7 @@ async def del_command(msg: Message):
await msg.reply(f'Tags {args} removed') await msg.reply(f'Tags {args} removed')
@dp.message_handler(commands=['list']) @dp.message_handler(commands=['list'], filter=filters.IDFilter(chat_id=user_ids))
async def list_tags(msg: Message): async def list_tags(msg: Message):
tags = [t.decode() for t in await redis.smembers('e621:subs')] tags = [t.decode() for t in await redis.smembers('e621:subs')]
tags.sort() tags.sort()
@ -155,7 +158,7 @@ async def list_tags(msg: Message):
await msg.reply(f'Monitored tags:\n\n{lines}') await msg.reply(f'Monitored tags:\n\n{lines}')
@dp.message_handler(commands=['update']) @dp.message_handler(commands=['update'], filter=filters.IDFilter(chat_id=user_ids))
async def update(msg: Message): async def update(msg: Message):
await check_updates() await check_updates()