add error_shot to toolkit

ckan · Dec 10, 2019 · b47341d · b47341d
1 parent dd4f1c1
commit b47341d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ckan/plugins/toolkit.py b/ckan/plugins/toolkit.py
@@ -76,6 +76,8 @@ class _Toolkit(object):
         'CkanCommand',
         # function for initializing CLI interfaces
         'load_config',
+        # function to promt the exception in CLI command
+        'error_shout',
         # base class for IDatasetForm plugins
         'DefaultDatasetForm',
         # base class for IGroupForm plugins
@@ -137,7 +139,8 @@ def _initialize(self):
         import ckan.logic.validators as logic_validators
         import ckan.lib.navl.dictization_functions as dictization_functions
         import ckan.lib.helpers as h
-        import ckan.lib.cli as cli
+        import ckan.lib.cli as old_cli
+        import ckan.cli as cli
         import ckan.lib.plugins as lib_plugins
         import ckan.common as common
         from ckan.exceptions import (
@@ -250,8 +253,9 @@ def _initialize(self):
         t['UnknownValidator'] = logic.UnknownValidator
         t['Invalid'] = logic_validators.Invalid
 
-        t['CkanCommand'] = cli.CkanCommand
-        t['load_config'] = cli.load_config
+        t['CkanCommand'] = old_cli.CkanCommand
+        t['load_config'] = old_cli.load_config
+        t['error_shout'] = cli.error_shout
         t['DefaultDatasetForm'] = lib_plugins.DefaultDatasetForm
         t['DefaultGroupForm'] = lib_plugins.DefaultGroupForm
         t['DefaultOrganizationForm'] = lib_plugins.DefaultOrganizationForm