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]

You need to create a connection in Airflow UI with some other DB type like "MySQL" because it seems MongoDB does not worth to be listed.

There's an open issue for this in the Airflow development, I linked this gist there. There's also another similar gist here.
I look forward to see the hook with MongoDB working soon!

This hook works fine?

There's no option to add a connection for 'MongoDB' from airflow admin UI. Can anyone confirm how to setup mongo connection properly?