Каким образом выполнить множественную замену строк в файле?

Question

[teodoros]

Здравствуйте. Подскажите, как можно сократить этот код?
Т.е. сократить кол-во replace.

def editapache(self,oldmail,mail,name,oldname,charset,oldcharset,vhid):
      with open(filename,'r+') as config:
        config.replace(oldcharset,charset)
        config.replace(oldname,name)
        config.replace(oldmail,mail)
    config.write(config)

Comments

[aRegius]

Здравствуйте, teodoros.

replace() - строковый метод. Разве config, в вашем примере, это строка?

[JaxxDexx]

aRegius: Кто его знает, может он переопределил контекстный менеджер для open)))

[aRegius]

JaxxDexx:
И даже если это строка, в чем я лично сомневаюсь, на нашем пути возникает пункт 2)))
А именно: строковый метод не меняет строку in-place, он возвращает новую, поэтому в данном примере записывается все равно исходная строка. Нужно присваивать значение переменной.

[JaxxDexx]

aRegius: А почему сразу строка) Может его контекстный менеджер возвращает какой-то неведомый объект, который тоже имеет метод replace, который изменяет строку внутри себя)))
Я конечно же фантазирую! Но кто его знает, где ТС нарыл этот код, раз у него не возникло вопросов по его работоспособности)

[teodoros]

JaxxDexx: Действительно, в моем предыдущем коде переменная config строкой не является

def editapache(self,oldmail,mail,name,oldname,charset,oldcharset,vhid):
       path = '/etc/apache2/sites-available/%s.conf' %(vhid)
#Открываем файл
        with open(path, 'r+') as config_file:
#Читаем его содержимое в строку
            config = config_file.read()       
#Do smth here

[JaxxDexx]

teodoros: Если вы под 3 питон пишите, используйте fileinput, он уже поддерживает редактирование текста по месту.
Вот пример: ^_^

[teodoros]

JaxxDexx: И еще подскажите - в одной библиотеке увидел такой подход:

zone_file.seek(0)
      for line in file_content:
        # When we find the old line, write the new one instead
        # This may seem elaborate, but its a way of ensuring we're editing the right
        #   record and that we don't miss it
        if re.search(old_name,line) and re.search(old_type,line) and re.search(old_value,line):
          zone_file.write("%s IN %s %s \n" % (new_name,new_type,new_value))
        else:
          zone_file.write(line)

Т.е. перебор строк файла в цикле, поиск нужной строки и её замена.
Подход с replace идентичен приведенному выше коду?

[JaxxDexx]

teodoros:
В случае цикла, вы не загружаете весь файл целиком в память, поэтому если у вас файл очень большой, то лучше использовать цикл.
Поиск нужной строки, с помощью replace вы можете заменить только при явном совпадении, если у вас регулярка, то так не получится!
В общем подходы разные и выбор зависит от требований

Answer 1

[JaxxDexx]

def editapache(self,oldmail,mail,name,oldname,charset,oldcharset,vhid):
      with open(filename,'r+') as config:
        map(config.replace, [oldmail, oldname, oldcharset], [mail, name, charset])
    config.write(config)

Comments

[Alexej Simakov]

а я всегда думал в map нужно ставить lambda :) оказывается любая фунцкия подходит

Answer 2

[Алексей Черемисин]

Сильно не улучшить: но

def editapache(self,replacements):
      with open(filename,'r+') as config:
        for _from,_to in replacements:
            config.replace(_from,_to)
    config.write(config)


....
editapache(self,[(oldcharset,charset),(oldmail,mail),(oldname,name),("oldstring","newstring")])

А вообще-то лучше использовать темплейты и генерировать когфигурацию из них. Посоветую например "нинзя2" - jinja.pocoo.org/docs/2.9
А еще лучше, возьмите ansible например, в нем и jinja2 в качестве шаблонизатора в комплекте :-)

Answer 3

[Alexej Simakov]

нашел тут на просторах интернета такое решение:

config = "1 2 3 4 5 6"
substitutions = {"2":"7", "3":"6"}
def replace_all(text, dic):
    for i, j in dic.iteritems():
        text = text.replace(i, j)
    return text

print replace_all(config, substitutions)

вернет
1 7 6 4 5 6