show collection icon for collection parent

Author: FuhuXia

ckanext/datagovtheme/helpers.py

@@ -9,6 +9,7 @@
 
 import pkg_resources
 
+import ckan.logic as logic
 from ckan import plugins as p, model
 from ckan.lib import base, helpers as h
 from ckan.plugins.toolkit import asbool, config
@@ -670,6 +671,24 @@ def is_tagged_ngda(pkg_dict):
     return False
 
 
+def is_collection_parent(pkg_dict):
+    '''Returns True if the package is a collection parent
+       this relies on the searching with collection_info fq
+       collection_info is handled in geodatagov'''
+    sid = get_pkg_dict_extra(pkg_dict, 'harvest_source_id', None)
+    pid = get_pkg_dict_extra(pkg_dict, 'identifier', None)
+    package_search = logic.get_action('package_search')
+    search_params = {
+        'fq': f'collection_info:"{sid} {pid}"'
+    }
+    base_results = package_search(
+        {'ignore_auth': True},
+        search_params
+    )
+
+    return asbool(base_results['results'])
+
+
 # TODO can we drop this dependency on ckanext-harvest? Can this be moved to ckanext-harvest? geodatagov?
 def get_pkg_dict_extra(pkg_dict, key, default=None):
     '''Override the CKAN core helper to add rolled up extras

ckanext/datagovtheme/plugin.py

@@ -89,6 +89,7 @@ def get_helpers(self):
             'datagovtheme_get_reference_date': datagovtheme_helpers.get_reference_date,
             'datagovtheme_get_responsible_party': datagovtheme_helpers.get_responsible_party,
             'is_tagged_ngda': datagovtheme_helpers.is_tagged_ngda,
+            'is_collection_parent': datagovtheme_helpers.is_collection_parent,
             'render_datetime_datagov': datagovtheme_helpers.render_datetime_datagov,
             'get_harvest_object_formats': datagovtheme_helpers.get_harvest_object_formats,
             'get_bureau_info': datagovtheme_helpers.get_bureau_info,

ckanext/datagovtheme/templates/snippets/package_item.html

@@ -43,7 +43,7 @@ <h3 class="dataset-heading" pkg_id="{{ package.id }}">
           </span>
         {% endif %}
 
-        {% if h.get_pkg_dict_extra(package, 'collection_metadata', '') %}
+        {% if h.is_collection_parent(package) %}
             <span class="dataset-collection label {{ 'label-default' }}">
                 <i class="icon-collection"></i>
                 {{ _('Collection') }}

dev-requirements.txt

@@ -3,5 +3,4 @@ factory-boy>=2
 flake8
 mock
 pytest
-pytest-ckan
 pytest-cov