diff --git a/swh/graphql/resolvers/content.py b/swh/graphql/resolvers/content.py --- a/swh/graphql/resolvers/content.py +++ b/swh/graphql/resolvers/content.py @@ -8,6 +8,7 @@ from .base_node import BaseSWHNode from .directory_entry import BaseDirectoryEntryNode from .release import BaseReleaseNode +from .search import SearchResultNode from .snapshot_branch import BaseSnapshotBranchNode @@ -87,7 +88,12 @@ This request could be from directory entry, release or a branch """ - obj: Union[BaseDirectoryEntryNode, BaseReleaseNode, BaseSnapshotBranchNode] + obj: Union[ + SearchResultNode, + BaseDirectoryEntryNode, + BaseReleaseNode, + BaseSnapshotBranchNode, + ] def _get_node_data(self): return self._get_content_by_hash(checksums={"sha1_git": self.obj.target_hash}) diff --git a/swh/graphql/resolvers/directory.py b/swh/graphql/resolvers/directory.py --- a/swh/graphql/resolvers/directory.py +++ b/swh/graphql/resolvers/directory.py @@ -11,6 +11,7 @@ from .base_node import BaseSWHNode from .release import BaseReleaseNode from .revision import BaseRevisionNode +from .search import SearchResultNode from .snapshot_branch import BaseSnapshotBranchNode @@ -65,7 +66,12 @@ from .directory_entry import BaseDirectoryEntryNode - obj: Union[BaseSnapshotBranchNode, BaseReleaseNode, BaseDirectoryEntryNode] + obj: Union[ + BaseSnapshotBranchNode, + BaseReleaseNode, + BaseDirectoryEntryNode, + SearchResultNode, + ] def _get_node_data(self): return self._get_directory_by_id(self.obj.target_hash) diff --git a/swh/graphql/resolvers/release.py b/swh/graphql/resolvers/release.py --- a/swh/graphql/resolvers/release.py +++ b/swh/graphql/resolvers/release.py @@ -6,6 +6,7 @@ from typing import Union from .base_node import BaseSWHNode +from .search import SearchResultNode from .snapshot_branch import BaseSnapshotBranchNode @@ -45,7 +46,7 @@ Node resolver for a release requested as a target """ - obj: Union[BaseSnapshotBranchNode, BaseReleaseNode] + obj: Union[BaseSnapshotBranchNode, BaseReleaseNode, SearchResultNode] def _get_node_data(self): # self.obj.target_hash is the requested release id diff --git a/swh/graphql/resolvers/revision.py b/swh/graphql/resolvers/revision.py --- a/swh/graphql/resolvers/revision.py +++ b/swh/graphql/resolvers/revision.py @@ -14,6 +14,7 @@ from .base_node import BaseSWHNode from .directory_entry import BaseDirectoryEntryNode from .release import BaseReleaseNode +from .search import SearchResultNode from .snapshot_branch import BaseSnapshotBranchNode @@ -60,7 +61,12 @@ Node resolver for a revision requested as a target """ - obj: Union[BaseSnapshotBranchNode, BaseReleaseNode, BaseDirectoryEntryNode] + obj: Union[ + BaseSnapshotBranchNode, + BaseReleaseNode, + BaseDirectoryEntryNode, + SearchResultNode, + ] def _get_node_data(self): # self.obj.target_hash is the requested revision id diff --git a/swh/graphql/resolvers/snapshot.py b/swh/graphql/resolvers/snapshot.py --- a/swh/graphql/resolvers/snapshot.py +++ b/swh/graphql/resolvers/snapshot.py @@ -13,6 +13,7 @@ from .base_connection import BaseConnection from .base_node import BaseSWHNode from .origin import OriginNode +from .search import SearchResultNode from .visit_status import BaseVisitStatusNode @@ -68,7 +69,7 @@ from .snapshot_branch import BaseSnapshotBranchNode - obj: Union[BaseVisitStatusNode, BaseSnapshotBranchNode] + obj: Union[SearchResultNode, BaseSnapshotBranchNode] def _get_node_data(self): snapshot_id = self.obj.target_hash