maggma

import os from maggma.stores import MemoryStore # Sample data turtles = [ {"name": "Leonardo", "color": "blue", "tool": "sword"}, {"name": "Donatello", "color": "purple", "tool": "staff"}, {"name": "Michelangelo", "color": "orange", "tool": "nunchuks"}, {"name": "Raphael", "color": "red", "tool": "sai"} ] # Create a MemoryStore (in-memory, data not persistent) # 'key' argument specifies the unique identifier for documents store = MemoryStore(key="name") # Connect to the store (for MemoryStore, this just initializes it) store.connect() # Add data to the store using update # upsert=True means insert if not found, update if found store.update(turtles, key_field='name', upsert=True) # Query the store print(f"Total documents: {store.count()}") print(f"Blue turtle: {store.query(criteria={'color': 'blue'}).current()}") # Find distinct values print(f"Distinct colors: {list(store.distinct(field='color'))}") # Close the store connection (important for persistent stores) store.close() # Example of using a persistent store (e.g., MongoStore) # Requires a MongoDB instance running and pymongo installed. # uri = os.environ.get('MONGO_URI', 'mongodb://localhost:27017/test_db') # from maggma.stores import MongoStore # mongo_store = MongoStore(collection_name='my_collection', database_name='test_db', host=uri, key='name') # try: # mongo_store.connect() # mongo_store.update(turtles, key_field='name', upsert=True) # print(f"MongoStore count: {mongo_store.count()}") # finally: # mongo_store.close()