From 6cc81453d46077cf6bb2f40adc5e020609ca6e61 Mon Sep 17 00:00:00 2001 From: Jonathan Frederic Date: Wed, 24 Feb 2016 12:18:07 -0800 Subject: [PATCH] Simplify mappings --- notebook/nbextensions.py | 24 ++++-------------------- 1 file changed, 4 insertions(+), 20 deletions(-) diff --git a/notebook/nbextensions.py b/notebook/nbextensions.py index 50fc5f5e3..cfb43fcc9 100644 --- a/notebook/nbextensions.py +++ b/notebook/nbextensions.py @@ -293,16 +293,8 @@ def enable_nbextension_python(package, user=False, sys_prefix=False, log=None): for nbext in nbexts: require = nbext['require'] section = nbext['section'] - if section == 'common': - diff = {'NotebookApp': {'nbextensions_common': {require: True}}} - elif section == 'notebook': - diff = {'NotebookApp': {'nbextensions_notebook': {require: True}}} - elif section == 'tree': - diff = {'NotebookApp': {'nbextensions_tree': {require: True}}} - elif section == 'edit': - diff = {'NotebookApp': {'nbextensions_edit': {require: True}}} - elif section == 'terminal': - diff = {'NotebookApp': {'nbextensions_terminal': {require: True}}} + if section in ['common', 'notebook', 'tree', 'edit', 'terminal']: + diff = {'NotebookApp': {'nbextensions_' + section: {require: True}}} _recursive_update(data, diff) _write_config_data(data, user=user, sys_prefix=sys_prefix) @@ -314,16 +306,8 @@ def disable_nbextension_python(package, user=False, sys_prefix=False): for nbext in nbexts: require = nbext['require'] section = nbext['section'] - if section == 'common': - diff = {'NotebookApp': {'nbextensions_common': {require: False}}} - elif section == 'notebook': - diff = {'NotebookApp': {'nbextensions_notebook': {require: False}}} - elif section == 'tree': - diff = {'NotebookApp': {'nbextensions_tree': {require: False}}} - elif section == 'edit': - diff = {'NotebookApp': {'nbextensions_edit': {require: False}}} - elif section == 'terminal': - diff = {'NotebookApp': {'nbextensions_terminal': {require: False}}} + if section in ['common', 'notebook', 'tree', 'edit', 'terminal']: + diff = {'NotebookApp': {'nbextensions_' + section: {require: False}}} _recursive_update(data, diff) _write_config_data(data, user=user, sys_prefix=sys_prefix)