5cbe3e2254
* style: apply some fixes from Sorcery * Apply suggestions from code review Co-authored-by: Yannick Jadoul <yannick.jadoul@belgacom.net> Co-authored-by: Joe Rickerby <joerick@mac.com> Co-authored-by: Yannick Jadoul <yannick.jadoul@belgacom.net> Co-authored-by: Joe Rickerby <joerick@mac.com>
24 lines
751 B
Python
24 lines
751 B
Python
"""
|
|
These are utilities for the `/bin` scripts, not for the `cibuildwheel` program.
|
|
"""
|
|
|
|
from typing import Any, Dict
|
|
|
|
import toml.encoder
|
|
from packaging.version import Version
|
|
|
|
|
|
class InlineArrayDictEncoder(toml.encoder.TomlEncoder): # type: ignore
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.dump_funcs[Version] = lambda v: f'"{v}"'
|
|
|
|
def dump_sections(self, o: Dict[str, Any], sup: str) -> Any:
|
|
if not all(isinstance(a, list) for a in o.values()):
|
|
return super().dump_sections(o, sup)
|
|
val = ""
|
|
for k, v in o.items():
|
|
inner = ",\n ".join(self.dump_inline_table(d_i).strip() for d_i in v)
|
|
val += f"{k} = [\n {inner},\n]\n"
|
|
return val, self._dict()
|