mongo.py

import logging
from motor.motor_asyncio import AsyncIOMotorClient

from opsdroid.database import Database


class DatabaseMongo(Database):
    """A module for opsdroid to allow memory to persist in a mongo database."""

    def __init__(self, config):
        """Start the database connection."""
        logging.debug("Loaded mongo database connector")
        self.name = "mongo"
        self.config = config
        self.client = None
        self.db = None

    async def connect(self, opsdroid):
        """Connect to the database."""
        host = self.config["host"] if "host" in self.config else "localhost"
        port = self.config["port"] if "port" in self.config else "27017"
        database = self.config["database"] \
            if "database" in self.config else "opsdroid"
        path = "mongodb://" + host + ":" + port
        self.client = AsyncIOMotorClient(path)
        self.db = self.client[database]
        logging.info("Connected to mongo")

    async def put(self, key, data):
        """Insert or replace an object into the database for a given key."""
        logging.debug("Putting " + key + " into mongo")
        if "_id" in data:
            await self.db[key].update_one({"_id": data["_id"]}, {"$set": data})
        else:
            await self.db[key].insert_one(data)

    async def get(self, key):
        """Get a document from the database for a given key."""
        logging.debug("Getting " + key + " from mongo")
        return await self.db[key].find_one(
                        {"$query": {}, "$orderby": {"$natural" : -1}}
                        )