289 строки
12 KiB
Python
289 строки
12 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}')
|
|
|
|
template_path = app_path.parent / 'templates' / 'network.html'
|
|
if not template_path.exists():
|
|
raise SystemExit(f'network.html not found: {template_path}')
|
|
|
|
dashboard_template_path = app_path.parent / 'templates' / 'dashboard.html'
|
|
if not dashboard_template_path.exists():
|
|
raise SystemExit(f'dashboard.html not found: {dashboard_template_path}')
|
|
|
|
text = app_path.read_text()
|
|
changed = []
|
|
|
|
helpers = r'''
|
|
|
|
def parse_dhcp_leases_output(output: str) -> list[dict[str, str]]:
|
|
leases: list[dict[str, str]] = []
|
|
for line in (output or '').splitlines():
|
|
raw = line.strip()
|
|
if not raw or raw.startswith('Expiry') or raw.startswith('-'):
|
|
continue
|
|
parts = raw.split()
|
|
if len(parts) < 5:
|
|
continue
|
|
leases.append({
|
|
'expiry': ' '.join(parts[0:2]),
|
|
'mac': parts[2].lower(),
|
|
'protocol': parts[3],
|
|
'ip': parts[4].split('/')[0],
|
|
'ip_with_prefix': parts[4],
|
|
'hostname': parts[5] if len(parts) > 5 else '—',
|
|
'client_id': parts[6] if len(parts) > 6 else '—',
|
|
})
|
|
return leases
|
|
|
|
|
|
def dhcp_leases_hint(info: dict[str, Any], leases: list[dict[str, str]]) -> str:
|
|
if leases:
|
|
return 'DHCP leases найдены. IP можно использовать для проброса портов.'
|
|
if not info.get('exists'):
|
|
return 'NAT-сеть ещё не создана. Нажми «Создать / починить NAT-сеть».'
|
|
info_text = str(info.get('info', ''))
|
|
if 'Active: yes' not in info_text:
|
|
return 'NAT-сеть есть, но не активна. Нажми «Создать / починить NAT-сеть» или проверь virsh net-start virtuality-nat.'
|
|
return 'Leases пусты: запусти VM с сетью virtuality-nat, дождись загрузки ОС и DHCP. У статических IP и некоторых готовых образов lease может не появиться.'
|
|
|
|
|
|
def vm_mac_addresses(vm_name: str) -> list[str]:
|
|
result = run_cmd(['virsh', 'domiflist', vm_name], timeout=8)
|
|
if not result['ok']:
|
|
return []
|
|
macs: list[str] = []
|
|
for line in result['stdout'].splitlines():
|
|
match = re.search(r'([0-9a-fA-F]{2}:){5}[0-9a-fA-F]{2}', line)
|
|
if match:
|
|
macs.append(match.group(0).lower())
|
|
return macs
|
|
|
|
|
|
def resolve_vm_ip_for_table(vm_name: str) -> str:
|
|
if not vm_name:
|
|
return '—'
|
|
result = run_cmd(['virsh', 'domifaddr', vm_name], timeout=8)
|
|
if result['ok']:
|
|
match = re.search(r'\b(\d{1,3}(?:\.\d{1,3}){3})/\d+', result['stdout'])
|
|
if match:
|
|
return match.group(1)
|
|
|
|
leases = run_cmd(['virsh', 'net-dhcp-leases', NETWORK_NAME], timeout=8)
|
|
if not leases['ok']:
|
|
return '—'
|
|
macs = set(vm_mac_addresses(vm_name))
|
|
for lease in parse_dhcp_leases_output(leases['stdout']):
|
|
if macs and lease.get('mac', '').lower() not in macs:
|
|
continue
|
|
ip = lease.get('ip') or ''
|
|
if ip:
|
|
return ip
|
|
return '—'
|
|
|
|
|
|
def enrich_vms_with_ips(vms: list[dict[str, str]]) -> list[dict[str, str]]:
|
|
enriched: list[dict[str, str]] = []
|
|
for vm in vms:
|
|
item = dict(vm)
|
|
item['ip'] = resolve_vm_ip_for_table(item.get('name', '')) if item.get('name') else '—'
|
|
enriched.append(item)
|
|
return enriched
|
|
'''
|
|
|
|
if 'def parse_dhcp_leases_output(' not in text:
|
|
markers = [
|
|
'\n\ndef libvirt_network_info() -> dict[str, Any]:',
|
|
'\n\ndef network_context() -> dict[str, Any]:',
|
|
'\n\ndef parse_virsh_list() -> list[dict[str, str]]:',
|
|
]
|
|
inserted = False
|
|
for marker in markers:
|
|
if marker in text:
|
|
text = text.replace(marker, helpers + marker, 1)
|
|
inserted = True
|
|
changed.append('DHCP leases and VM IP helpers added')
|
|
break
|
|
if not inserted:
|
|
print('WARN: DHCP helper marker not found, skip helper injection')
|
|
changed.append('DHCP leases parser skipped')
|
|
else:
|
|
changed.append('DHCP leases parser already present')
|
|
if 'def enrich_vms_with_ips(' not in text:
|
|
marker = '\n\ndef parse_virsh_list() -> list[dict[str, str]]:'
|
|
extra_helpers = r'''
|
|
|
|
def vm_mac_addresses(vm_name: str) -> list[str]:
|
|
result = run_cmd(['virsh', 'domiflist', vm_name], timeout=8)
|
|
if not result['ok']:
|
|
return []
|
|
macs: list[str] = []
|
|
for line in result['stdout'].splitlines():
|
|
match = re.search(r'([0-9a-fA-F]{2}:){5}[0-9a-fA-F]{2}', line)
|
|
if match:
|
|
macs.append(match.group(0).lower())
|
|
return macs
|
|
|
|
|
|
def resolve_vm_ip_for_table(vm_name: str) -> str:
|
|
if not vm_name:
|
|
return '—'
|
|
result = run_cmd(['virsh', 'domifaddr', vm_name], timeout=8)
|
|
if result['ok']:
|
|
match = re.search(r'\b(\d{1,3}(?:\.\d{1,3}){3})/\d+', result['stdout'])
|
|
if match:
|
|
return match.group(1)
|
|
leases = run_cmd(['virsh', 'net-dhcp-leases', NETWORK_NAME], timeout=8)
|
|
if not leases['ok']:
|
|
return '—'
|
|
macs = set(vm_mac_addresses(vm_name))
|
|
for lease in parse_dhcp_leases_output(leases['stdout']):
|
|
if macs and lease.get('mac', '').lower() not in macs:
|
|
continue
|
|
ip = lease.get('ip') or ''
|
|
if ip:
|
|
return ip
|
|
return '—'
|
|
|
|
|
|
def enrich_vms_with_ips(vms: list[dict[str, str]]) -> list[dict[str, str]]:
|
|
enriched: list[dict[str, str]] = []
|
|
for vm in vms:
|
|
item = dict(vm)
|
|
item['ip'] = resolve_vm_ip_for_table(item.get('name', '')) if item.get('name') else '—'
|
|
enriched.append(item)
|
|
return enriched
|
|
'''
|
|
if marker in text:
|
|
text = text.replace(marker, extra_helpers + marker, 1)
|
|
changed.append('VM IP helpers added')
|
|
else:
|
|
print('WARN: parse_virsh_list marker not found, skip VM IP helpers')
|
|
changed.append('VM IP helpers skipped')
|
|
|
|
old_info = '''def libvirt_network_info() -> dict[str, Any]:
|
|
info = run_cmd(['virsh', 'net-info', NETWORK_NAME], timeout=8)
|
|
leases = run_cmd(['virsh', 'net-dhcp-leases', NETWORK_NAME], timeout=8)
|
|
return {
|
|
'name': NETWORK_NAME,
|
|
'bridge': NAT_BRIDGE,
|
|
'subnet': NAT_SUBNET,
|
|
'gateway': NAT_GATEWAY,
|
|
'dhcp': f'{DHCP_START} - {DHCP_END}',
|
|
'exists': info['ok'],
|
|
'info': info['stdout'] if info['ok'] else info['stderr'],
|
|
'leases': leases['stdout'] if leases['ok'] else leases['stderr'],
|
|
}
|
|
'''
|
|
new_info = '''def libvirt_network_info() -> dict[str, Any]:
|
|
info = run_cmd(['virsh', 'net-info', NETWORK_NAME], timeout=8)
|
|
leases = run_cmd(['virsh', 'net-dhcp-leases', NETWORK_NAME], timeout=8)
|
|
leases_text = leases['stdout'] if leases['ok'] else leases['stderr']
|
|
lease_rows = parse_dhcp_leases_output(leases['stdout'] if leases['ok'] else '') if 'parse_dhcp_leases_output' in globals() else []
|
|
data = {
|
|
'name': NETWORK_NAME,
|
|
'bridge': NAT_BRIDGE,
|
|
'subnet': NAT_SUBNET,
|
|
'gateway': NAT_GATEWAY,
|
|
'dhcp': f'{DHCP_START} - {DHCP_END}',
|
|
'exists': info['ok'],
|
|
'info': info['stdout'] if info['ok'] else info['stderr'],
|
|
'leases': leases_text,
|
|
'lease_rows': lease_rows,
|
|
'lease_count': len(lease_rows),
|
|
'lease_hint': '',
|
|
}
|
|
data['lease_hint'] = dhcp_leases_hint(data, lease_rows) if 'dhcp_leases_hint' in globals() else 'DHCP leases пусты.'
|
|
return data
|
|
'''
|
|
if old_info in text:
|
|
text = text.replace(old_info, new_info, 1)
|
|
changed.append('libvirt network info enriched with lease rows and hint')
|
|
elif "'lease_rows':" in text:
|
|
changed.append('libvirt network info already enriched')
|
|
else:
|
|
print('WARN: libvirt_network_info marker not found, skip function rewrite')
|
|
changed.append('libvirt network info rewrite skipped')
|
|
|
|
# Add IP enrichment directly into parse_virsh_list return path.
|
|
if "return enrich_vms_with_ips(rows)" not in text:
|
|
old_return = " return rows\n\n\ndef parse_pool_list() -> list[dict[str, str]]:"
|
|
new_return = " return enrich_vms_with_ips(rows) if 'enrich_vms_with_ips' in globals() else rows\n\n\ndef parse_pool_list() -> list[dict[str, str]]:"
|
|
if old_return in text:
|
|
text = text.replace(old_return, new_return, 1)
|
|
changed.append('VM list enriched with IP addresses')
|
|
else:
|
|
print('WARN: parse_virsh_list return marker not found, skip VM IP enrichment')
|
|
changed.append('VM IP enrichment skipped')
|
|
else:
|
|
changed.append('VM IP enrichment already present')
|
|
|
|
app_path.write_text(text)
|
|
|
|
html = template_path.read_text()
|
|
old_html = ''' <article class="card">
|
|
<div class="card-head">
|
|
<h2>DHCP leases</h2>
|
|
<span class="pill">virsh net-dhcp-leases</span>
|
|
</div>
|
|
<pre>{{ ctx.nat.leases }}</pre>
|
|
<div class="notice top-space">IP для проброса определяется автоматически: сначала через qemu-guest-agent, затем по DHCP leases и MAC-адресу VM.</div>
|
|
</article>
|
|
'''
|
|
new_html = ''' <article class="card">
|
|
<div class="card-head">
|
|
<h2>DHCP leases</h2>
|
|
<span class="pill">{{ ctx.nat.lease_count|default(0) }} active</span>
|
|
</div>
|
|
{% if ctx.nat.lease_rows %}
|
|
<table>
|
|
<thead><tr><th>IP</th><th>MAC</th><th>Host</th><th>Expiry</th></tr></thead>
|
|
<tbody>
|
|
{% for lease in ctx.nat.lease_rows %}
|
|
<tr>
|
|
<td class="strong">{{ lease.ip }}</td>
|
|
<td>{{ lease.mac }}</td>
|
|
<td>{{ lease.hostname }}</td>
|
|
<td class="muted">{{ lease.expiry }}</td>
|
|
</tr>
|
|
{% endfor %}
|
|
</tbody>
|
|
</table>
|
|
{% else %}
|
|
<div class="notice">{{ ctx.nat.lease_hint|default('DHCP leases пусты.') }}</div>
|
|
<pre>virsh net-dhcp-leases {{ ctx.nat.name }}
|
|
virsh net-info {{ ctx.nat.name }}
|
|
virsh domiflist ИМЯ_VM
|
|
virsh domifaddr ИМЯ_VM</pre>
|
|
{% endif %}
|
|
<div class="notice top-space">IP для проброса определяется автоматически: сначала через qemu-guest-agent, затем по DHCP leases и MAC-адресу VM. Если внутри образа статический IP, leases может быть пустым — тогда лучше добавить ручной IP-режим следующим шагом.</div>
|
|
</article>
|
|
'''
|
|
if old_html in html:
|
|
html = html.replace(old_html, new_html, 1)
|
|
changed.append('DHCP leases UI replaced with structured table and empty hint')
|
|
elif 'ctx.nat.lease_rows' in html:
|
|
changed.append('DHCP leases UI already updated')
|
|
else:
|
|
print('WARN: DHCP leases template marker not found, skip UI rewrite')
|
|
changed.append('DHCP leases UI rewrite skipped')
|
|
|
|
template_path.write_text(html)
|
|
|
|
dashboard_html = dashboard_template_path.read_text()
|
|
if '<th>IP</th>' not in dashboard_html:
|
|
dashboard_html = dashboard_html.replace('<tr><th>ID</th><th>Name</th><th>State</th><th>Actions</th></tr>', '<tr><th>ID</th><th>Name</th><th>IP</th><th>State</th><th>Actions</th></tr>')
|
|
dashboard_html = dashboard_html.replace('<td class="strong"><a class="table-link" href="/vm/{{ vm.name }}">{{ vm.name }}</a></td>\n <td><span', '<td class="strong"><a class="table-link" href="/vm/{{ vm.name }}">{{ vm.name }}</a></td>\n <td class="strong">{{ vm.ip|default(\'—\') }}</td>\n <td><span')
|
|
dashboard_html = dashboard_html.replace('colspan="4" class="muted">Виртуальных машин пока нет', 'colspan="5" class="muted">Виртуальных машин пока нет')
|
|
changed.append('Dashboard VM IP column added')
|
|
else:
|
|
changed.append('Dashboard VM IP column already present')
|
|
dashboard_template_path.write_text(dashboard_html)
|
|
|
|
print('DHCP leases and VM IP patch applied:')
|
|
for item in changed:
|
|
print(f'- {item}')
|
© 2026 StackWorks