[forms, commands] move code for creating example vocab to a paster command
--- /dev/null
+++ b/ckanext/example/commands.py
@@ -1,1 +1,66 @@
+from ckan import model
+from ckan.lib.cli import CkanCommand
+from ckan.logic import get_action, NotFound
+import forms
+import logging
+log = logging.getLogger()
+
+
+class ExampleCommand(CkanCommand):
+ '''
+ CKAN Example Extension
+
+ Usage::
+
+ paster example create-example-vocab -c <path to config file>
+
+ paster example clean -c <path to config file>
+ - Remove all data created by ckanext-example
+
+ The commands should be run from the ckanext-example directory.
+ '''
+ summary = __doc__.split('\n')[0]
+ usage = __doc__
+
+ def command(self):
+ '''
+ Parse command line arguments and call appropriate method.
+ '''
+ if not self.args or self.args[0] in ['--help', '-h', 'help']:
+ print ExampleCommand.__doc__
+ return
+
+ cmd = self.args[0]
+ self._load_config()
+
+ if cmd == 'create-example-vocab':
+ self.create_example_vocab()
+ else:
+ log.error('Command "%s" not recognized' % (cmd,))
+
+ def create_example_vocab(self):
+ '''
+ Adds an example vocabulary to the database if it doesn't
+ already exist.
+ '''
+ user = get_action('get_site_user')({'model': model, 'ignore_auth': True}, {})
+ context = {'model': model, 'session': model.Session, 'user': user['name']}
+ data = {'id': forms.EXAMPLE_VOCAB}
+
+ try:
+ get_action('vocabulary_show')(context, data)
+ log.info("Example tag vocabulary already exists, skipping.")
+ except NotFound:
+ log.info("Creating example vocab %s" % forms.EXAMPLE_VOCAB)
+ data = {'name': forms.EXAMPLE_VOCAB}
+ vocab = get_action('vocabulary_create')(context, data)
+
+ log.info("Adding tag %s to vocab %s" % ('vocab-tag-example-1', forms.EXAMPLE_VOCAB))
+ data = {'name': 'vocab-tag-example-1', 'vocabulary_id': vocab['id']}
+ get_action('tag_create')(context, data)
+
+ log.info("Adding tag %s to vocab %s" % ('vocab-tag-example-2', forms.EXAMPLE_VOCAB))
+ data = {'name': 'vocab-tag-example-2', 'vocabulary_id': vocab['id']}
+ get_action('tag_create')(context, data)
+
--- a/ckanext/example/forms.py
+++ b/ckanext/example/forms.py
@@ -1,23 +1,26 @@
-import os, logging
+import os
+import logging
from ckan.authz import Authorizer
from ckan.logic.converters import convert_to_extras,\
convert_from_extras, convert_to_tags, convert_from_tags, free_tags_only
-from ckan.logic import get_action
+from ckan.logic import get_action, NotFound
from ckan.logic.schema import package_form_schema, group_form_schema
from ckan.lib.base import c, model
-from ckan.plugins import IDatasetForm, IGroupForm, IConfigurer, IConfigurable
-from ckan.plugins import IGenshiStreamFilter
+from ckan.plugins import IDatasetForm, IGroupForm, IConfigurer
from ckan.plugins import implements, SingletonPlugin
-from ckan.lib.navl.validators import ignore_missing, not_empty, keep_extras
+from ckan.lib.navl.validators import ignore_missing, keep_extras
log = logging.getLogger(__name__)
+
+EXAMPLE_VOCAB = u'example_vocab'
+
class ExampleGroupForm(SingletonPlugin):
"""This plugin demonstrates how a class packaged as a CKAN
extension might extend CKAN behaviour by providing custom forms
based on the type of a Group.
- In this case, we implement two extension interfaces to provide custom
+ In this case, we implement two extension interfaces to provide custom
forms for specific types of group.
- ``IConfigurer`` allows us to override configuration normally
@@ -30,7 +33,7 @@
class will be used.
"""
implements(IGroupForm, inherit=True)
- implements(IConfigurer, inherit=True)
+ implements(IConfigurer, inherit=True)
def update_config(self, config):
"""
@@ -48,7 +51,7 @@
"""
Returns a string representing the location of the template to be
rendered. e.g. "forms/group_form.html".
- """
+ """
return 'forms/group_form.html'
def group_types(self):
@@ -111,14 +114,12 @@
found in the ``ini``-file. Here we use it to specify where the
form templates can be found.
- ``IDatasetForm`` allows us to provide a custom form for a dataset
- based on the type_name that may be set for a package. Where the
- type_name matches one of the values in package_types then this
- class will be used.
+ based on the type_name that may be set for a package. Where the
+ type_name matches one of the values in package_types then this
+ class will be used.
"""
implements(IDatasetForm, inherit=True)
- implements(IConfigurer, inherit=True)
- implements(IConfigurable)
- implements(IGenshiStreamFilter)
+ implements(IConfigurer, inherit=True)
def update_config(self, config):
"""
@@ -132,45 +133,11 @@
config['extra_template_paths'] = ','.join([template_dir,
config.get('extra_template_paths', '')])
- def configure(self, config):
- '''
- Adds some new vocabularies to the database if they don't already exist.
-
- '''
- # Add a 'genre' vocabulary with some tags.
- self.genre_vocab = model.Vocabulary.get('Genre')
- if not self.genre_vocab:
- log.info("Adding vocab Genre")
- self.genre_vocab = model.Vocabulary('Genre')
- model.Session.add(self.genre_vocab)
- model.Session.commit()
- log.info("Adding example tags to vocab %s" % self.genre_vocab.name)
- jazz_tag = model.Tag('jazz', self.genre_vocab.id)
- soul_tag = model.Tag('soul', self.genre_vocab.id)
- model.Session.add(jazz_tag)
- model.Session.add(soul_tag)
- model.Session.commit()
-
- # Add a 'composer' vocabulary with some tags.
- self.composer_vocab = model.Vocabulary.get('Composer')
- if not self.composer_vocab:
- log.info("Adding vocab Composer")
- self.composer_vocab = model.Vocabulary('Composer')
- model.Session.add(self.composer_vocab)
- model.Session.commit()
- log.info("Adding example tags to vocab %s" %
- self.composer_vocab.name)
- mintzer_tag = model.Tag('Bob Mintzer', self.composer_vocab.id)
- lewis_tag = model.Tag('Steve Lewis', self.composer_vocab.id)
- model.Session.add(mintzer_tag)
- model.Session.add(lewis_tag)
- model.Session.commit()
-
def package_form(self):
"""
Returns a string representing the location of the template to be
rendered. e.g. "package/new_package_form.html".
- """
+ """
return 'forms/dataset_form.html'
def is_fallback(self):
@@ -178,7 +145,7 @@
Returns true iff this provides the fallback behaviour, when no other
plugin instance matches a package's type.
- As this is not the fallback controller we should return False. If
+ As this is not the fallback controller we should return False. If
we were wanting to act as the fallback, we'd return True
"""
return True
@@ -200,13 +167,15 @@
"""
Adds variables to c just prior to the template being rendered that can
then be used within the form
- """
+ """
c.licences = [('', '')] + model.Package.get_license_options()
c.publishers = [('Example publisher', 'Example publisher 2')]
c.is_sysadmin = Authorizer().is_sysadmin(c.user)
c.resource_columns = model.Resource.get_columns()
- c.genre_tags = get_action('tag_list')(context, {'vocabulary_id': self.genre_vocab.name})
- c.composer_tags = get_action('tag_list')(context, {'vocabulary_id': self.composer_vocab.name})
+ try:
+ c.vocab_tags = get_action('tag_list')(context, {'vocabulary_id': EXAMPLE_VOCAB})
+ except NotFound:
+ c.vocab_tags = None
## This is messy as auths take domain object not data_dict
pkg = context.get('package') or c.pkg
@@ -221,9 +190,8 @@
"""
schema = package_form_schema()
schema.update({
- 'published_by': [not_empty, unicode, convert_to_extras],
- 'genre_tags': [ignore_missing, convert_to_tags(self.genre_vocab.name)],
- 'composer_tags': [ignore_missing, convert_to_tags(self.composer_vocab.name)],
+ 'published_by': [ignore_missing, unicode, convert_to_extras],
+ 'vocab_tags': [ignore_missing, convert_to_tags(EXAMPLE_VOCAB)],
})
return schema
@@ -237,10 +205,7 @@
'tags': {
'__extras': [keep_extras, free_tags_only]
},
- 'genre_tags_selected': [convert_from_tags(self.genre_vocab.name),
- ignore_missing],
- 'composer_tags_selected': [
- convert_from_tags(self.composer_vocab.name), ignore_missing],
+ 'vocab_tags_selected': [convert_from_tags(EXAMPLE_VOCAB), ignore_missing],
'published_by': [convert_from_extras, ignore_missing],
})
return schema
@@ -251,26 +216,4 @@
"""
return
- def filter(self, stream):
- # Add vocab tags to the bottom of the sidebar.
- from pylons import request
- from genshi.filters import Transformer
- from genshi.input import HTML
- routes = request.environ.get('pylons.routes_dict')
- if routes.get('controller') == 'package' \
- and routes.get('action') == 'read':
- for vocab in (self.genre_vocab, self.composer_vocab):
- vocab_tags = [tag for tag in c.pkg_dict.get('tags', [])
- if tag.get('vocabulary_id') == vocab.id]
- if not vocab_tags:
- continue
- html = '<li class="sidebar-section">'
- html = html + '<h3>%s</h3>' % vocab.name
- html = html + '<ul class="tags clearfix">'
- for tag in vocab_tags:
- html = html + '<li>%s</li>' % tag.get('name')
- html = html + "</ul></li>"
- stream = stream | Transformer("//div[@id='sidebar']")\
- .append(HTML(html))
- return stream
-
+
--- a/ckanext/example/theme/templates/forms/dataset_form.html
+++ b/ckanext/example/theme/templates/forms/dataset_form.html
@@ -143,39 +143,22 @@
<dd class="tags-instructions hints">e.g. pollution, rivers, water quality</dd>
<dd class="tags-instructions field_error" py:if="errors.get('tag_string', '')">${errors.get('tag_string', '')}</dd>
</dl>
-
- <h3>Musical Genre</h3>
- <dl>
+ <h3 py:if="c.vocab_tags">Tags With Example Vocabulary</h3>
+ <dl py:if="c.vocab_tags">
<dt class="tags-label"><label class="field_opt" for="tags_vocab">Tags</label></dt>
<dd class="tags-field">
- <select id="genre_tags" class="chzn-select" name="genre_tags" size="60" multiple="multiple">
- <py:for each="tag in c.genre_tags">
+ <select id="vocab_tags" class="chzn-select" name="vocab_tags" size="60" multiple="multiple">
+ <py:for each="tag in c.vocab_tags">
<py:choose test="">
- <option py:when="tag in data.get('genre_tags_selected', [])" selected="selected" value="${tag}">${tag}</option>
+ <option py:when="tag in data.get('vocab_tags_selected', [])" selected="selected" value="${tag}">${tag}</option>
<option py:otherwise="" value="${tag}">${tag}</option>
</py:choose>
</py:for>
</select>
</dd>
- <dd class="tags-instructions field_error" py:if="errors.get('genre_tag_string', '')">${errors.get('genre_tag_string', '')}</dd>
- </dl>
-
- <h3>Composer</h3>
- <dl>
- <dt class="tags-label"><label class="field_opt" for="tags_vocab">Tags</label></dt>
- <dd class="tags-field">
- <select id="composer_tags" class="chzn-select" name="composer_tags" size="60" multiple="multiple">
- <py:for each="tag in c.composer_tags">
- <py:choose test="">
- <option py:when="tag in data.get('composer_tags_selected', [])" selected="selected" value="${tag}">${tag}</option>
- <option py:otherwise="" value="${tag}">${tag}</option>
- </py:choose>
- </py:for>
- </select>
- </dd>
- <dd class="tags-instructions field_error" py:if="errors.get('composer_tag_string', '')">${errors.get('composer_tag_string', '')}</dd>
- </dl>
-
+ <dd class="tags-instructions instructions basic" i18n:msg="">Similar to standard CKAN tags, but these use a fixed vocabulary.</dd>
+ <dd class="tags-instructions field_error" py:if="errors.get('vocab_tag_string', '')">${errors.get('vocab_tag_string', '')}</dd>
+ </dl>
</fieldset>
<fieldset id='further-information'>
<dl>
--- a/setup.py
+++ b/setup.py
@@ -17,9 +17,7 @@
namespace_packages=['ckanext', 'ckanext.example'],
include_package_data=True,
zip_safe=False,
- install_requires=[
- # -*- Extra requirements: -*-
- ],
+ install_requires=[],
entry_points=\
"""
[ckan.plugins]
@@ -29,6 +27,9 @@
[ckan.forms]
example_form = ckanext.example.package_form:get_example_fieldset
+
+ [paste.paster_command]
+ example=ckanext.example.commands:ExampleCommand
""",
)