mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 01:00:29 +08:00
69 lines
2.8 KiB
Python
69 lines
2.8 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main PiKVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018-2024 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 os
|
|
import argparse
|
|
|
|
import mako.template
|
|
|
|
from ... import network
|
|
|
|
from .. import init
|
|
|
|
|
|
# =====
|
|
def main(argv: (list[str] | None)=None) -> None:
|
|
(parent_parser, argv, config) = init(
|
|
add_help=False,
|
|
argv=argv,
|
|
)
|
|
parser = argparse.ArgumentParser(
|
|
prog="kvmd-nginx-mkconf",
|
|
description="Generate KVMD-Nginx config",
|
|
parents=[parent_parser],
|
|
)
|
|
parser.add_argument("-p", "--print", action="store_true", help="Print the result to stdout besides the output file")
|
|
parser.add_argument("input", help="Input Mako template")
|
|
parser.add_argument("output", help="Output Nginx config")
|
|
options = parser.parse_args(argv[1:])
|
|
|
|
with open(options.input, "r") as in_file:
|
|
template = in_file.read()
|
|
|
|
rendered = mako.template.Template(template).render(
|
|
http_port=config.nginx.http.port,
|
|
https_enabled=config.nginx.https.enabled,
|
|
https_port=config.nginx.https.port,
|
|
ipv6_enabled=network.is_ipv6_enabled(),
|
|
)
|
|
|
|
if options.print:
|
|
print(rendered)
|
|
|
|
try:
|
|
os.remove(options.output)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
with open(options.output, "w") as out_file:
|
|
out_file.write(rendered)
|