get mongo docs

juddges/_modidx.py

@@ -5,4 +5,7 @@
                 'doc_host': 'https://laugustyniak.github.io',
                 'git_url': 'https://github.com/laugustyniak/juddges',
                 'lib_path': 'juddges'},
-  'syms': {'juddges.data.pl_court_api': {}, 'juddges.prompts.information_extraction': {}, 'juddges.settings': {}}}
+  'syms': { 'juddges.data.models': {},
+            'juddges.data.pl_court_api': {},
+            'juddges.prompts.information_extraction': {},
+            'juddges.settings': {}}}

juddges/data/models.py

@@ -0,0 +1,16 @@
+import os
+
+from pymongo import MongoClient
+
+if os.environ.get("MONGO_URI", None) is None:
+    raise Exception("Missing `MONGO_URI` environment variable.")
+
+
+if os.environ.get("MONGO_DB_NAME", None) is None:
+    raise Exception("Missing `MONGO_DB_NAME` environment variable.")
+
+
+def get_mongo_collection(collection_name: str):
+    client = MongoClient(os.environ["MONGO_URI"])
+    db = client[os.environ["MONGO_DB_NAME"]]
+    return db[collection_name]