Differential D7942 Diff 28624 swh/graphql/resolvers/content.py

Changeset View

Standalone View

swh/graphql/resolvers/content.py

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

				from typing import Union

	from swh.graphql.backends import archive			from swh.graphql.backends import archive

	from .base_node import BaseSWHNode			from .base_node import BaseSWHNode
				from .directory_entry import DirectoryEntryNode
				from .release import BaseReleaseNode


	class BaseContentNode(BaseSWHNode):			class BaseContentNode(BaseSWHNode):
	""" """			"""
				Base resolver for all the content nodes
				"""

	def _get_content_by_id(self, content_id):			def _get_content_by_id(self, content_id):
	content = archive.Archive().get_content(content_id)			content = archive.Archive().get_content(content_id)
	return content[0] if content else None			return content[0] if content else None

	@property			@property
	def checksum(self):			def checksum(self):
	# FIXME, return a Node object			# FIXME, return a Node object
	return {k: v.hex() for (k, v) in self._node.hashes().items()}			return {k: v.hex() for (k, v) in self._node.hashes().items()}

	@property			@property
	def id(self):			def id(self):
	return self._node.sha1_git			return self._node.sha1_git

	def is_type_of(self):			def is_type_of(self):
				# is_type_of is required only when resolving a UNION type
				# This is for ariadne to return the right type
	return "Content"			return "Content"


	class ContentNode(BaseContentNode):			class ContentNode(BaseContentNode):
	def _get_node_data(self):
	"""			"""
	When a content is requested directly			Node resolver for a content requested directly with its SWHID
	with its SWHID
	"""			"""

				def _get_node_data(self):
	return self._get_content_by_id(self.kwargs.get("swhid").object_id)			return self._get_content_by_id(self.kwargs.get("swhid").object_id)


	class TargetContentNode(BaseContentNode):			class TargetContentNode(BaseContentNode):
	def _get_node_data(self):
	"""			"""
	When a content is requested from a			Node resolver for a content requested from a
	directory entry or from a release target			directory entry or from a release target

	content id is obj.targetHash here
	"""			"""

				obj: Union[DirectoryEntryNode, BaseReleaseNode]

				vlorentzUnsubmitted Done Inline Actions redundant vlorentz: redundant
				def _get_node_data(self):
	content_id = self.obj.targetHash			content_id = self.obj.targetHash
	return self._get_content_by_id(content_id)			return self._get_content_by_id(content_id)