Skip to content

Commit 44fe299

Browse files
committed
Add MongoDB connection setup in handlers.py
1 parent ecfbfaa commit 44fe299

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

bot_package/handlers.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import time
55
import shutil
66
import re
7+
from pymongo import MongoClient
78
from pyrogram import Client, filters
89
from pyrogram.types import CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
910
from pyrogram.errors import FloodWait, PeerIdInvalid
@@ -40,6 +41,15 @@
4041
DL_COMPLETE_BUTTON,
4142
) # Assuming buttons.py exists
4243

44+
# --- MongoDB Connection ---
45+
pwd = os.getenv('MONGO_PWD')
46+
47+
# connection string
48+
connection_string = f'mongodb+srv://myAtlasDBUser:{pwd}@pyroytbot.p1vptc2.mongodb.net/?retryWrites=true&w=majority&appName=PyroYtBot'
49+
50+
# initialize the client
51+
client = MongoClient(connection_string)
52+
4353

4454
# --- MongoDB tools ---
4555
async def mongo_check_user_database(

0 commit comments

Comments
 (0)