Как редактировать конфигурационный файл Linux в Python?

Question

[gadzhikuliev]

Добрый день. Пишу скрипт для копирования конфигурационных файлов сетевого оборудования на TFTP сервер. Функцию обхода оборудования и копирования конфигов с него уже сделал, она работает. Осталось "добить" функцию редактирования конфига TFTP-сервера (в моём случае - tftpd-hpa). В конфиге необходимо корневой папке (параметр TFTP_DIRECTORY) менять название по умолчанию на определённое - например, текущая дата, когда было произведено копирование. В примере ниже новый каталог - переменная new. Так выглядит файл настроек tftpd-hpa до вмешательства скрипта:

TFTP_USERNAME="tftp"
TFTP_DIRECTORY="/tftpboot"
TFTP_ADDRESS="0.0.0.0:69"
TFTP_OPTIONS="--secure --create"

А вот сама функция. Подстановку нужного параметра сделал, но не придумал, как в файл записать, не создавая дополнительный.

new = '"/home/user/tfrp"'
with open('config.txt') as cfg:
    for line in cfg:   
        if line.startswith('TFTP_DIRECTORY'):
            break
sett = line.split('=')[1].strip()
print(line)
print(sett)
line = line.replace(sett,new)
print(line)

Подскажите, пожалуйста, если кто делал подобное. Заранее спасибо.

Answer 1

[tsarevfs]

Посмотрите на https://docs.python.org/3/library/configparser.html
Что касается изменения файла, то проще всего сначала считать в память, потом переоткрыть файл для записи и записать измененные данные из памяти.
https://stackoverflow.com/a/125713/1762922

Comments

[gadzhikuliev]

ConfigParser поможет, если формат - INI. В случае перекрытия на запись стираются все строки в файле.

[tsarevfs]

Ironhide, Все верно.
Вам нужно будет считать все строки, изменить некоторые из них и записать обратно все.

[tsarevfs]

Ironhide, А по поводу формата, ConfigParser похоже действительно не очень подходит. Возможно руками будет даже проще разобрать.

[gadzhikuliev]

tsarevfs, да, видимо, придётся с дополнительным файлом работать. Очевидный вариант, просто хотел покрасивее, без костылей, сделать.

[tsarevfs]

Ironhide, А зачем файл то? Запоминайте в память и потом опять в тот же файл.

aa="aaa"
bb="bbb"
cc="ccc"

lines = []
with open('config.txt', 'r') as cfg:
	lines = cfg.readlines()
key = 'bb'
newValue = 'qqq'
idx = next((i for i, l in enumerate(lines) if l.startswith(key)), None)

if idx is not None:
	lines[idx] = '{}="{}"'.format(key, newValue)

with open('config.txt', 'w') as cfg:
	cfg.write('\n'.join(lines))

[gadzhikuliev]

tsarevfs, спасибо! Попробую так.