Page MenuHomeSoftware Heritage
Files F9348809

base_node.py
No OneTemporary
Actions

base_node.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 abc import ABC
from collections import namedtuple
from swh.graphql.backends.archive import Archive
from swh.graphql.errors import ObjectNotFoundError
class BaseNode(ABC):
"""
Base resolver for all the nodes
"""
def __init__(self, obj, info, node_data=None, **kwargs):
self.obj = obj
self.info = info
self.kwargs = kwargs
self.archive = Archive()
self._node = self._get_node(node_data)
# handle the errors, if any, after _node is set
self._handle_node_errors()
def _get_node(self, node_data):
"""
Get the node object from the given data
if the data (node_data) is none make a function call
to get data from backend
"""
if node_data is None:
node_data = self._get_node_data()
return self._get_node_from_data(node_data)
def _get_node_from_data(self, node_data):
"""
Get the object from node_data
In case of a dict, convert it to an object
Override to support different data structures
"""
if type(node_data) is dict:
return namedtuple("NodeObj", node_data.keys())(*node_data.values())
return node_data
def _handle_node_errors(self):
"""
Handle any error related to node data
raise an error in case the object returned is None
override for specific behaviour
"""
if self._node is None:
raise ObjectNotFoundError("Requested object is not available")
def _get_node_data(self):
"""
Override for desired behaviour
This will be called only when node_data is None
"""
# FIXME, make this call async (not for v1)
return None
def __getattr__(self, name):
"""
Any property defined in the sub-class will get precedence over
the _node attributes
"""
return getattr(self._node, name)
def is_type_of(self):
return self.__class__.__name__
class BaseSWHNode(BaseNode):
"""
Base resolver for all the nodes with a SWHID field
"""
@property
def swhid(self):
return self._node.swhid()

File Metadata

Mime Type
text/x-python
Expires
Jul 4 2025, 6:51 PM (5 w, 6 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3293019

Event Timeline

About · Developer information · Legal