testmongdb port

app_utils.py

@@ -11,6 +11,8 @@ from tinydb_serialization import SerializationMiddleware
 from tinydb_sqlite import SQLiteStorage
 import tempfile
 
+import socket
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -29,6 +31,25 @@ logger.info(f"Session ID: {SESSION_ID}")
 ADMIN_PASSWD = os.getenv('ADMIN_PASSWD', 'default_password')
 import os
 
+
+def is_mongodb_port_open(ip: str, port: int) -> bool:
+    """Check if the MongoDB port is open on the given IP address."""
+    try:
+        with socket.create_connection((ip, port), timeout=5):
+            return True
+    except (socket.timeout, ConnectionRefusedError, OSError):
+        return False
+
+# Example usage:
+MongoDBDomain = os.environ.get("MONGODB_DOMAIN")
+port_number = 27017        # Default MongoDB port
+
+if is_mongodb_port_open(MongoDBDomain, port_number):
+    print(f"The MongoDB port {port_number} is open.")
+else:
+    print(f"The MongoDB port {port_number} on is blocked.")
+    
+
 def explore_persistent_volumes():
     # Known persistent directories in Hugging Face Spaces
     persistent_dirs = [

reboot.py

@@ -2,6 +2,7 @@ from huggingface_hub import HfApi
 from dotenv import load_dotenv
 import os
 
+load_dotenv()
 # Replace with your actual token and space details
 HF_TOKEN = os.environ.get("HF_TOKEN")
 REPO_ID = "zqu2004/Test"