From ff3c71f186c85625a4be73ff878ffc8a32439d39 Mon Sep 17 00:00:00 2001 From: David Raznick Date: Mon, 24 May 2021 17:57:08 +0100 Subject: [PATCH 1/2] Add choose your own magic command. --- noteql/__init__.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/noteql/__init__.py b/noteql/__init__.py index 36e89df..545e307 100755 --- a/noteql/__init__.py +++ b/noteql/__init__.py @@ -25,6 +25,8 @@ LOCAL_DB_MADE = False +REGISTERED_MAGICS = False + def get_engine(dburi): if not dburi: @@ -110,7 +112,8 @@ def __init__( df_viewer=None, df_viewer_kw=None, datasette_url=None, - cell_magic_output=False + cell_magic_name=None, + cell_magic_output=False, ): self.schema = schema self.dburi = dburi @@ -150,7 +153,17 @@ def __init__( connection.execute("create schema if not exists {};".format(self.schema)) self.ipython = get_ipython() - self.ipython.register_magics(Noteql) + + global REGISTERED_MAGICS + + if not REGISTERED_MAGICS: + if cell_magic_name: + Noteql.magics["line"][cell_magic_name] = Noteql.magics["line"].pop("nql") + Noteql.magics["cell"][cell_magic_name] = Noteql.magics["cell"].pop("nql") + + self.ipython.register_magics(Noteql) + REGISTERED_MAGICS = True + self.set() def tables(self): From e7700036f581ac585418052168607b11ecd3cc93 Mon Sep 17 00:00:00 2001 From: David Raznick Date: Mon, 24 May 2021 18:37:45 +0100 Subject: [PATCH 2/2] add changelog --- CHANGELOG.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index dfc2fd5..cd9a38d 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0. ## [Unreleased] +### Added + +- Way to define your own magic function. ## [0.5.0] - 2021-05-22