nikto_b/breakshaft
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Releases Wiki Activity

Add method add_conversion_points into a ConvRepo

Browse Source
This commit is contained in:
Артур Шевченко
2025-08-20 03:11:17 +03:00
parent 9142cb05fc
commit 5ac6ff102f
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/breakshaft/convertor.py
View File

@@ -79,11 +79,14 @@ class ConvRepo:
def convertor_set(self):
return self._convertor_set
def add_conversion_points(self, conversion_points: Iterable[ConversionPoint]):
self._convertor_set |= set(conversion_points)
def add_injector(self,
func: Callable,
rettype: Optional[type] = None,
type_remap: Optional[dict[str, type]] = None):
self._convertor_set |= set(ConversionPoint.from_fn(func, rettype=rettype, type_remap=type_remap))
self.add_conversion_points(ConversionPoint.from_fn(func, rettype=rettype, type_remap=type_remap))
def _callseq_from_callgraph(self, cg: Callgraph) -> list[ConversionPoint]:
if len(cg.variants) == 0: