Генерация и отладка Python-кода с помощью нейросетей

Расширенные примеры генерации и отладки Python-кода с помощью ИИ

Ниже приведены более сложные сценарии, демонстрирующие возможности нейросетевых помощников.

1. REST API клиент с обработкой ошибок

Запрос к ChatGPT: «Создай класс на Python для работы с REST API, который включает повторные попытки при таймауте и логирование».

import requests
import time
import logging

class APIClient:
    def __init__(self, base_url, retries=3, timeout=5):
        self.base_url = base_url
        self.retries = retries
        self.timeout = timeout
        logging.basicConfig(level=logging.INFO)
        self.logger = logging.getLogger(__name__)

    def get(self, endpoint):
        for attempt in range(self.retries):
            try:
                response = requests.get(f'{self.base_url}/{endpoint}', timeout=self.timeout)
                response.raise_for_status()
                return response.json()
            except requests.exceptions.RequestException as e:
                self.logger.warning(f'Attempt {attempt+1} failed: {e}')
                time.sleep(2 ** attempt)
        raise Exception('API request failed after max retries')

>>> client = APIClient('https://jsonplaceholder.typicode.com')
>>> client.get('todos/1')
{'userId': 1, 'id': 1, 'title': 'delectus aut autem', 'completed': False}

2. Парсинг сложной HTML-страницы с BeautifulSoup

Запрос: «Извлеки все цены и названия товаров со страницы интернет-магазина, используя селекторы CSS».

import requests
from bs4 import BeautifulSoup

url = 'https://books.toscrape.com/'
soup = BeautifulSoup(requests.get(url).text, 'html.parser')
products = soup.select('article.product_pod')
for product in products:
    title = product.h3.a.get('title')
    price = product.select_one('p.price_color').text
    print(f'{title}: {price}')

A Light in the Attic: £51.77
Tipping the Velvet: £53.74
Soumission: £50.10
...

3. Асинхронная загрузка нескольких URL с помощью asyncio

Запрос: «Напиши скрипт, который асинхронно загружает список URL и сохраняет содержимое в файлы».

import asyncio
import aiofiles
import aiohttp

async def download_one(session, url, filename):
    async with session.get(url) as resp:
        async with aiofiles.open(filename, 'wb') as f:
            await f.write(await resp.read())

async def download_all(urls):
    async with aiohttp.ClientSession() as session:
        tasks = [download_one(session, url, f'page_{i}.html') for i, url in enumerate(urls)]
        await asyncio.gather(*tasks)

urls = ['https://example.com', 'https://httpbin.org', 'https://python.org']
asyncio.run(download_all(urls))

(файлы page_0.html, page_1.html, page_2.html созданы в текущей директории)

4. Генерация тестов с помощью pytest

Запрос: «Создай набор тестов для функции, которая проверяет, является ли строка палиндромом».

# Исходная функция
def is_palindrome(s: str) -> bool:
    cleaned = ''.join(c.lower() for c in s if c.isalnum())
    return cleaned == cleaned[::-1]

# Тесты, сгенерированные ИИ:
import pytest
def test_simple_palindrome():
    assert is_palindrome('racecar') == True
    assert is_palindrome('hello') == False

def test_with_spaces_and_punctuation():
    assert is_palindrome('A man, a plan, a canal: Panama') == True
    assert is_palindrome('No lemon, no melon') == True

def test_empty_string():
    assert is_palindrome('') == True

def test_numbers():
    assert is_palindrome('12321') == True
    assert is_palindrome('12345') == False

$ pytest test_palindrome.py -v
... 4 passed in 0.03s

5. Работа с базой данных SQLite через ORM

Запрос: «Создай модель данных для библиотеки (книги, авторы) с использованием SQLAlchemy и добавь несколько записей».

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import declarative_base, relationship, Session

Base = declarative_base()

class Author(Base):
    __tablename__ = 'authors'
    id = Column(Integer, primary_key=True)
    name = Column(String, nullable=False)
    books = relationship('Book', back_populates='author')

class Book(Base):
    __tablename__ = 'books'
    id = Column(Integer, primary_key=True)
    title = Column(String, nullable=False)
    author_id = Column(Integer, ForeignKey('authors.id'))
    author = relationship('Author', back_populates='books')

engine = create_engine('sqlite:///library.db')
Base.metadata.create_all(engine)

with Session(engine) as session:
    author = Author(name='Jane Austen')
    session.add(author)
    session.add(Book(title='Pride and Prejudice', author=author))
    session.add(Book(title='Sense and Sensibility', author=author))
    session.commit()

(файл library.db создан, таблицы заполнены)

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