mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
af237074f3
Implement the new functionality of AnalysesList on the python's side of rev.ng and the GraphQL schema both in static and autogenerated form
100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
from collections.abc import Sequence
|
|
from itertools import islice
|
|
from typing import Any, Dict, Generator, Optional, overload
|
|
|
|
from ._capi import _api, ffi
|
|
from .kind import Kind
|
|
from .utils import make_generator, make_python_string
|
|
|
|
|
|
class Analysis:
|
|
def __init__(self, analysis):
|
|
self._analysis = analysis
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
_name = _api.rp_analysis_get_name(self._analysis)
|
|
return make_python_string(_name)
|
|
|
|
def get_arguments_count(self) -> int:
|
|
return _api.rp_analysis_get_arguments_count(self._analysis)
|
|
|
|
def arguments(self) -> Generator["AnalysisArgument", None, None]:
|
|
return make_generator(self.get_arguments_count(), lambda idx: AnalysisArgument(self, idx))
|
|
|
|
def as_dict(self) -> Dict[str, Any]:
|
|
return {"name": self.name, "arguments": [a.as_dict() for a in self.arguments()]}
|
|
|
|
|
|
class AnalysisArgument:
|
|
def __init__(self, analysis: Analysis, index: int):
|
|
self.analysis = analysis
|
|
self.index = index
|
|
|
|
@property
|
|
def name(self):
|
|
_name = _api.rp_analysis_get_argument_name(self.analysis._analysis, self.index)
|
|
return make_python_string(_name)
|
|
|
|
def acceptable_kinds(self) -> Generator[Kind, None, None]:
|
|
return make_generator(self.acceptable_kinds_count(), self._get_acceptable_kind_from_index)
|
|
|
|
def acceptable_kinds_count(self) -> int:
|
|
return _api.rp_analysis_get_argument_acceptable_kinds_count(
|
|
self.analysis._analysis, self.index
|
|
)
|
|
|
|
def _get_acceptable_kind_from_index(self, idx) -> Optional[Kind]:
|
|
_kind = _api.rp_analysis_get_argument_acceptable_kind(
|
|
self.analysis._analysis, self.index, idx
|
|
)
|
|
return Kind(_kind) if _kind != ffi.NULL else None
|
|
|
|
def as_dict(self):
|
|
return {
|
|
"name": self.name,
|
|
"acceptableKinds": [k.as_dict() for k in self.acceptable_kinds()],
|
|
}
|
|
|
|
|
|
class AnalysesList(Sequence[Analysis]):
|
|
def __init__(self, analyses_list, manager):
|
|
self._analyses_list = analyses_list
|
|
self._manager = manager
|
|
|
|
@property
|
|
def name(self):
|
|
_name = _api.rp_analyses_list_get_name(self._analyses_list)
|
|
return make_python_string(_name)
|
|
|
|
def __len__(self) -> int:
|
|
return _api.rp_analyses_list_count(self._analyses_list)
|
|
|
|
def _get_analysis(self, index: int) -> Analysis:
|
|
_analysis = _api.rp_manager_get_analysis(self._manager, self._analyses_list, index)
|
|
return Analysis(_analysis)
|
|
|
|
@overload
|
|
def __getitem__(self, idx: int) -> Analysis:
|
|
...
|
|
|
|
@overload
|
|
def __getitem__(self, idx: slice) -> Sequence[Analysis]:
|
|
...
|
|
|
|
def __getitem__(self, idx: int | slice):
|
|
if isinstance(idx, int):
|
|
if idx < len(self):
|
|
return self._get_analysis(idx)
|
|
else:
|
|
raise IndexError("list index out of range")
|
|
else:
|
|
return list(islice(self, idx.start, idx.stop, idx.step))
|
|
|
|
def as_dict(self):
|
|
return {"name": self.name, "analyses": [a.as_dict() for a in self]}
|