mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
7c3d9a3fcd
Use the owning annotation in PipelineC/Prototypes.h to automatically destroy resources when they go out of scope on the python side.
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
from pathlib import Path
|
|
from typing import Callable, Generator, Optional, TypeVar, Union
|
|
|
|
from ._capi import ffi
|
|
|
|
|
|
def make_python_string(s: ffi.CData) -> str:
|
|
if s == ffi.NULL:
|
|
return ""
|
|
bytestring = ffi.string(s)
|
|
return bytestring.decode("utf-8")
|
|
|
|
|
|
def make_c_string(s: Union[bytes, str]):
|
|
if isinstance(s, bytes):
|
|
return ffi.new("char[]", s)
|
|
if isinstance(s, str):
|
|
return ffi.new("char[]", s.encode("utf-8"))
|
|
raise TypeError(f"Invalid type: {s.__class__}")
|
|
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def make_generator(count: int, get_idx: Callable[[int], Optional[T]]) -> Generator[T, None, None]:
|
|
for i in range(count):
|
|
ret = get_idx(i)
|
|
if ret is not None:
|
|
yield ret
|
|
|
|
|
|
def save_file(path: Path, content: Union[bytes, str]):
|
|
if isinstance(content, str):
|
|
with open(path.resolve(), "w", encoding="utf-8") as str_f:
|
|
str_f.write(content)
|
|
else:
|
|
with open(path.resolve(), "wb") as bytes_f:
|
|
bytes_f.write(content)
|