1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/usr/bin/env python3
#
# Copyright 2022 VyOS maintainers and contributors <maintainers@vyos.io>
#
# This file is part of VyOS.
#
# VyOS 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.
#
# VyOS 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
# VyOS. If not, see <https://www.gnu.org/licenses/>.
import sys
from typing import List, Union
from hurry.filesize import size
from tabulate import tabulate
from vyos import opmode
from vyos.system import disk, grub, image
def _format_show_images_summary(images_summary: image.BootDetails) -> str:
headers: list[str] = ['Name', 'Default boot', 'Running']
table_data: list[list[str]] = list()
for image_item in images_summary.get('images_available', []):
name: str = image_item
if images_summary.get('image_default') == name:
default: str = 'Yes'
else:
default: str = ''
if images_summary.get('image_running') == name:
running: str = 'Yes'
else:
running: str = ''
table_data.append([name, default, running])
tabulated: str = tabulate(table_data, headers)
return tabulated
def _format_show_images_details(
images_details: list[image.ImageDetails]) -> str:
headers: list[str] = [
'Name', 'Version', 'Storage Read-Only', 'Storage Read-Write',
'Storage Total'
]
table_data: list[list[Union[str, int]]] = list()
for image_item in images_details:
name: str = image_item.get('name')
version: str = image_item.get('version')
disk_ro: int = size(image_item.get('disk_ro'))
disk_rw: int = size(image_item.get('disk_rw'))
disk_total: int = size(image_item.get('disk_total'))
table_data.append([name, version, disk_ro, disk_rw, disk_total])
tabulated: str = tabulate(table_data, headers)
return tabulated
def show_images_summary(raw: bool) -> Union[image.BootDetails, str]:
images_available: list[str] = grub.version_list()
root_dir: str = disk.find_persistence()
boot_vars: dict = grub.vars_read(f'{root_dir}/{image.CFG_VYOS_VARS}')
images_summary: image.BootDetails = dict()
images_summary['image_default'] = image.get_default_image()
images_summary['image_running'] = image.get_running_image()
images_summary['images_available'] = images_available
images_summary['console_type'] = boot_vars.get('console_type')
images_summary['console_num'] = boot_vars.get('console_num')
if raw:
return images_summary
else:
return _format_show_images_summary(images_summary)
def show_images_details(raw: bool) -> Union[list[image.ImageDetails], str]:
images: list[str] = grub.version_list()
images_details: list[image.ImageDetails] = list()
for image_name in images:
images_details.append(image.get_details(image_name))
if raw:
return images_details
else:
return _format_show_images_details(images_details)
if __name__ == '__main__':
try:
res = opmode.run(sys.modules[__name__])
if res:
print(res)
except (ValueError, opmode.Error) as e:
print(e)
sys.exit(1)
|
