Как сделать запись в файл после определенного метода?

Question

[Стас Поваляев]

Привет! я вот тут не могу решить проблему, подскажите как сделать так что бы пайтон записывал в файл с кодом, текст, но в определённом методе?
вот есть такой

шаблон

from kivy.app import App
from kivy.uix.boxlayout import BoxLayout
from kivy.uix.button import Button
from kivy.uix.label import Label
from kivy.uix.screenmanager import ScreenManager, Screen
from kivy.animation import Animation
from kivy.clock import Clock
from kivy.config import Config
from kivy.uix.floatlayout import FloatLayout
from kivy.uix.image import Image
from kivy.core import audio
from kivy.core.audio import SoundLoader
#import socket
#sock = socket.socket()
#sock.bind(('', 9090))
#sock.listen(1)
 
 
Config.set('graphics', 'resizable', '0')
Config.set('graphics', 'width', '1280')
Config.set('graphics', 'height', '720')
 
class MainApp(App):
    def build(self):
        main_layout = FloatLayout()
 
 
 
 
 
 
 
 
 
        return main_layout
 
 
 
if __name__ == '__main__':
    app = MainApp()
    app.run()

как сделать так что бы пайтон записал текст после строчек

def build(self):
        main_layout = FloatLayout()
        #тут

но не номеру строчки, а по названию функции, что бы пайтон искал метод "def build" и после еще 1 строки записал текст?

Answer 1

[st0ne_c0ld]

По исходному условию, можно написать regexp или "стейт машину" с хардкодом содержимого файла, но если содержимое будет меняться - код надо будет переписывать. Ниже вариант проще, но он отличается от исходного условия.

Если нет ограничения на формат файла и код надо вписывать всегда в однотипные места, можно использовать модуль шаблонов jinja2 либо плейсхолдеры для работы со строками, вида:

your_new_code="""
    print(123)
    return False
"""
...
updated_text = """
...
class MainApp(App):
    def build(self):
        main_layout = FloatLayout()
        {some_text}
...
""".format(some_text=your_new_code)

Comments

[Стас Поваляев]

хм, но если допустим выше шаблона появился еще один метод, и сместился текст сместился вниз, то тогда при записании кода один код налезет на другой? я просто делаю что-то на подобии скретча, но техника была такая: код записывался на конкретную строку, и когда я импортировал какой-то модуль вверху, то весь текст вмещался вниз, и весь код был поломан, потому что нужная строка уже занята текстом, и один текст налазит на другой, именно можно ли зделать так, что бы пайтон искал метод поименно? типа есть 3 метода, и пайтон искал именно 1 конкретный метод и записывал туда?

[Стас Поваляев]

вот у меня запись шла так:

with open(FILE_NAME, 'r') as NewPrj
            lines = NewPrj.readlines()
            lines.insert(26, '#my code\n')
            # переменная/вставить/номер строки/код

но можно ли вместо номера строки вписать что-то что бы в конкретный метод записывало?

[st0ne_c0ld]

cout_HelloWorld_andl, если хочется с файлом просто как с текстом работать - можно использовать вариант предложенный Vindicar - там не учитываются номера строк, а идет замена нужного токена.
Если хочется прям "по правильному" можно полистать исходники модуля parser и/или ast(стандартные библиотеки, версии 3.7+), который вроде как умеет делать синтаксический разбор, возможно там получится и куски кода вставить в нужные места, но сам я таким не занимался.
Ну и ещё вариант, можно посмотреть в сторону шаблонизаторов вроде jinja, которые придуманы, чтобы вставлять текст в нужные места в шаблоне - там, например есть функции-фильтры, которые сделают нужное количество отступов, чтобы уместить вставляемый код с правильным количеством пробелов.

[Стас Поваляев]

st0ne_c0ld, ваш код уже работает или так еще надо доделать?

Answer 2

[Vindicar]

Ну самый лобовой способ - изменить файл-шаблон так (сокращено):

class MainApp(App):
    def build(self):
        main_layout = FloatLayout()
        # INSERT CODE HERE
        return main_layout
 
if __name__ == '__main__':
    app = MainApp()
    app.run()

Тогда можно будет сделать просто

with open('file.py', 'rt', encoding='utf-8') as src:
    data = src.read()
with open('file2.py', 'wt', encoding='utf-8') as dest:
    dest.write(data.replace('# INSERT CODE HERE', 'тут твой код'))

Разумеется, строка # INSERT CODE HERE не должна встречаться в других местах файла-шаблона.
Также подразумевается, что файл-шаблон сам по себе, и не изменяется программно - вместо этого результат записывается в другой файл, который уже используется.

Comments

[Стас Поваляев]

а если выше добавится еще один метод?

[Vindicar]

cout_HelloWorld_andl, код просто заменит указанный фрагмент строки на другой. Независимо от того, сколько раз и где он встречается.

[Стас Поваляев]

Vindicar, спасибо, но я заметил что он заменяет но не записывает, тобеж если я хочу дважды код добавить, что бы было допустим 2 кнопки, это не работает

[Стас Поваляев]

Vindicar, еще он каждый раз создает новый файл, и полностью перезаписывает его содержимое, мне надо что бы он изменял конкретный файл шаблон и в нем дописывал код, извините что надоедаю( очень это надо решить

[Vindicar]

cout_HelloWorld_andl,
1. Просто сделай два разных маркера, делов то. # INSERT CODE 1 HERE и # INSERT CODE 2 HERE
2. "он каждый раз создает новый файл, и полностью перезаписывает его содержимое"
Да, я сразу предупредил что код так работает. И если ты хоть немножко подумаешь головой, то поймёшь почему.

Подсказка

Потому что корректно определить, где именно в файле находится неизвестный ранее вставленный код, будет в разы сложнее, чем просто сгенерировать новый код на базе старого шаблона. Не надо усложнять решение без нужды.

Так что лучше объясни, чем тебе так помешало присутствие отдельного файла-шаблона.

[Vindicar]

cout_HelloWorld_andl, вот ещё раз, ты совершенно не ответил на мой вопрос.
Чем тебе так мешает присутствие отдельного файла-шаблона для генерации кода, что ты настаиваешь на том, чтобы пытаться редактировать уде существующий (и уже изменённый) файл?

Всё, что ты написал выше, можно реализовывать по разному, в том числе и моим способом. Я просто не вижу в чём у тебя возникает проблема, потому что решение задачи, озвученной в вопросе - тривиальнейшее. Впрочем, ты даже не озаботился привести пример "что хочется - что получилось на практике", так что сам виноват. Телепаты все на карантине, учись формулировать свои вопросы так, чтобы тебя понимали.

Единственная потенциальная сложность - если ты заранее не знаешь, на каком уровне отступа должен находиться вставляемый код. Вот тогда да, придётся повозиться. Но опять же, ты словом не обмолвился, что с этим возникает проблема. Если ты вставляешь код в методы одного класса, то они все должны быть на одному уровне отступа.