mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 01:00:29 +08:00
77 lines
3.3 KiB
Python
77 lines
3.3 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main Pi-KVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018 Maxim Devaev <mdevaev@gmail.com> #
|
|
# #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
|
# #
|
|
# ========================================================================== #
|
|
|
|
|
|
import textwrap
|
|
import operator
|
|
|
|
from typing import Generator
|
|
from typing import Any
|
|
|
|
import yaml
|
|
|
|
from . import Section
|
|
|
|
|
|
# =====
|
|
def make_config_dump(config: Section, indent: int=4) -> str:
|
|
return "\n".join(_inner_make_dump(config, indent))
|
|
|
|
|
|
def _inner_make_dump(config: Section, indent: int, _level: int=0) -> Generator[str, None, None]:
|
|
for (key, value) in sorted(config.items(), key=operator.itemgetter(0)):
|
|
if isinstance(value, Section):
|
|
prefix = " " * indent * _level
|
|
yield f"{prefix}{key}:"
|
|
yield from _inner_make_dump(value, indent, _level + 1)
|
|
yield ""
|
|
else:
|
|
default = config._get_default(key) # pylint: disable=protected-access
|
|
comment = config._get_help(key) # pylint: disable=protected-access
|
|
if default == value:
|
|
yield _make_yaml_kv(key, value, indent, _level, comment=comment)
|
|
else:
|
|
yield _make_yaml_kv(key, default, indent, _level, comment=comment, commented=True)
|
|
yield _make_yaml_kv(key, value, indent, _level)
|
|
|
|
|
|
def _make_yaml_kv(key: str, value: Any, indent: int, level: int, comment: str="", commented: bool=False) -> str:
|
|
text = yaml.dump(value, indent=indent, allow_unicode=True)
|
|
text = text.replace("\n...\n", "").strip()
|
|
if (
|
|
isinstance(value, dict) and text[0] != "{"
|
|
or isinstance(value, list) and text[0] != "["
|
|
):
|
|
text = "\n" + textwrap.indent(text, prefix=" " * indent)
|
|
else:
|
|
text = " " + text
|
|
|
|
prefix = " " * indent * level
|
|
if commented:
|
|
prefix = prefix + "# "
|
|
text = textwrap.indent(f"{key}:{text}", prefix=prefix)
|
|
|
|
if comment:
|
|
lines = text.split("\n")
|
|
lines[0] += " # " + comment
|
|
text = "\n".join(lines)
|
|
return text
|