Merge pull request #180 from realpython/python-mongodb

MongoDB tutorial code examples

Author: jaschmitt

python-mongodb/README.md

@@ -0,0 +1,3 @@
+# Description
+
+The files in this folder provide the code examples discussed in the tutorial [Python and MongoDB: Connecting to NoSQL Databases](https://realpython.com/introduction-to-mongodb-and-python/).

python-mongodb/mongoengine_example.py

@@ -0,0 +1,30 @@
+from mongoengine import Document, ListField, StringField, URLField, connect
+
+# Establish a connection
+connect(db="rptutorials", host="localhost", port=27017)
+
+
+# Define a document schema or model
+class Tutorial(Document):
+    title = StringField(required=True, max_length=70)
+    author = StringField(required=True, max_length=20)
+    contributors = ListField(StringField(max_length=20))
+    url = URLField(required=True)
+
+
+# Create and save documents
+tutorial1 = Tutorial(
+    title="Beautiful Soup: Build a Web Scraper With Python",
+    author="Martin",
+    contributors=["Aldren", "Geir", "Jaya", "Joanna", "Mike"],
+    url="https://realpython.com/beautiful-soup-web-scraper-python/",
+)
+tutorial1.save()  # Insert the new tutorial
+
+# Retrieving all documents and print their title
+for doc in Tutorial.objects:
+    print(doc.title)
+
+# Filtering documents by author
+for doc in Tutorial.objects(author="Alex"):
+    print(doc.title)

python-mongodb/pymongo_example.py

@@ -0,0 +1,54 @@
+import pprint
+
+from pymongo import MongoClient
+
+# Establish a connection
+client = MongoClient("localhost", 27017)
+
+# Access the database
+db = client.rptutorials
+
+# Access a collection
+tutorial = db.tutorial
+
+# Insert a single document
+tutorial1 = {
+    "title": "Working With JSON Data in Python",
+    "author": "Lucas",
+    "contributors": ["Aldren", "Dan", "Joanna"],
+    "url": "https://realpython.com/python-json/",
+}
+result = tutorial.insert_one(tutorial1)
+print(f"One tutorial: {result.inserted_id}")
+
+# Insert several documents
+tutorial2 = {
+    "title": "Python’s Requests Library (Guide)",
+    "author": "Alex",
+    "contributors": ["Aldren", "Brad", "Joanna"],
+    "url": "https://realpython.com/python-requests/",
+}
+
+tutorial3 = {
+    "title": "Object-Oriented Programming (OOP) in Python 3",
+    "author": "David",
+    "contributors": ["Aldren", "Joanna", "Jacob"],
+    "url": "https://realpython.com/python3-object-oriented-programming/",
+}
+
+new_result = tutorial.insert_many([tutorial2, tutorial3])
+print(f"Multiple tutorials: {new_result.inserted_ids}")
+
+# Retrieve all documents
+for doc in tutorial.find():
+    pprint.pprint(doc)
+
+# Retrieve a single document
+jon_tutorial = tutorial.find_one({"author": "Jon"})
+pprint.pprint(jon_tutorial)
+
+# Close a connection
+with MongoClient() as client:
+    db = client.rptutorials
+    for doc in db.tutorial.find():
+        pprint.pprint(doc)