64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""TOML format converter using tomli/tomllib."""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any, Dict, Optional
|
|
|
|
from config_convert.converters.base import Converter
|
|
|
|
if sys.version_info >= (3, 11):
|
|
import tomllib
|
|
else:
|
|
import tomli as tomllib
|
|
|
|
|
|
class TOMLConverter(Converter):
|
|
"""Converter for TOML format."""
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "toml"
|
|
|
|
def loads(self, data: str) -> Dict[str, Any]:
|
|
return tomllib.loads(data)
|
|
|
|
def dumps(self, data: Dict[str, Any], indent: Optional[int] = None) -> str:
|
|
output = []
|
|
self._write_dict(data, output, indent=0)
|
|
return "\n".join(output)
|
|
|
|
def _write_dict(self, data: Dict[str, Any], output: list, indent: int) -> None:
|
|
for key, value in data.items():
|
|
if isinstance(value, dict):
|
|
output.append(f"[{key}]")
|
|
self._write_dict(value, output, indent + 1)
|
|
elif isinstance(value, list):
|
|
if len(value) > 0 and isinstance(value[0], dict):
|
|
for item in value:
|
|
output.append(f"[[{key}]]")
|
|
self._write_dict(item, output, indent + 1)
|
|
else:
|
|
items_str = ", ".join(self._format_value(item) for item in value)
|
|
output.append(f"{key} = [{items_str}]")
|
|
else:
|
|
output.append(f"{key} = {self._format_value(value)}")
|
|
|
|
def _format_value(self, value: Any) -> str:
|
|
if isinstance(value, bool):
|
|
return "true" if value else "false"
|
|
elif isinstance(value, str):
|
|
return f'"{value}"'
|
|
elif value is None:
|
|
return ""
|
|
else:
|
|
return str(value)
|
|
|
|
def load(self, file_path: str) -> Dict[str, Any]:
|
|
with open(file_path, "rb") as f:
|
|
return tomllib.load(f)
|
|
|
|
def dump(self, data: Dict[str, Any], file_path: str, indent: Optional[int] = None) -> None:
|
|
Path(file_path).parent.mkdir(parents=True, exist_ok=True)
|
|
with open(file_path, "wb") as f:
|
|
f.write(self.dumps(data).encode("utf-8"))
|