Методы автоматической генерации скриптов Python

Расширенный пример 1: Генерация FastAPI эндпоинта с помощью Jinja2

Шаблон для создания обработчика GET-запроса с одним параметром. Jinja2 позволяет подставить имя функции, путь и возвращаемое значение.

from jinja2 import Template

endpoint_template = Template('''
@app.get("{{ path }}")
async def {{ func_name }}({{ param }}: {{ param_type }}):
    return {"{{ param }}": {{ param }}}
''')

code = endpoint_template.render(
    path="/items/{item_id}",
    func_name="get_item",
    param="item_id",
    param_type="int"
)
print(code)

@app.get("/items/{item_id}")
async def get_item(item_id: int):
    return {"item_id": item_id}

Расширенный пример 2: Генерация pytest теста через f-строки

Создание тестовой функции для проверки сложения. f-строки удобны для небольших динамических тестов.

import textwrap

func_name = "test_add"
a, b, expected = 2, 3, 5

code = textwrap.dedent(f'''
def {func_name}():
    result = add({a}, {b})
    assert result == {expected}, "Expected {expected}, got " + str(result)
''')
print(code)

def test_add():
    result = add(2, 3)
    assert result == 5, "Expected 5, got " + str(result)

Расширенный пример 3: Генерация SQLAlchemy модели с помощью string.Template

Шаблон для модели с полями и типами. string.Template безопасен при вводе из внешних источников.

from string import Template

model_template = Template('''
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class ${class_name}(Base):
    __tablename__ = '${table_name}'
    id = Column(Integer, primary_key=True)
    ${field_name} = Column(String(50))
''')

code = model_template.substitute(
    class_name="User",
    table_name="users",
    field_name="username"
)
print(code)

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    username = Column(String(50))

Расширенный пример 4: Генерация argparse скрипта через AST

Создание дерева AST для парсера аргументов и компиляция в исполняемый код. Пример демонстрирует добавление одного аргумента.

import ast
import argparse

# Создаём узлы AST для парсера
import_node = ast.Import(names=[ast.alias(name='argparse', asname=None)])

parser_call = ast.Call(
    func=ast.Attribute(value=ast.Name(id='argparse', ctx=ast.Load()), attr='ArgumentParser', ctx=ast.Load()),
    args=[],
    keywords=[ast.keyword(arg='description', value=ast.Constant(value='Sample script'))]
)

add_arg = ast.Call(
    func=ast.Attribute(value=ast.Name(id='parser', ctx=ast.Load()), attr='add_argument', ctx=ast.Load()),
    args=[],
    keywords=[
        ast.keyword(arg='name', value=ast.Constant(value='--name')),
        ast.keyword(arg='help', value=ast.Constant(value='Your name'))
    ]
)

assign_parser = ast.Assign(targets=[ast.Name(id='parser', ctx=ast.Store())], value=parser_call)
assign_args = ast.Call(
    func=ast.Attribute(value=ast.Name(id='parser', ctx=ast.Load()), attr='parse_args', ctx=ast.Load()),
    args=[],
    keywords=[]
)
assign_result = ast.Assign(targets=[ast.Name(id='args', ctx=ast.Store())], value=assign_args)

print_hello = ast.Call(
    func=ast.Name(id='print', ctx=ast.Load()),
    args=[ast.BinOp(left=ast.Constant(value='Hello, '), op=ast.Add(), right=ast.Attribute(value=ast.Name(id='args', ctx=ast.Load()), attr='name', ctx=ast.Load()))],
    keywords=[]
)

mod = ast.Module(body=[import_node, assign_parser, add_arg, assign_result, print_hello], type_ignores=[])

code = compile(mod, filename='', mode='exec')
exec(code)
# Запуск с параметром: python script.py --name World

При запуске с аргументом --name World выведет: Hello, World

Расширенный пример 5: Генерация dataclass с декоратором

Использование комбинации f-строк и ручной сборки для создания класса данных с полями.

from dataclasses import dataclass

class_name = "Point"
fields = [("x", "float"), ("y", "float")]

field_str = "\n    ".join([f"{name}: {typ}" for name, typ in fields])
code = f'''
from dataclasses import dataclass

@dataclass
class {class_name}:
    {field_str}
'''
print(code)

from dataclasses import dataclass

@dataclass
class Point:
    x: float
    y: float