summaryrefslogtreecommitdiff
path: root/cloudinit/distros/bsd_utils.py
blob: 079d0d5356263331cfb6771bd7e4262a3050ddf4 (plain)
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
# This file is part of cloud-init. See LICENSE file for license information.

import shlex

from cloudinit import util

# On NetBSD, /etc/rc.conf comes with a if block:
#   if [ -r /etc/defaults/rc.conf ]; then
# as a consequence, the file is not a regular key/value list
# anymore and we cannot use cloudinit.distros.parsers.sys_conf
# The module comes with a more naive parser, but is able to
# preserve these if blocks.


def _unquote(value):
    if value[0] == value[-1] and value[0] in ['"', "'"]:
        return value[1:-1]
    return value


def get_rc_config_value(key, fn='/etc/rc.conf'):
    key_prefix = '{}='.format(key)
    for line in util.load_file(fn).splitlines():
        if line.startswith(key_prefix):
            value = line.replace(key_prefix, '')
            return _unquote(value)


def set_rc_config_value(key, value, fn='/etc/rc.conf'):
    lines = []
    done = False
    value = shlex.quote(value)
    original_content = util.load_file(fn)
    for line in original_content.splitlines():
        if '=' in line:
            k, v = line.split('=', 1)
            if k == key:
                v = value
                done = True
            lines.append('='.join([k, v]))
        else:
            lines.append(line)
    if not done:
        lines.append('='.join([key, value]))
    new_content = '\n'.join(lines) + '\n'
    if new_content != original_content:
        util.write_file(fn, new_content)


# vi: ts=4 expandtab