lab2_mongo.py

from pymongo import MongoClient

client = MongoClient('localhost', 27017)
db = client['store']

# Create collections
customers = db['customers']
orders = db['orders']

#reset collections
customers.drop()
orders.drop()

#insert documents
customers.insert_many([
    {'customer_id': 1, 'name': 'Alice Smith', 'email': 'alicesmith@gmail.com'},
    {'customer_id': 2, 'name': 'Bob Johnson', 'email': 'bobjohnson@gmail.com'}
])

orders.insert_many([
    {'order_id': 1, 'customer_id': 1, 'product': {'name': 'Laptop', 'price': 999.99}, 'quantity': 1, 'order_date': '2023-10-01'},
    {'order_id': 2, 'customer_id': 2, 'product': {'name': 'Smartphone', 'price': 499.99}, 'quantity': 2, 'order_date': '2023-10-02'}
])

print("\nFind one Customer:")
print(customers.find_one({'customer_id': 1}))

print("\nOrders for Bob Johnson:")
for order in orders.find({'customer_id': 2}):
    print(order)

print("\nAggregate total revenue per customer:")
pipeline = [
    {
        "$group": {
            "_id": "$customer_id",
            "total_spent": {
                "$sum": { "$multiply": ["$quantity", "$product.price"] }
            }
        }
    }
]

for res in orders.aggregate(pipeline):
    print(res)