omdev 0.0.0.dev191__py3-none-any.whl → 0.0.0.dev193__py3-none-any.whl

omdev/toml/writer.py

@@ -1,114 +0,0 @@
-import dataclasses as dc
-import string
-import typing as ta
-
-
-class TomlWriter:
-    @dc.dataclass(frozen=True)
-    class Literal:
-        s: str
-
-    def __init__(self, out: ta.TextIO) -> None:
-        super().__init__()
-        self._out = out
-
-        self._indent = 0
-        self._wrote_indent = False
-
-    #
-
-    def _w(self, s: str) -> None:
-        if not self._wrote_indent:
-            self._out.write('    ' * self._indent)
-            self._wrote_indent = True
-        self._out.write(s)
-
-    def _nl(self) -> None:
-        self._out.write('\n')
-        self._wrote_indent = False
-
-    def _needs_quote(self, s: str) -> bool:
-        return (
-            not s or
-            any(c in s for c in '\'"\n') or
-            s[0] not in string.ascii_letters
-        )
-
-    def _maybe_quote(self, s: str) -> str:
-        if self._needs_quote(s):
-            return repr(s)
-        else:
-            return s
-
-    #
-
-    def write_root(self, obj: ta.Mapping) -> None:
-        for i, (k, v) in enumerate(obj.items()):
-            if i:
-                self._nl()
-            self._w('[')
-            self._w(self._maybe_quote(k))
-            self._w(']')
-            self._nl()
-            self.write_table_contents(v)
-
-    def write_table_contents(self, obj: ta.Mapping) -> None:
-        for k, v in obj.items():
-            self.write_key(k)
-            self._w(' = ')
-            self.write_value(v)
-            self._nl()
-
-    def write_array(self, obj: ta.Sequence) -> None:
-        self._w('[')
-        self._nl()
-        self._indent += 1
-        for e in obj:
-            self.write_value(e)
-            self._w(',')
-            self._nl()
-        self._indent -= 1
-        self._w(']')
-
-    def write_inline_table(self, obj: ta.Mapping) -> None:
-        self._w('{')
-        for i, (k, v) in enumerate(obj.items()):
-            if i:
-                self._w(', ')
-            self.write_key(k)
-            self._w(' = ')
-            self.write_value(v)
-        self._w('}')
-
-    def write_inline_array(self, obj: ta.Sequence) -> None:
-        self._w('[')
-        for i, e in enumerate(obj):
-            if i:
-                self._w(', ')
-            self.write_value(e)
-        self._w(']')
-
-    def write_key(self, obj: ta.Any) -> None:
-        if isinstance(obj, TomlWriter.Literal):
-            self._w(obj.s)
-        elif isinstance(obj, str):
-            self._w(self._maybe_quote(obj.replace('_', '-')))
-        elif isinstance(obj, int):
-            self._w(repr(str(obj)))
-        else:
-            raise TypeError(obj)
-
-    def write_value(self, obj: ta.Any) -> None:
-        if isinstance(obj, bool):
-            self._w(str(obj).lower())
-        elif isinstance(obj, (str, int, float)):
-            self._w(repr(obj))
-        elif isinstance(obj, ta.Mapping):
-            self.write_inline_table(obj)
-        elif isinstance(obj, ta.Sequence):
-            if not obj:
-                self.write_inline_array(obj)
-            else:
-                self.write_array(obj)
-        else:
-            raise TypeError(obj)