Changeset View
Changeset View
Standalone View
Standalone View
swh/graphql/resolvers/directory.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.model.model import Directory | from swh.model.model import Directory | ||||
| from swh.model.swhids import ObjectType | from swh.model.swhids import ObjectType | ||||
| from .base_node import BaseSWHNode | from .base_node import BaseSWHNode | ||||
| from .release import BaseReleaseNode | |||||
| from .revision import BaseRevisionNode | from .revision import BaseRevisionNode | ||||
| from .search import SearchResultNode | from .target import TargetNode | ||||
| from .snapshot_branch import BaseSnapshotBranchNode | |||||
| class BaseDirectoryNode(BaseSWHNode): | class BaseDirectoryNode(BaseSWHNode): | ||||
| """ | """ | ||||
| Base resolver for all the directory nodes | Base resolver for all the directory nodes | ||||
| """ | """ | ||||
| def _get_directory_by_id(self, directory_id): | def _get_directory_by_id(self, directory_id): | ||||
| Show All 36 Lines | def _get_node_data(self): | ||||
| return self._get_directory_by_id(self.obj.directory_hash) | return self._get_directory_by_id(self.obj.directory_hash) | ||||
| class TargetDirectoryNode(BaseDirectoryNode): | class TargetDirectoryNode(BaseDirectoryNode): | ||||
| """ | """ | ||||
| Node resolver for a directory requested as a target | Node resolver for a directory requested as a target | ||||
| """ | """ | ||||
| from .directory_entry import BaseDirectoryEntryNode | |||||
| _can_be_null = True | _can_be_null = True | ||||
| obj: Union[ | obj: TargetNode | ||||
| BaseSnapshotBranchNode, | |||||
| BaseReleaseNode, | |||||
| BaseDirectoryEntryNode, | |||||
| SearchResultNode, | |||||
| ] | |||||
| def _get_node_data(self): | def _get_node_data(self): | ||||
| return self._get_directory_by_id(self.obj.target_hash) | return self._get_directory_by_id(self.obj.target_id) | ||||