Deleted mysql support, redis password, webhook, keyboard constructors, filters, loguru.

Added support of decorators in handlers
Added function to notify admins on startup
This commit is contained in:
Kostiantyn Kriuchkov 2020-04-14 01:50:57 +03:00
parent 13ff8d0478
commit eab45a5088
30 changed files with 141 additions and 223 deletions

44
bot.py
View File

@ -1,42 +1,36 @@
import logging
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.fsm_storage.redis import RedisStorage2
from aiohttp import web
from loguru import logger
import filters
import handlers
import middlewares
from data import config
bot = Bot(token=config.BOT_TOKEN, parse_mode=types.ParseMode.HTML)
storage = RedisStorage2(**config.aiogram_redis)
dp = Dispatcher(bot, storage=storage)
logging.basicConfig(format=u'%(filename)s [LINE:%(lineno)d] #%(levelname)-8s [%(asctime)s] %(message)s',
level=logging.INFO,
# level=logging.DEBUG,
)
# noinspection PyUnusedLocal
async def on_startup(web_app: web.Application):
async def on_startup(dp):
import filters
import middlewares
filters.setup(dp)
middlewares.setup(dp)
handlers.errors.setup(dp)
handlers.user.setup(dp)
await dp.bot.delete_webhook()
await dp.bot.set_webhook(config.WEBHOOK_URL)
from utils.notify_admins import on_startup_notify
await on_startup_notify(dp)
async def execute(req: web.Request) -> web.Response:
upds = [types.Update(**(await req.json()))]
Bot.set_current(dp.bot)
Dispatcher.set_current(dp)
try:
await dp.process_updates(upds)
except Exception as e:
logger.error(e)
finally:
return web.Response()
async def on_shutdown(dp):
await bot.close()
await storage.close()
if __name__ == '__main__':
app = web.Application()
app.on_startup.append(on_startup)
app.add_routes([web.post(config.WEBHOOK_PATH, execute)])
web.run_app(app, port=5151, host='localhost')
from aiogram import executor
from handlers import dp
executor.start_polling(dp, on_startup=on_startup)

View File

@ -1,31 +1,20 @@
BOT_TOKEN = ''
BASE_URL = 'https://example.com' # Webhook domain
WEBHOOK_PATH = f'/webhook/bot/{BOT_TOKEN}'
WEBHOOK_URL = f'{BASE_URL}{WEBHOOK_PATH}'
import os
admins = []
from dotenv import load_dotenv
ip = {
'db': '',
'redis': '',
}
load_dotenv()
mysql_info = {
'host': ip['db'],
'user': '',
'password': '',
'db': '',
'maxsize': 5,
'port': 3306,
}
BOT_TOKEN = str(os.getenv("BOT_TOKEN"))
admins = [
]
ip = os.getenv("ip")
aiogram_redis = {
'host': ip['redis'],
'password': ''
'host': ip,
}
redis = {
'address': (ip['redis'], 6379),
'password': '',
'address': (ip, 6379),
'encoding': 'utf8'
}

View File

@ -1,7 +1,9 @@
from aiogram import Dispatcher
from .is_admin import AdminFilter
# from .is_admin import AdminFilter
def setup(dp: Dispatcher):
dp.filters_factory.bind(AdminFilter)
# dp.filters_factory.bind(AdminFilter)
pass

View File

@ -1,14 +0,0 @@
from aiogram import types
from aiogram.dispatcher.filters import BoundFilter
from data import config
class AdminFilter(BoundFilter):
key = 'is_admin'
def __init__(self, is_admin):
self.is_admin = is_admin
async def check(self, message: types.Message):
return message.from_user.id in config.admins

View File

@ -1,2 +1,4 @@
from . import errors
from . import user
from .errors import dp
from .users import dp
__all__ = ["dp"]

View File

@ -1,9 +1,3 @@
from aiogram import Dispatcher
from aiogram.utils import exceptions
from .error_handler import dp
from .not_modified import message_not_modified, message_to_delete_not_found
def setup(dp: Dispatcher):
dp.register_errors_handler(message_not_modified, exception=exceptions.MessageNotModified)
dp.register_errors_handler(message_to_delete_not_found, exception=exceptions.MessageToDeleteNotFound)
__all__ = ["dp"]

View File

@ -0,0 +1,56 @@
import logging
from bot import dp
@dp.errors_handler()
async def errors_handler(update, exception):
"""
Exceptions handler. Catches all exceptions within task factory tasks.
:param dispatcher:
:param update:
:param exception:
:return: stdout logging
"""
from aiogram.utils.exceptions import (Unauthorized, InvalidQueryID, TelegramAPIError,
CantDemoteChatCreator, MessageNotModified, MessageToDeleteNotFound,
MessageTextIsEmpty, RetryAfter,
CantParseEntities, MessageCantBeDeleted)
if isinstance(exception, CantDemoteChatCreator):
logging.debug("Can't demote chat creator")
return True
if isinstance(exception, MessageNotModified):
logging.debug('Message is not modified')
return True
if isinstance(exception, MessageCantBeDeleted):
logging.debug('Message cant be deleted')
return True
if isinstance(exception, MessageToDeleteNotFound):
logging.debug('Message to delete not found')
return True
if isinstance(exception, MessageTextIsEmpty):
logging.debug('MessageTextIsEmpty')
return True
if isinstance(exception, Unauthorized):
logging.info(f'Unauthorized: {exception}')
return True
if isinstance(exception, InvalidQueryID):
logging.exception(f'InvalidQueryID: {exception} \nUpdate: {update}')
return True
if isinstance(exception, TelegramAPIError):
logging.exception(f'TelegramAPIError: {exception} \nUpdate: {update}')
return True
if isinstance(exception, RetryAfter):
logging.exception(f'RetryAfter: {exception} \nUpdate: {update}')
return True
if isinstance(exception, CantParseEntities):
logging.exception(f'CantParseEntities: {exception} \nUpdate: {update}')
return True
logging.exception(f'Update: {update} \n{exception}')

View File

@ -1,10 +0,0 @@
from aiogram import types
from aiogram.utils import exceptions
async def message_not_modified(update: types.Update, error: exceptions.MessageNotModified):
return True
async def message_to_delete_not_found(update: types.Update, error: exceptions.MessageToDeleteNotFound):
return True

View File

View File

@ -1,10 +0,0 @@
from aiogram import Dispatcher
from aiogram.dispatcher.filters import CommandStart, CommandHelp
from .help import bot_help
from .start import bot_start
def setup(dp: Dispatcher):
dp.register_message_handler(bot_start, CommandStart())
dp.register_message_handler(bot_help, CommandHelp())

View File

@ -1,5 +0,0 @@
from aiogram import types
async def bot_start(msg: types.Message):
await msg.answer(f'Привет, {msg.from_user.full_name}!')

View File

@ -0,0 +1,5 @@
from .help import dp
from .start import dp
from .echo import dp
__all__ = ["dp"]

7
handlers/users/echo.py Normal file
View File

@ -0,0 +1,7 @@
from aiogram import types
from bot import dp
@dp.message_handler()
async def bot_start(message: types.Message):
await message.answer(message.text)

View File

@ -1,13 +1,16 @@
from aiogram import types
from aiogram.dispatcher.filters.builtin import CommandStart
from bot import dp
from utils.misc import rate_limit
@rate_limit(5, 'help')
async def bot_help(msg: types.Message):
@dp.message_handler(CommandStart())
async def bot_help(message: types.Message):
text = [
'Список команд: ',
'/start - Начать диалог',
'/help - Получить справку'
]
await msg.answer('\n'.join(text))
await message.answer('\n'.join(text))

9
handlers/users/start.py Normal file
View File

@ -0,0 +1,9 @@
from aiogram import types
from aiogram.dispatcher.filters.builtin import CommandStart
from bot import dp
@dp.message_handler(CommandStart())
async def bot_start(message: types.Message):
await message.answer(f'Привет, {message.from_user.full_name}!')

View File

@ -1,15 +0,0 @@
from typing import List
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton
from . import utils
class DefaultConstructor:
@staticmethod
def _create_kb(actions: List[str], schema: List[int]) -> ReplyKeyboardMarkup:
btns = []
for a in actions:
btns.append(KeyboardButton(a))
kb = utils.misc.arrange_default_schema(btns, schema)
return kb

View File

@ -1 +0,0 @@
from . import misc

View File

@ -1,17 +0,0 @@
from typing import List
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton
def arrange_default_schema(buttons: List[KeyboardButton], count: List[int]) -> ReplyKeyboardMarkup:
kb = ReplyKeyboardMarkup(resize_keyboard=True)
kb.row_width = max(count)
if sum(count) != len(buttons):
raise ValueError('Количество кнопок не совпадает со схемой')
tmplist = []
for a in count:
tmplist.append([])
for _ in range(a):
tmplist[-1].append(buttons.pop(0))
kb.keyboard = tmplist
return kb

View File

@ -0,0 +1,3 @@
from aiogram.utils.callback_data import CallbackData
some_callback = CallbackData("new")

View File

@ -1,21 +0,0 @@
from typing import List, Tuple, Dict
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from aiogram.utils.callback_data import CallbackData
from . import utils
class InlineConstructor:
@staticmethod
def _create_kb(actions: List[Tuple[str, Dict[str, str], CallbackData]], schema: List[int]) -> InlineKeyboardMarkup:
btns = []
for a, b, c in actions:
btns.append(
InlineKeyboardButton(
text=a,
callback_data=c.new(**b)
)
)
kb = utils.misc.arrange_inline_schema(btns, schema)
return kb

View File

@ -1 +0,0 @@
from . import misc

View File

@ -1,17 +0,0 @@
from typing import List
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
def arrange_inline_schema(buttons: List[InlineKeyboardButton], count: List[int]) -> InlineKeyboardMarkup:
kb = InlineKeyboardMarkup()
kb.row_width = max(count)
if sum(count) != len(buttons):
raise ValueError('Количество кнопок не совпадает со схемой')
tmplist = []
for a in count:
tmplist.append([])
for _ in range(a):
tmplist[-1].append(buttons.pop(0))
kb.inline_keyboard = tmplist
return kb

View File

@ -4,4 +4,4 @@ from .throttling import ThrottlingMiddleware
def setup(dp: Dispatcher):
dp.middleware.setup(ThrottlingMiddleware)
dp.middleware.setup(ThrottlingMiddleware())

View File

@ -1,5 +1,3 @@
aiomysql==0.0.20
aiogram==2.7
aiohttp==3.6.2
aioredis==1.3.1
loguru==0.4.1
aioredis==1.3.1

View File

@ -1 +0,0 @@
from . import user

View File

@ -1,3 +1,4 @@
from . import db_api
from . import misc
from . import redis
from .notify_admins import on_startup_notify

View File

@ -1,47 +0,0 @@
import asyncio
from typing import Optional, Dict, Any, Union, List
import aiomysql
from data import config
class RawConnection:
connection_pool = None
@staticmethod
async def _make_request(
sql: str,
params: Union[tuple, List[tuple]] = None,
fetch: bool = False,
mult: bool = False,
retries_count: int = 5
) -> Optional[Union[List[Dict[str, Any]], Dict[str, Any]]]:
if RawConnection.connection_pool is None:
RawConnection.connection_pool = await aiomysql.create_pool(**config.mysql_info)
async with RawConnection.connection_pool.acquire() as conn:
conn: aiomysql.Connection = conn
async with conn.cursor(aiomysql.DictCursor) as cur:
cur: aiomysql.DictCursor = cur
for i in range(retries_count):
try:
if isinstance(params, list):
await cur.executemany(sql, params)
else:
await cur.execute(sql, params)
except aiomysql.OperationalError as e:
if 'Deadlock found' in str(e):
await asyncio.sleep(1)
except aiomysql.InternalError as e:
if 'Deadlock found' in str(e):
await asyncio.sleep(1)
else:
break
if fetch:
if mult:
r = await cur.fetchall()
else:
r = await cur.fetchone()
return r
else:
await conn.commit()

14
utils/notify_admins.py Normal file
View File

@ -0,0 +1,14 @@
import logging
from aiogram import Dispatcher
from data.config import admins
async def on_startup_notify(dp: Dispatcher):
for admin in admins:
try:
await dp.bot.send_message(admin, "Бот Запущен")
except Exception as err:
logging.exception(err)