#########################################################################
#
# Copyright (C) 2023 Open Source Geospatial Foundation - all rights reserved
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
#########################################################################
import logging
from django.db.models import Count
from geonode.base.models import TopicCategory
from geonode.facets.models import FacetProvider, DEFAULT_FACET_PAGE_SIZE, FACET_TYPE_CATEGORY
[docs]
logger = logging.getLogger(__name__)
[docs]
class CategoryFacetProvider(FacetProvider):
"""
Implements faceting for resource's topicCategory
"""
@property
[docs]
def name(self) -> str:
return "category"
[docs]
def get_info(self, lang="en", **kwargs) -> dict:
return {
"name": self.name,
"filter": "filter{category.identifier.in}",
"label": "Category",
"type": FACET_TYPE_CATEGORY,
}
[docs]
def get_facet_items(
self,
queryset,
start: int = 0,
end: int = DEFAULT_FACET_PAGE_SIZE,
lang="en",
topic_contains: str = None,
keys: set = {},
**kwargs,
) -> (int, list):
logger.debug("Retrieving facets for %s", self.name)
filters = {"resourcebase__in": queryset}
if topic_contains:
filters["gn_description__icontains"] = topic_contains
if keys:
logger.debug("Filtering by keys %r", keys)
filters["identifier__in"] = keys
q = (
TopicCategory.objects.values("identifier", "gn_description", "fa_class")
.filter(**filters)
.annotate(count=Count("resourcebase"))
.order_by("-count")
)
logger.debug(" PREFILTERED QUERY ---> %s\n\n", queryset.query)
logger.debug(" ADDITIONAL FILTERS ---> %s\n\n", filters)
logger.debug(" FINAL QUERY ---> %s\n\n", q.query)
cnt = q.count()
logger.info("Found %d facets for %s", cnt, self.name)
logger.debug(" ---> %s\n\n", q.query)
logger.debug(" ---> %r\n\n", q.all())
topics = [
{
"key": r["identifier"],
"label": r["gn_description"],
"count": r["count"],
"fa_class": r["fa_class"],
}
for r in q[start:end].all()
]
return cnt, topics
[docs]
def get_topics(self, keys: list, lang="en", **kwargs) -> list:
q = TopicCategory.objects.filter(identifier__in=keys)
logger.debug(" ---> %s\n\n", q.query)
logger.debug(" ---> %r\n\n", q.all())
return [
{
"key": r.identifier,
"label": r.gn_description,
"fa_class": r.fa_class,
}
for r in q.all()
]
@classmethod
[docs]
def register(cls, registry, **kwargs) -> None:
registry.register_facet_provider(CategoryFacetProvider(**kwargs))