Basic MongoDB Operations in Python

1
# Note:
2
# On Debian & Ubuntu systems you'll first need to install virtualenv with:
3
# sudo apt install python3-venv
4
python3 -m venv venv

1
# Run the following on OSX & Linux:
2
source venv/bin/activate
3
4
# Run the following on Windows:
5
.\\venv\\Scripts\\activate

1
python -m pip install pymongo[srv]==3.10.1

1
python -m pip install python-dotenv==0.13.0

1
import datetime   # This will be needed later
2
import os
3
4
from dotenv import load_dotenv
5
from pymongo import MongoClient
6
7
# Load config from a .env file:
8
load_dotenv()
9
MONGODB_URI = os.environ['MONGODB_URI']
10
11
# Connect to your MongoDB cluster:
12
client = MongoClient(MONGODB_URI)
13
14
# List all the databases in the cluster:
15
for db_info in client.list_database_names():
16
   print(db_info)

1
# Unix:
2
export MONGODB_URI='mongodb+srv://yourusername:yourpasswordgoeshere@pythonquickstart-123ab.mongodb.net/test?retryWrites=true&w=majority'

1
$ python basic_operations.py
2
sample_airbnb
3
sample_analytics
4
sample_geospatial
5
sample_mflix
6
sample_supplies
7
sample_training
8
sample_weatherdata
9
twitter_analytics
10
admin
11
local

1
# Get a reference to the 'sample_mflix' database:
2
db = client['sample_mflix']
3
4
# List all the collections in 'sample_mflix':
5
collections = db.list_collection_names()
6
for collection in collections:
7
   print(collection)

1
$ python basic_operations.py
2
movies
3
sessions
4
comments
5
users
6
theaters

1
# Import the `pprint` function to print nested data:
2
from pprint import pprint
3
4
# Get a reference to the 'movies' collection:
5
movies = db['movies']
6
7
# Get the document with the title 'Blacksmith Scene':
8
pprint(movies.find_one({'title': 'Blacksmith Scene'}))

1
{'_id': ObjectId('573a1390f29313caabcd4135'),
2
'awards': {'nominations': 0, 'text': '1 win.', 'wins': 1},
3
'cast': ['Charles Kayser', 'John Ott'],
4
'countries': ['USA'],
5
'directors': ['William K.L. Dickson'],
6
'fullplot': 'A stationary camera looks at a large anvil with a blacksmith '
7
            'behind it and one on either side. The smith in the middle draws '
8
            'a heated metal rod from the fire, places it on the anvil, and '
9
            'all three begin a rhythmic hammering. After several blows, the '
10
            'metal goes back in the fire. One smith pulls out a bottle of '
11
            'beer, and they each take a swig. Then, out comes the glowing '
12
            'metal and the hammering resumes.',
13
'genres': ['Short'],
14
'imdb': {'id': 5, 'rating': 6.2, 'votes': 1189},
15
'lastupdated': '2015-08-26 00:03:50.133000000',
16
'num_mflix_comments': 1,
17
'plot': 'Three men hammer on an anvil and pass a bottle of beer around.',
18
'rated': 'UNRATED',
19
'released': datetime.datetime(1893, 5, 9, 0, 0),
20
'runtime': 1,
21
'title': 'Blacksmith Scene',
22
'tomatoes': {'lastUpdated': datetime.datetime(2015, 6, 28, 18, 34, 9),
23
            'viewer': {'meter': 32, 'numReviews': 184, 'rating': 3.0}},
24
'type': 'movie',
25
'year': 1893}

1
# Insert a document for the movie 'Parasite':
2
insert_result = movies.insert_one({
3
      "title": "Parasite",
4
      "year": 2020,
5
      "plot": "A poor family, the Kims, con their way into becoming the servants of a rich family, the Parks. "
6
      "But their easy life gets complicated when their deception is threatened with exposure.",
7
      "released": datetime(2020, 2, 7, 0, 0, 0),
8
   })
9
10
# Save the inserted_id of the document you just created:
11
parasite_id = insert_result.inserted_id
12
print("_id of inserted document: {parasite_id}".format(parasite_id=parasite_id))

1
import bson # <- Put this line near the start of the file if you prefer.
2
3
# Look up the document you just created in the collection:
4
print(movies.find_one({'_id': bson.ObjectId(parasite_id)}))

1
# Look up the documents you've created in the collection:
2
for doc in movies.find({"title": "Parasite"}):
3
   pprint(doc)

1
{
2
   'year': {
3
      '$lt': 1920
4
   }, 
5
   'genres': 'Romance'
6
}

1
# Update the document with the correct year:
2
update_result = movies.update_one({ '_id': parasite_id }, {
3
   '$set': {"year": 2019}
4
})
5
6
# Print out the updated record to make sure it's correct:
7
pprint(movies.find_one({'_id': ObjectId(parasite_id)}))

1
# Update *all* the Parasite movie docs to the correct year:
2
update_result = movies.update_many({"title": "Parasite"}, {"$set": {"year": 2019}})

1
movies.delete_many(
2
   {"title": "Parasite",}
3
)