mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-26 10:16:32 +00:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
|
import os
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
import sentry_sdk
|
||
|
|
from git.repo import Repo
|
||
|
|
from git.repo.fun import rev_parse
|
||
|
|
from sentry_sdk.integrations.excepthook import ExcepthookIntegration
|
||
|
|
from sentry_sdk.integrations.httpx import HttpxIntegration
|
||
|
|
from sentry_sdk.integrations.logging import LoggingIntegration
|
||
|
|
from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration
|
||
|
|
from telegram import Update
|
||
|
|
|
||
|
|
__all__ = ["SentryClient", "SentryClientException"]
|
||
|
|
|
||
|
|
|
||
|
|
class SentryClientException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class SentryClient:
|
||
|
|
def __init__(self, sentry_dsn: Optional[str] = None):
|
||
|
|
self.sentry_dsn = sentry_dsn
|
||
|
|
if sentry_dsn:
|
||
|
|
repo = Repo(os.getcwd())
|
||
|
|
sentry_sdk_git_hash = rev_parse(repo, "HEAD").hexsha
|
||
|
|
sentry_sdk.init(
|
||
|
|
sentry_dsn,
|
||
|
|
traces_sample_rate=1.0,
|
||
|
|
release=sentry_sdk_git_hash,
|
||
|
|
environment="production",
|
||
|
|
integrations=[
|
||
|
|
HttpxIntegration(),
|
||
|
|
ExcepthookIntegration(always_run=False),
|
||
|
|
LoggingIntegration(event_level=50),
|
||
|
|
SqlalchemyIntegration(),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def enabled(self) -> bool:
|
||
|
|
return bool(self.sentry_dsn)
|
||
|
|
|
||
|
|
def report_error(self, update: object, exc_info):
|
||
|
|
try:
|
||
|
|
return self._report_error(update, exc_info)
|
||
|
|
except Exception as exc:
|
||
|
|
raise SentryClientException from exc
|
||
|
|
|
||
|
|
def _report_error(self, update: object, exc_info):
|
||
|
|
if not self.sentry_dsn:
|
||
|
|
return
|
||
|
|
message: str = ""
|
||
|
|
chat_id: int = 0
|
||
|
|
user_id: int = 0
|
||
|
|
if isinstance(update, Update):
|
||
|
|
if update.effective_user:
|
||
|
|
chat_id = update.effective_user.id
|
||
|
|
if update.effective_chat:
|
||
|
|
user_id = update.effective_chat.id
|
||
|
|
if update.effective_message and update.effective_message.text:
|
||
|
|
message = update.effective_message.text
|
||
|
|
sentry_sdk.set_context("Target", {"ChatID": str(chat_id), "UserID": str(user_id), "Msg": message})
|
||
|
|
sentry_sdk.capture_exception(exc_info)
|