search.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

search.py
View Options

	# Copyright (C) 2022 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information

	from swh.storage.interface import PagedResult

	from .base_connection import BaseConnection
	from .base_node import BaseNode


	class SearchResultNode(BaseNode):
	""" """

	@property
	def targetType(self): # To support the schema naming convention
	return self._node.type


	class ResolveSwhidConnection(BaseConnection):

	_node_class = SearchResultNode

	def _get_paged_result(self) -> PagedResult:
	swhid = self.kwargs.get("swhid")
	results = []
	if self.archive.is_object_available(swhid.object_id, swhid.object_type):
	results = [
	{
	"target_hash": swhid.object_id,
	"type": swhid.object_type.name.lower(),
	}
	]
	return PagedResult(results=results)


	class SearchConnection(BaseConnection):

	_node_class = SearchResultNode

	def _get_paged_result(self) -> PagedResult:
	origins = self.search.get_origins(
	query=self.kwargs.get("query"),
	after=self._get_after_arg(),
	first=self._get_first_arg(),
	)

	# FIXME hard coding type to origin for now, as it is the only searchable object
	results = [
	{"target_url": ori["url"], "type": "origin"} for ori in origins.results
	]
	return PagedResult(results=results, next_page_token=origins.next_page_token)