Implemented IClick

CivityNL · Oct 9, 2024 · 8fa08bd · 8fa08bd
1 parent 072fd01
commit 8fa08bd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/ckanext/civity/cli.py b/ckanext/civity/cli.py
@@ -27,6 +27,7 @@ def init():
                 {'model': model, 'ignore_auth': True, 'user': site_user['name']},
                 pkg_dict
             )
+            click.secho("Updated package: %s" % pkg_id, fg="green", bold=True)
         except Exception as e:
             error_shout(e)
             error_shout(u"Package '{}' was not found".format(pkg_id))
diff --git a/ckanext/civity/plugin.py b/ckanext/civity/plugin.py
@@ -8,6 +8,7 @@
 import ckanext.civity.logic.action.get as action_get
 from ckanext.civity.uploader import CivityResourceUpload
 from ckan.lib.plugins import DefaultTranslation
+import ckanext.civity.cli as cli
 
 
 class CivityPlugin(plugins.SingletonPlugin, DefaultTranslation):
@@ -18,6 +19,7 @@ class CivityPlugin(plugins.SingletonPlugin, DefaultTranslation):
     plugins.implements(plugins.IPackageController, inherit=True)
     plugins.implements(plugins.ITemplateHelpers)
     plugins.implements(plugins.IActions)
+    plugins.implements(plugins.IClick)
 
 
     users_assign_all_groups = toolkit.config.get('civity.users_assign_all_groups', False)
@@ -118,3 +120,7 @@ def get_actions(self):
 
     def get_resource_uploader(self, data_dict):
         return CivityResourceUpload(data_dict)
+
+    # IClick
+    def get_commands(self):
+        return [cli.civity]