mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
46 lines
1.2 KiB
Python
46 lines
1.2 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 _api, ffi
|
|
|
|
|
|
def make_python_string(s: ffi.CData, free: bool = False) -> str: # type: ignore
|
|
if s == ffi.NULL:
|
|
return ""
|
|
bytestring = ffi.string(s)
|
|
ret = bytestring.decode("utf-8")
|
|
if free:
|
|
_api.rp_string_destroy(s)
|
|
return ret
|
|
|
|
|
|
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)
|