Garmelon/pasch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow specifying formatting options for json files

Browse source
This commit is contained in:
Joscha 2026-02-17 23:07:13 +01:00
parent e236973e82
commit 034d56b733
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pasch/file/json.py
View file

@ -42,8 +42,15 @@ class JsonFileProxy:
class JsonFile(File): class JsonFile(File):
def __init__(self, data: Any = None) -> None: def __init__(
self,
data: Any = None,
indent: int = 2,
trailing_newline: bool = True,
) -> None:
self.data = {} if data is None else data self.data = {} if data is None else data
self.indent = indent
self.trailing_newline = trailing_newline
def at(self, *path: str) -> JsonFileProxy: def at(self, *path: str) -> JsonFileProxy:
return JsonFileProxy(self, path) return JsonFileProxy(self, path)
@ -76,7 +83,12 @@ class JsonFile(File):
self.set(path, TAG) self.set(path, TAG)
def to_text(self) -> TextFile: def to_text(self) -> TextFile:
return TextFile(json.dumps(self.data, sort_keys=True, indent=2) + "\n") file = TextFile()
file.append(
json.dumps(self.data, sort_keys=True, indent=self.indent),
newline=self.trailing_newline,
)
return file
def to_bytes(self) -> bytes: def to_bytes(self) -> bytes:
return self.to_text().to_bytes() return self.to_text().to_bytes()