python script to test connection into mongo db

main.py

import sys
import pymongo
from pymongo.errors import ConnectionFailure

def connect_to_mongo(connection_string):
    try:
        # Create a MongoDB client
        client = pymongo.MongoClient(connection_string)

        # The ismaster command is cheap and does not require authentication
        client.admin.command('ismaster')

        print("MongoDB connection successful!")
    except ConnectionFailure as e:
        print(f"MongoDB connection failed: {e}")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python mongo_connect.py <mongo_connection_string>")
        sys.exit(1)

    connection_string = sys.argv[1]
    connect_to_mongo(connection_string)