mirror of
https://github.com/nexus-stc/hyperboria
synced 2025-01-25 18:07:34 +01:00
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import logging
|
|
import traceback
|
|
from contextlib import asynccontextmanager
|
|
from typing import (
|
|
Awaitable,
|
|
Callable,
|
|
Optional,
|
|
)
|
|
|
|
from library.logging import error_log
|
|
from telethon import (
|
|
errors,
|
|
events,
|
|
)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def safe_execution(
|
|
error_log=error_log,
|
|
on_fail: Optional[Callable[[], Awaitable]] = None,
|
|
level=logging.WARNING,
|
|
is_logging_enabled: bool = True
|
|
):
|
|
try:
|
|
try:
|
|
yield
|
|
except events.StopPropagation:
|
|
raise
|
|
except errors.MessageNotModifiedError:
|
|
pass
|
|
except (
|
|
errors.UserIsBlockedError,
|
|
errors.QueryIdInvalidError,
|
|
errors.MessageDeleteForbiddenError,
|
|
errors.MessageIdInvalidError,
|
|
errors.ChatAdminRequiredError,
|
|
) as e:
|
|
if is_logging_enabled:
|
|
error_log(e, level=level)
|
|
traceback.print_exc()
|
|
except Exception as e:
|
|
if is_logging_enabled:
|
|
error_log(e, level=level)
|
|
traceback.print_exc()
|
|
if on_fail:
|
|
await on_fail()
|
|
except events.StopPropagation:
|
|
raise
|
|
except Exception as e:
|
|
if is_logging_enabled:
|
|
error_log(e, level=level)
|