MongoDB Hook for Airflow system.

mongodb_hook.py

from airflow.hooks.base_hook import BaseHook
from pymongo import MongoClient


class MongoDBHook(BaseHook):
    def __init__(self, conn_id='mongodb_default'):
        self.conn = self.get_connection(conn_id)
        self.client = MongoClient(host=self.conn.host, port=self.conn.port)

    def __getattr__(self, name):
        return getattr(self.client, name)

    def __getitem__(self, item):
        return self.client[item]