mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
4f7e295a36
Add the revng.api module: a wrapper around the PipelineC API.
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
from typing import Dict, Optional
|
|
|
|
from ._capi import _api, ffi
|
|
from .container import Container, ContainerIdentifier
|
|
from .kind import Kind
|
|
from .utils import make_python_string
|
|
|
|
|
|
class Step:
|
|
def __init__(self, step):
|
|
self._step = step
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
name = _api.rp_step_get_name(self._step)
|
|
return make_python_string(name)
|
|
|
|
def get_parent(self) -> Optional["Step"]:
|
|
_step = _api.rp_step_get_parent(self._step)
|
|
return Step(_step) if _step != ffi.NULL else None
|
|
|
|
def get_container(self, container_identifier: ContainerIdentifier) -> Optional[Container]:
|
|
_container = _api.rp_step_get_container(
|
|
self._step, container_identifier._container_identifier
|
|
)
|
|
return Container(_container, self.name) if _container != ffi.NULL else None
|
|
|
|
def get_artifacts_kind(self) -> Optional[Kind]:
|
|
_kind = _api.rp_step_get_artifacts_kind(self._step)
|
|
return Kind(_kind) if _kind != ffi.NULL else None
|
|
|
|
def get_artifacts_container(self) -> Optional[Container]:
|
|
_container = _api.rp_step_get_artifacts_container(self._step)
|
|
return Container(_container, self.name) if _container != ffi.NULL else None
|
|
|
|
def as_dict(self) -> Dict[str, str]:
|
|
ret = {"name": self.name}
|
|
|
|
parent = self.get_parent()
|
|
if parent is not None:
|
|
ret["parent"] = parent.name
|
|
|
|
return ret
|