TRX/tools/generate_init
Fabrice Delliaux 420143d76e
build: improve pyjson5 module import
Trying to build on Gentoo Linux is failing with the following :
> Traceback (most recent call last):
>   File "[...]/TR1X/tools/generate_init", line 7, in <module>
>     import pyjson5
> ModuleNotFoundError: No module named 'pyjson5'

On Gentoo, pyjson5 module is installed as json5.

Signed-off-by: Fabrice Delliaux <netbox253@gmail.com>
2023-12-13 09:31:45 +01:00

67 lines
1.7 KiB
Python
Executable file

#!/usr/bin/env python3
import argparse
import io
import json
from pathlib import Path
try:
import pyjson5
except ImportError:
import json5 as pyjson5
REPO_DIR = Path(__file__).parent.parent
BIN_DIR = REPO_DIR / "bin"
SRC_DIR = REPO_DIR / "src"
CONFIG_DIR = BIN_DIR / "cfg"
GAMEFLOW_PATH = CONFIG_DIR / "TR1X_gameflow.json5"
BODY_PREFIX = """
#include "game/gameflow.h"
#include "global/types.h"
#include <stddef.h>
const char *g_TR1XVersion = "{version}";
GAMEFLOW_DEFAULT_STRING g_GameFlowDefaultStrings[] = {{"""
BODY_SUFFIX = """ { 0, NULL },
};
"""
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser()
parser.add_argument("--version-file", type=Path)
parser.add_argument('-o', "--output", type=Path)
return parser.parse_args()
def get_init_c(version: str) -> str:
gameflow = pyjson5.loads(GAMEFLOW_PATH.read_text(encoding="utf-8"))
with io.StringIO() as handle:
print(BODY_PREFIX.format(version=version).lstrip(), file=handle)
for key, value in gameflow["strings"].items():
print(f" {{ GS_{key}, {json.dumps(value)} }},", file=handle)
print(BODY_SUFFIX.rstrip(), file=handle)
return handle.getvalue()
def update_init_c(output_path: Path, version: str) -> None:
new_text = get_init_c(version=version)
if not output_path.exists() or output_path.read_text() != new_text:
output_path.write_text(new_text)
def main() -> None:
args = parse_args()
if args.version_file and args.version_file.exists():
version = args.version_file.read_text().strip()
else:
version = ""
update_init_c(output_path=args.output, version=version)
if __name__ == "__main__":
main()