mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-21 09:07:46 +01:00
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
import logging
|
||
|
|
import traceback
|
||
|
|
from contextlib import asynccontextmanager
|
||
|
|
from typing import (
|
||
|
|
Awaitable,
|
||
|
|
Callable,
|
||
|
|
Optional,
|
||
|
|
)
|
||
|
|
|
||
|
|
from telethon import (
|
||
|
|
errors,
|
||
|
|
events,
|
||
|
|
)
|
||
|
|
|
||
|
|
from .base import RequestContext
|
||
|
|
|
||
|
|
|
||
|
|
@asynccontextmanager
|
||
|
|
async def safe_execution(
|
||
|
|
request_context: RequestContext,
|
||
|
|
on_fail: Optional[Callable[[], Awaitable]] = None,
|
||
|
|
):
|
||
|
|
try:
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
except events.StopPropagation:
|
||
|
|
raise
|
||
|
|
except (
|
||
|
|
errors.UserIsBlockedError,
|
||
|
|
errors.QueryIdInvalidError,
|
||
|
|
errors.MessageDeleteForbiddenError,
|
||
|
|
errors.MessageIdInvalidError,
|
||
|
|
errors.MessageNotModifiedError,
|
||
|
|
errors.ChatAdminRequiredError,
|
||
|
|
) as e:
|
||
|
|
request_context.error_log(e, level=logging.WARNING)
|
||
|
|
except Exception as e:
|
||
|
|
traceback.print_exc()
|
||
|
|
request_context.error_log(e)
|
||
|
|
if on_fail:
|
||
|
|
await on_fail()
|
||
|
|
except events.StopPropagation:
|
||
|
|
raise
|
||
|
|
except Exception as e:
|
||
|
|
request_context.error_log(e)
|