Skip to content

Commit

Permalink
T6703: fix related and unrelated lint issues
Browse files Browse the repository at this point in the history
  • Loading branch information
nvollmar committed Sep 11, 2024
1 parent 333672b commit b2f3aea
Showing 1 changed file with 19 additions and 12 deletions.
31 changes: 19 additions & 12 deletions src/conf_mode/system_option.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,27 +37,26 @@
from vyos.configdep import call_dependents
from vyos import ConfigError
from vyos import airbag

airbag.enable()

curlrc_config = r'/etc/curlrc'
ssh_config = r'/etc/ssh/ssh_config.d/91-vyos-ssh-client-options.conf'
systemd_action_file = '/lib/systemd/system/ctrl-alt-del.target'
usb_autosuspend = r'/etc/udev/rules.d/40-usb-autosuspend.rules'
kernel_dynamic_debug = r'/sys/kernel/debug/dynamic_debug/control'
time_format_to_locale = {
'12-hour': 'en_US.UTF-8',
'24-hour': 'en_GB.UTF-8'
}
time_format_to_locale = {'12-hour': 'en_US.UTF-8', '24-hour': 'en_GB.UTF-8'}


def get_config(config=None):
if config:
conf = config
else:
conf = Config()
base = ['system', 'option']
options = conf.get_config_dict(base, key_mangling=('-', '_'),
get_first_key=True,
with_recursive_defaults=True)
options = conf.get_config_dict(
base, key_mangling=('-', '_'), get_first_key=True, with_recursive_defaults=True
)

if 'performance' in options:
# Update IPv4/IPv6 and sysctl options after tuned applied it's settings
Expand All @@ -66,14 +65,17 @@ def get_config(config=None):

return options


def verify(options):
if 'http_client' in options:
config = options['http_client']
if 'source_interface' in config:
verify_interface_exists(options, config['source_interface'])

if {'source_address', 'source_interface'} <= set(config):
raise ConfigError('Can not define both HTTP source-interface and source-address')
raise ConfigError(
'Can not define both HTTP source-interface and source-address'
)

if 'source_address' in config:
if not is_addr_assigned(config['source_address']):
Expand All @@ -94,7 +96,9 @@ def verify(options):
address = config['source_address']
interface = config['source_interface']
if not is_intf_addr_assigned(interface, address):
raise ConfigError(f'Address "{address}" not assigned on interface "{interface}"!')
raise ConfigError(
f'Address "{address}" not assigned on interface "{interface}"!'
)

if 'kernel' in options:
cpu_vendor = get_cpus()[0]['vendor_id']
Expand All @@ -105,6 +109,7 @@ def verify(options):

return None


def generate(options):
render(curlrc_config, 'system/curlrc.j2', options)
render(ssh_config, 'system/ssh_config.j2', options)
Expand All @@ -125,6 +130,7 @@ def generate(options):

return None


def apply(options):
# System bootup beep
beep_service = 'vyos-beep.service'
Expand Down Expand Up @@ -163,7 +169,7 @@ def apply(options):
if 'performance' in options:
cmd('systemctl restart tuned.service')
# wait until daemon has started before sending configuration
while (not is_systemd_service_running('tuned.service')):
while not is_systemd_service_running('tuned.service'):
sleep(0.250)
cmd('tuned-adm profile network-{performance}'.format(**options))
else:
Expand All @@ -178,9 +184,9 @@ def apply(options):

# Enable/diable root-partition-auto-resize SystemD service
if 'root_partition_auto_resize' in options:
cmd('systemctl enable root-partition-auto-resize.service')
cmd('systemctl enable root-partition-auto-resize.service')
else:
cmd('systemctl disable root-partition-auto-resize.service')
cmd('systemctl disable root-partition-auto-resize.service')

# Time format 12|24-hour
if 'time_format' in options:
Expand All @@ -200,6 +206,7 @@ def apply(options):
else:
write_file(kernel_dynamic_debug, f'module {module} -p')


if __name__ == '__main__':
try:
c = get_config()
Expand Down

0 comments on commit b2f3aea

Please sign in to comment.