48 строки
2.4 KiB
Python
48 строки
2.4 KiB
Python
#!/usr/bin/env python3
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
app_path = Path(sys.argv[1]) if len(sys.argv) > 1 else Path('/opt/virtuality/web/app.py')
|
|
if not app_path.exists():
|
|
raise SystemExit(f'app.py not found: {app_path}')
|
|
|
|
text = app_path.read_text()
|
|
|
|
if 'import network_templates' not in text:
|
|
text = text.replace('import network_core\n', 'import network_core\nimport network_templates\n', 1)
|
|
|
|
text = text.replace('"ctx": network_core.network_context(), "error": error}', '"ctx": network_core.network_context(), "templates": network_templates.list_templates(), "error": error}', 1)
|
|
text = text.replace('"ctx": network_core.network_context(), "error": str(exc)}', '"ctx": network_core.network_context(), "templates": network_templates.list_templates(), "error": str(exc)}')
|
|
text = text.replace('"ctx": network_core.network_context(), "error": str(exc)}, status_code=400)', '"ctx": network_core.network_context(), "templates": network_templates.list_templates(), "error": str(exc)}, status_code=400)')
|
|
text = text.replace('"ctx": network_core.network_context(), "error": str(exc)}, status_code=500)', '"ctx": network_core.network_context(), "templates": network_templates.list_templates(), "error": str(exc)}, status_code=500)')
|
|
|
|
if 'def network_template_apply(' not in text:
|
|
handler = '''
|
|
|
|
@app.post("/network/template/apply")
|
|
def network_template_apply(request: Request, vm_name: str = Form(...), template_key: str = Form(...), external_base_port: int = Form(0)):
|
|
auth_redirect = require_auth(request)
|
|
if auth_redirect:
|
|
return auth_redirect
|
|
try:
|
|
network_templates.apply_template(vm_name, template_key, external_base_port if external_base_port > 0 else None)
|
|
except NetworkError as exc:
|
|
return templates.TemplateResponse("network.html", {
|
|
"request": request,
|
|
"app_name": APP_NAME,
|
|
"user": AUTH_USER,
|
|
"vms": parse_virsh_list(),
|
|
"ctx": network_core.network_context(),
|
|
"templates": network_templates.list_templates(),
|
|
"error": str(exc),
|
|
}, status_code=400)
|
|
return RedirectResponse(url="/network", status_code=303)
|
|
'''
|
|
marker = '\n\n@app.post("/network/forward/add")'
|
|
if marker not in text:
|
|
raise SystemExit('forward/add marker not found')
|
|
text = text.replace(marker, handler + marker, 1)
|
|
|
|
app_path.write_text(text)
|
|
print(f'network templates patch applied: {app_path}')
|
© 2026 StackWorks