Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,908 Bytes
df6c67d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import json
import os.path
from typing import List, Optional, Union
def read_text_file(
path: str,
split_lines: bool = False,
strip_white_chars: bool = False,
) -> Union[str, List[str]]:
with open(path) as f:
if split_lines:
lines = list(f.readlines())
if strip_white_chars:
return [line.strip() for line in lines if len(line.strip()) > 0]
else:
return lines
content = f.read()
if strip_white_chars:
content = content.strip()
return content
def read_json(path: str, **kwargs) -> Optional[Union[dict, list]]:
with open(path) as f:
return json.load(f, **kwargs)
def dump_json(
path: str, content: Union[dict, list], allow_override: bool = False, **kwargs
) -> None:
ensure_write_is_allowed(path=path, allow_override=allow_override)
ensure_parent_dir_exists(path=path)
with open(path, "w") as f:
json.dump(content, fp=f, **kwargs)
def dump_text_lines(
path: str, content: List[str], allow_override: bool = False
) -> None:
ensure_write_is_allowed(path=path, allow_override=allow_override)
ensure_parent_dir_exists(path=path)
with open(path, "w") as f:
f.write("\n".join(content))
def dump_bytes(path: str, content: bytes, allow_override: bool = False) -> None:
ensure_write_is_allowed(path=path, allow_override=allow_override)
ensure_parent_dir_exists(path=path)
with open(path, "wb") as f:
f.write(content)
def ensure_parent_dir_exists(path: str) -> None:
absolute_path = os.path.abspath(path)
parent_dir = os.path.dirname(absolute_path)
os.makedirs(parent_dir, exist_ok=True)
def ensure_write_is_allowed(path: str, allow_override: bool) -> None:
if os.path.exists(path) and not allow_override:
raise RuntimeError(f"File {path} exists and override is forbidden.")
|