Code for connecting to and inserting into Mongo

mediabridge/db/connect.py

@@ -1 +1,19 @@
 # MongoDB connection setup
+import os
+from functools import cache
+
+from dotenv import load_dotenv
+from pymongo import MongoClient
+
+load_dotenv()
+
+
+@cache
+def connect_to_mongo():
+    # This will raise an exception if MONGODB_URI is not defined in the environment.
+    # If that isn't enough information to help developers populate their environment,
+    # we should use `get` and throw a custom message if the value is missing.
+    mongo_uri = os.environ["MONGODB_URI"]
+    client = MongoClient(mongo_uri)
+    db = client["mediabridge"]
+    return db

mediabridge/db/queries.py

@@ -1 +1,20 @@
-# Functions to query MongoDB for movies and interactions
+from mediabridge.db.connect import connect_to_mongo
+
+
+def insert_into_mongo(movie):
+    db = connect_to_mongo()
+    collection = db["movies"]
+    collection.update_one(
+        {"wikidata_id": movie[1]},
+        {
+            "set": {
+                "netflix_id": movie[0],
+                "wikidata_id": movie[1],
+                "title": movie[2],
+                "year": movie[3],
+                "genre": movie[4],
+                "director": movie[5],
+            }
+        },
+        upsert=True,
+    )