used black
This commit is contained in:
@@ -8,19 +8,20 @@ from bot.modules.settings import UserSettings
|
||||
|
||||
class SettingsInjectorMiddleware(BaseMiddleware):
|
||||
async def __call__(
|
||||
self,
|
||||
handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
|
||||
event: TelegramObject,
|
||||
data: Dict[str, Any],
|
||||
self,
|
||||
handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
|
||||
event: TelegramObject,
|
||||
data: Dict[str, Any],
|
||||
):
|
||||
if (not hasattr(event, 'from_user') and
|
||||
(not hasattr(event, 'inline_query') or event.inline_query is None)):
|
||||
if not hasattr(event, "from_user") and (
|
||||
not hasattr(event, "inline_query") or event.inline_query is None
|
||||
):
|
||||
return await handler(event, data)
|
||||
elif hasattr(event, 'inline_query') and event.inline_query is not None:
|
||||
elif hasattr(event, "inline_query") and event.inline_query is not None:
|
||||
settings = UserSettings(event.inline_query.from_user.id)
|
||||
data['settings'] = settings
|
||||
data["settings"] = settings
|
||||
else:
|
||||
settings = UserSettings(event.from_user.id)
|
||||
data['settings'] = settings
|
||||
data["settings"] = settings
|
||||
|
||||
return await handler(event, data)
|
||||
|
||||
@@ -8,12 +8,12 @@ from bot.modules.database import db
|
||||
|
||||
class PrivateButtonMiddleware(BaseMiddleware):
|
||||
async def __call__(
|
||||
self,
|
||||
handler: Callable[[CallbackQuery, Dict[str, Any]], Awaitable[Any]],
|
||||
event: CallbackQuery,
|
||||
data: Dict[str, Any],
|
||||
self,
|
||||
handler: Callable[[CallbackQuery, Dict[str, Any]], Awaitable[Any]],
|
||||
event: CallbackQuery,
|
||||
data: Dict[str, Any],
|
||||
):
|
||||
if event.from_user.id == db.inline[event.inline_message_id].from_user.id:
|
||||
return await handler(event, data)
|
||||
else:
|
||||
await event.answer('This button is not for you')
|
||||
await event.answer("This button is not for you")
|
||||
|
||||
@@ -26,10 +26,10 @@ class SavedResult:
|
||||
|
||||
class SaveChosenMiddleware(BaseMiddleware):
|
||||
async def __call__(
|
||||
self,
|
||||
handler: Callable[[ChosenInlineResult, Dict[str, Any]], Awaitable[Any]],
|
||||
event: ChosenInlineResult,
|
||||
data: Dict[str, Any],
|
||||
self,
|
||||
handler: Callable[[ChosenInlineResult, Dict[str, Any]], Awaitable[Any]],
|
||||
event: ChosenInlineResult,
|
||||
data: Dict[str, Any],
|
||||
):
|
||||
db.inline[event.inline_message_id] = SavedResult(
|
||||
result_id=event.result_id,
|
||||
@@ -38,9 +38,9 @@ class SaveChosenMiddleware(BaseMiddleware):
|
||||
first_name=event.from_user.first_name,
|
||||
last_name=event.from_user.last_name,
|
||||
username=event.from_user.username,
|
||||
language_code=event.from_user.language_code
|
||||
language_code=event.from_user.language_code,
|
||||
),
|
||||
query=event.query,
|
||||
inline_message_id=event.inline_message_id
|
||||
inline_message_id=event.inline_message_id,
|
||||
)
|
||||
return await handler(event, data)
|
||||
|
||||
Reference in New Issue
Block a user