[forms, commands, templates] Add extended example vocabs and tags to paster command and reintegrate with form
[forms, commands, templates] Add extended example vocabs and tags to paster command and reintegrate with form

--- a/ckanext/example/commands.py
+++ b/ckanext/example/commands.py
@@ -13,7 +13,7 @@
 
     Usage::
 
-        paster example create-example-vocab -c <path to config file>
+        paster example create-example-vocabs -c <path to config file>
 
         paster example clean -c <path to config file>
             - Remove all data created by ckanext-example
@@ -34,33 +34,45 @@
         cmd = self.args[0]
         self._load_config()
 
-        if cmd == 'create-example-vocab':
-            self.create_example_vocab()
+        if cmd == 'create-example-vocabs':
+            self.create_example_vocabs()
         else:
             log.error('Command "%s" not recognized' % (cmd,))
 
-    def create_example_vocab(self):
+    def create_example_vocabs(self):
         '''
-        Adds an example vocabulary to the database if it doesn't
-        already exist.
+        Adds example vocabularies to the database if they don'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:
+            data = {'id': forms.GENRE_VOCAB}
             get_action('vocabulary_show')(context, data)
-            log.info("Example tag vocabulary already exists, skipping.")
+            log.info("Example genre vocabulary already exists, skipping.")
         except NotFound:
-            log.info("Creating example vocab %s" % forms.EXAMPLE_VOCAB)
-            data = {'name': forms.EXAMPLE_VOCAB}
+            log.info("Creating vocab %s" % forms.GENRE_VOCAB)
+            data = {'name': forms.GENRE_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']}
+            log.info("Adding tag %s to vocab %s" % ('jazz', forms.GENRE_VOCAB))
+            data = {'name': 'jazz', 'vocabulary_id': vocab['id']}
+            get_action('tag_create')(context, data)
+            log.info("Adding tag %s to vocab %s" % ('soul', forms.GENRE_VOCAB))
+            data = {'name': 'soul', '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']}
+        try:
+            data = {'id': forms.COMPOSER_VOCAB}
+            get_action('vocabulary_show')(context, data)
+            log.info("Example composer vocabulary already exists, skipping.")
+        except NotFound:
+            log.info("Creating vocab %s" % forms.COMPOSER_VOCAB)
+            data = {'name': forms.COMPOSER_VOCAB}
+            vocab = get_action('vocabulary_create')(context, data)
+            log.info("Adding tag %s to vocab %s" % ('Bob Mintzer', forms.COMPOSER_VOCAB))
+            data = {'name': 'Bob Mintzer', 'vocabulary_id': vocab['id']}
+            get_action('tag_create')(context, data)
+            log.info("Adding tag %s to vocab %s" % ('Steve Lewis', forms.COMPOSER_VOCAB))
+            data = {'name': 'Steve Lewis', 'vocabulary_id': vocab['id']}
             get_action('tag_create')(context, data)
 

--- a/ckanext/example/forms.py
+++ b/ckanext/example/forms.py
@@ -7,13 +7,13 @@
 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
-from ckan.plugins import IGenshiStreamFilter
 from ckan.plugins import implements, SingletonPlugin
 from ckan.lib.navl.validators import ignore_missing, keep_extras
 
 log = logging.getLogger(__name__)
 
-EXAMPLE_VOCAB = u'example_vocab'
+GENRE_VOCAB = u'genre_vocab'
+COMPOSER_VOCAB = u'composer_vocab'
 
 
 class ExampleGroupForm(SingletonPlugin):
@@ -121,7 +121,6 @@
     """
     implements(IDatasetForm, inherit=True)
     implements(IConfigurer, inherit=True)    
-    implements(IGenshiStreamFilter)
     
     def update_config(self, config):
         """
@@ -135,40 +134,6 @@
         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
@@ -209,11 +174,11 @@
         c.is_sysadmin = Authorizer().is_sysadmin(c.user)
         c.resource_columns = model.Resource.get_columns()
         try:
-            c.vocab_tags = get_action('tag_list')(context, {'vocabulary_id': EXAMPLE_VOCAB})
-            # 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})
+            c.genre_tags = get_action('tag_list')(context, {'vocabulary_id': GENRE_VOCAB})
+            c.composer_tags = get_action('tag_list')(context, {'vocabulary_id': COMPOSER_VOCAB})
         except NotFound:
             c.vocab_tags = None
+            c.composer_tags = None
 
         ## This is messy as auths take domain object not data_dict
         pkg = context.get('package') or c.pkg
@@ -229,9 +194,8 @@
         schema = package_form_schema()
         schema.update({
             'published_by': [ignore_missing, unicode, convert_to_extras],
-            'vocab_tags': [ignore_missing, convert_to_tags(EXAMPLE_VOCAB)],
-            # 'genre_tags': [ignore_missing, convert_to_tags(self.genre_vocab.name)],
-            # 'composer_tags': [ignore_missing, convert_to_tags(self.composer_vocab.name)],
+            'genre_tags': [ignore_missing, convert_to_tags(GENRE_VOCAB)],
+            'composer_tags': [ignore_missing, convert_to_tags(COMPOSER_VOCAB)]
         })
         return schema
     
@@ -245,11 +209,12 @@
             'tags': {
                 '__extras': [keep_extras, free_tags_only]
             },
-            'vocab_tags_selected': [convert_from_tags(EXAMPLE_VOCAB), ignore_missing],
-            # 'genre_tags_selected': [convert_from_tags(self.genre_vocab.name),
-            #     ignore_missing],
-            # 'composer_tags_selected': [
-            #     convert_from_tags(self.composer_vocab.name), ignore_missing],
+            'genre_tags_selected': [
+                convert_from_tags(GENRE_VOCAB), ignore_missing
+            ],
+            'composer_tags_selected': [
+                convert_from_tags(COMPOSER_VOCAB), ignore_missing
+            ],
             'published_by': [convert_from_extras, ignore_missing],
         })
         return schema
@@ -259,9 +224,6 @@
         Check if the return data is correct and raises a DataError if not.
         """
         return
-
-    def filter(self, stream):
-        return stream
 
     # def filter(self, stream):
     #     # Add vocab tags to the bottom of the sidebar.

--- a/ckanext/example/theme/templates/forms/dataset_form.html
+++ b/ckanext/example/theme/templates/forms/dataset_form.html
@@ -143,40 +143,38 @@
     <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 py:if="c.vocab_tags">Tags With Example Vocabulary</h3>
-  <dl py:if="c.vocab_tags">
-
-  <!-- <h3>Musical Genre</h3> -->
-  <!-- <dl> -->
-  <!--   <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"> -->
-  <!--         <py:choose test=""> -->
-  <!--         <option py:when="tag in data.get('genre_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> -->
+
+  <h3 py:if="c.genre_tags">Musical Genre</h3>
+  <dl py:if="c.genre_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">
+          <py:choose test="">
+          <option py:when="tag in data.get('genre_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 py:if="c.composer_tags">Composer</h3>
+  <dl py:if="c.composer_tags">
+    <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>
 
 </fieldset>
 <fieldset id='further-information'>