from pymongo import MongoClient

client = MongoClient("mongodb://localhost:27017")

def get_database():
    return client["neonsr"]

def get_collection(collection_name: str):
    database = client["neonsr"]
    return database[collection_name]