From fcce567efbcbb6ad72feac7a3834f135a0a4cb35 Mon Sep 17 00:00:00 2001 From: ComicIronic Date: Thu, 21 May 2020 19:24:46 +0100 Subject: [PATCH] Add argument to format JSON output (#3) --- rsi/rsi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rsi/rsi.py b/rsi/rsi.py index ed3c3a6..6764146 100644 --- a/rsi/rsi.py +++ b/rsi/rsi.py @@ -28,7 +28,7 @@ def new_state(self, directions: int, name: str) -> State: self.set_state(newstate, name) return newstate - def write(self, path: Union[str, Path], make_parent_dirs: bool = True) -> None: + def write(self, path: Union[str, Path], make_parent_dirs: bool = True, indent: Optional[int] = None) -> None: if isinstance(path, str): path = Path(path) @@ -67,7 +67,7 @@ def write(self, path: Union[str, Path], make_parent_dirs: bool = True) -> None: metajson["states"] = states with metapath.open("w") as f: - f.write(json.dumps(metajson)) + f.write(json.dumps(metajson, indent=indent)) # Write PNG files. for state in self.states.values():