NetworkSecurity / verify_mongo.py
Inder-26
feat: Add Hugging Face Hub sync workflow, MongoDB connection verification script, and update Dockerfile for Python 3.11 and non-root user.
e1d0f32
raw
history blame contribute delete
771 Bytes
import os
import sys
from dotenv import load_dotenv
from pymongo.mongo_client import MongoClient
# Load environment variables
load_dotenv()
mongo_url = os.getenv("MONGO_DB_URL")
print(f"Checking connection for URL: {mongo_url[:20]}..." if mongo_url else "No MONGO_DB_URL found in .env")
if not mongo_url:
print("❌ MONGO_DB_URL is not set in .env")
sys.exit(1)
try:
client = MongoClient(mongo_url, serverSelectionTimeoutMS=5000)
# Send a ping to confirm a successful connection
client.admin.command('ping')
print("✅ Successfully connected to MongoDB!")
# List databases to be sure
dbs = client.list_database_names()
print(f" Databases found: {dbs}")
except Exception as e:
print(f"❌ Connection Failed: {e}")