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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
from __future__ import annotations
import logging
from pathlib import Path
from typing import Any, Dict, TYPE_CHECKING
from ..configuration import ConfigurationOutput
from ..general import SysCommand
from ..menu import Menu
from ..menu.menu import MenuSelectionType
from ..output import log
if TYPE_CHECKING:
_: Any
def save_config(config: Dict):
def preview(selection: str):
if options['user_config'] == selection:
json_config = config_output.user_config_to_json()
return f'{config_output.user_configuration_file}\n{json_config}'
elif options['user_creds'] == selection:
if json_config := config_output.user_credentials_to_json():
return f'{config_output.user_credentials_file}\n{json_config}'
else:
return str(_('No configuration'))
elif options['disk_layout'] == selection:
if json_config := config_output.disk_layout_to_json():
return f'{config_output.disk_layout_file}\n{json_config}'
else:
return str(_('No configuration'))
elif options['all'] == selection:
output = f'{config_output.user_configuration_file}\n'
if json_config := config_output.user_credentials_to_json():
output += f'{config_output.user_credentials_file}\n'
if json_config := config_output.disk_layout_to_json():
output += f'{config_output.disk_layout_file}\n'
return output[:-1]
return None
config_output = ConfigurationOutput(config)
options = {
'user_config': str(_('Save user configuration')),
'user_creds': str(_('Save user credentials')),
'disk_layout': str(_('Save disk layout')),
'all': str(_('Save all'))
}
choice = Menu(
_('Choose which configuration to save'),
list(options.values()),
sort=False,
skip=True,
preview_size=0.75,
preview_command=preview
).run()
if choice.type_ == MenuSelectionType.Skip:
return
dirs_to_exclude = [
'/bin',
'/dev',
'/lib',
'/lib64',
'/lost+found',
'/opt',
'/proc',
'/run',
'/sbin',
'/srv',
'/sys',
'/usr',
'/var',
]
log(
_('When picking a directory to save configuration files to,'
' by default we will ignore the following folders: ') + ','.join(dirs_to_exclude),
level=logging.DEBUG
)
log(_('Finding possible directories to save configuration files ...'), level=logging.INFO)
find_exclude = '-path ' + ' -prune -o -path '.join(dirs_to_exclude) + ' -prune '
file_picker_command = f'find / {find_exclude} -o -type d -print0'
possible_save_dirs = list(
filter(None, SysCommand(file_picker_command).decode().split('\x00'))
)
selection = Menu(
_('Select directory (or directories) for saving configuration files'),
possible_save_dirs,
multi=True,
skip=True,
allow_reset=False,
).run()
match selection.type_:
case MenuSelectionType.Skip:
return
case _:
save_dirs = selection.value
prompt = _('Do you want to save {} configuration file(s) in the following locations?\n\n{}').format(
list(options.keys())[list(options.values()).index(choice.value)],
save_dirs
)
save_confirmation = Menu(prompt, Menu.yes_no(), default_option=Menu.yes()).run()
if save_confirmation == Menu.no():
return
log(
_('Saving {} configuration files to {}').format(
list(options.keys())[list(options.values()).index(choice.value)],
save_dirs
),
level=logging.DEBUG
)
if save_dirs is not None:
for save_dir_str in save_dirs:
save_dir = Path(save_dir_str)
if options['user_config'] == choice.value:
config_output.save_user_config(save_dir)
elif options['user_creds'] == choice.value:
config_output.save_user_creds(save_dir)
elif options['disk_layout'] == choice.value:
config_output.save_disk_layout(save_dir)
elif options['all'] == choice.value:
config_output.save_user_config(save_dir)
config_output.save_user_creds(save_dir)
config_output.save_disk_layout(save_dir)
|