Simple cell and line IPython Notebook magics for connecting to a ODBC DB and querying

sqlmagic.py

from IPython.core.magic import (Magics, magics_class, line_magic,
                                cell_magic, line_cell_magic)
import pyodbc
import pandas.io.sql as sql


@magics_class
class SQLMagic(Magics):
    con = None

    def __init__(self, shell):
        super(SQLMagic, self).__init__(shell)

    @line_magic
    def odbc(self, line):
        "Connect to an ODBC database (DSN) configured in your odbc.ini"
        self.con = pyodbc.connect('DSN='+line.strip())

    @line_cell_magic
    def sql(self, line, cell=None):
        "Run a SQL query"
        query = None
        if cell:
            query = cell
        else:
            query = line

        df = sql.read_frame(query, self.con)
        self.shell.push({'last_df': df})
        return df

def load_ipython_extension(ip):
    ip.register_magics(SQLMagic)