Add sentry integration

Author: hemantapkh

app.json

@@ -51,6 +51,16 @@
             "value": "",
             "required": true
         },
+        "SENTRY_DSN": {
+            "description": "Sentry.io token for error tracking",
+            "value": "",
+            "required": true
+        },
+        "ENVIRONMENT": {
+            "description": "Type of app environment i.e local or production",
+            "value": "production",
+            "required": false
+        },
         "START_ADS": {
             "description": "Send message from a channel to a user when they start the bot",
             "value": "false",

app/plugins/functions/init.py

@@ -28,7 +28,7 @@ async def add_admins(self):
                     await self.Client.DB.merge(new_admin)
 
                 except ValueError:
-                    logger.error(f"Invalid user id: {user_id}")
+                    logger.warning(f"Invalid user id: {user_id}")
 
     async def add_commands(self):
         logger.info("Setting bot commands")
@@ -66,7 +66,7 @@ async def add_commands(self):
                     ),
                 )
             except pyrogram.errors.exceptions.bad_request_400.PeerIdInvalid as err:
-                logger.error(f"Error setting commands for admins: {err}")
+                logger.warning(f"Error setting commands for admins: {err}")
 
 
 all_commands = [

app/plugins/functions/misc.py

@@ -44,7 +44,7 @@ async def message_admins(self, message):
                 )
 
             except Exception as err:
-                logger.error(f"Error sending message to admin: {err}")
+                logger.warning(f"Error sending message to admin: {err}")
                 pass
 
     # Get user language

app/torrenthunt.py

@@ -5,6 +5,7 @@
 from sys import argv
 
 import uvloop
+import sentry_sdk
 from apis.requests import Requests
 from apis.torrenthunt import TorrentHunt
 from database import DataBase
@@ -21,6 +22,12 @@
 from py1337x import py1337x
 from pyrogram import Client, filters
 
+# Initializing sentry for error tracking
+sentry_sdk.init(
+    dsn=environ.get("SENTRY_DSN"),
+    environment=environ.get("ENVIRONMENT") or "local"
+)
+
 # Installing UVloop for better performance
 logger.info("Installing uvloop")
 uvloop.install()

requirements.txt

@@ -35,6 +35,7 @@ requests==2.31.0
 requests-cache==1.1.1
 scikit-learn==1.2.2
 scipy==1.12.0
+sentry-sdk==2.1.1
 six==1.16.0
 soupsieve==2.5
 SQLAlchemy==2.0.25

sample.env

@@ -16,6 +16,10 @@ START_ADS=false
 START_ADS_CHANNEL=
 START_ADS_MESSAGE=
 
+# Sentry.io for error tracing
+SENTRY_DSN=
+
 # MISC
 WORKDIR=
 LOGFILE=
+ENVIRONMENT=local