Збереження даних з ModelForm у базу

У реальних проєктах ми майже завжди працюємо з даними користувачів, будь то реєстрація, відправка коментарів чи оновлення профілю. ModelForm дозволяє нам спростити процес: не потрібно вручну створювати поля для форми чи писати зайвий код для валідації даних. Збереження даних у базу стає не лише ефективним, але й безпечним.

Основи збереження даних у базу

Найголовніше, що ти маєш знати: ModelForm вміє зберігати дані прямо в твою базу, використовуючи метод save(). Цей метод бере введені користувачем дані, перевіряє їх і створює (або оновлює) відповідний об'єкт моделі. Давай розберемо це на практиці.

Приклад 1: просте збереження даних

Припустимо, у нас є модель Article:

from django.db import models

class Article(models.Model):
    title = models.CharField(max_length=100)
    content = models.TextField()
    published_at = models.DateTimeField(null=True, blank=True)

    def __str__(self):
        return self.title

Тепер створимо ModelForm для цієї моделі:

from django import forms
from .models import Article

class ArticleForm(forms.ModelForm):
    class Meta:
        model = Article
        fields = ['title', 'content', 'published_at']

І ось наше представлення:

from django.shortcuts import render, redirect
from .forms import ArticleForm

def create_article(request):
    if request.method == 'POST':
        form = ArticleForm(request.POST)
        if form.is_valid():  # Перевіряємо, що всі поля заповнені коректно
            article = form.save()  # Зберігаємо дані у базу
            return redirect('article_success')  # Перенаправляємо користувача
    else:
        form = ArticleForm()

    return render(request, 'create_article.html', {'form': form})

А це наш шаблон (дуже простий — без CSS, але зате мінімалістичний):

<form method="post">
    {% csrf_token %}
    {{ form.as_p }}  <!-- Відображаємо форму -->
    <button type="submit">Зберегти статтю</button>
</form>

Такий код створює новий об'єкт Article і зберігає його у базі даних. Просто, але ефективно.

Куди ми зберігаємо дані?

Коли ти викликаєш метод save(), Django створює та зберігає об'єкт у базу. Якщо ти хочеш зберегти об'єкт, але не одразу, використовуй метод save(commit=False). Тоді об'єкт тільки створюється в пам'яті, але не записується в базу. Цей трюк часто використовується для попередньої роботи з об'єктом перед його остаточним записом.

Приклад 2: Використання save(commit=False)

Уяви, у нас з'явилася нова ідея: при створенні статті автоматично призначати її статус "чернетка", якщо поле published_at порожнє. Ось як це можна зробити:

def create_article(request):
    if request.method == 'POST':
        form = ArticleForm(request.POST)
        if form.is_valid():
            # Зберігаємо об'єкт без запису в базу
            article = form.save(commit=False)
            if not article.published_at:
                article.published_at = None  # Тут ми можемо задати якісь значення
            article.save()  # Тепер дійсно зберігаємо в базу
            return redirect('article_success')
    else:
        form = ArticleForm()

    return render(request, 'create_article.html', {'form': form})

Робіть безпечно: обробка помилок

Іноді база даних може дати тобі "зворотний зв'язок" не в найпривітнішій формі, наприклад, викинути виключення. Це може статися, якщо дані форми не пройшли ретельної валідації (або якщо хтось навмисно вирішив пожартувати та надіслав тобі невалідний запит). Тому завжди оточуй свої дії з базою обробкою помилок.

from django.db import DatabaseError

def create_article(request):
    if request.method == 'POST':
        form = ArticleForm(request.POST)
        if form.is_valid():
            try:
                form.save()
                return redirect('article_success')
            except DatabaseError:
                # Тут можна обробити помилку, наприклад, вивести повідомлення користувачу
                form.add_error(None, "Сталася помилка під час збереження даних. Спробуйте пізніше.")
    else:
        form = ArticleForm()

    return render(request, 'create_article.html', {'form': form})

Користувач не помітить різниці (якщо ти коректно покажеш повідомлення), але твій сервер не впаде від непередбаченого бага.

Додаткові параметри методу save()

Метод save() у ModelForm дозволяє передавати додаткові параметри. Наприклад, якщо ви хочете вказати поточного користувача (автора) для створеного об'єкта.

Приклад 3: додавання додаткових даних

Припустимо, в Article є поле author, яке пов'язане з моделлю User:

from django.contrib.auth.models import User

class Article(models.Model):
    title = models.CharField(max_length=100)
    content = models.TextField()
    published_at = models.DateTimeField(null=True, blank=True)
    author = models.ForeignKey(User, on_delete=models.CASCADE)

    def __str__(self):
        return self.title

Тепер у представленні ми можемо передати поточного користувача:

def create_article(request):
    if request.method == 'POST':
        form = ArticleForm(request.POST)
        if form.is_valid():
            article = form.save(commit=False)  # Зберігаємо, але не пишемо у базу
            article.author = request.user  # Додаємо поточного користувача
            article.save()  # Зберігаємо остаточно
            return redirect('article_success')
    else:
        form = ArticleForm()

    return render(request, 'create_article.html', {'form': form})

Як обробляти збережені дані?

Коли форма успішно зберігає об'єкт, вона повертає цей об'єкт. Це дозволяє тобі одразу використовувати його, наприклад, для генерації повідомлень, редиректів або передачі даних у шаблони.

Приклад 4: виведення інформації про збережений об'єкт

def create_article(request):
    if request.method == 'POST':
        form = ArticleForm(request.POST)
        if form.is_valid():
            article = form.save()
            return render(request, 'article_created.html', {'article': article})  # Передаємо об'єкт у шаблон
    else:
        form = ArticleForm()

    return render(request, 'create_article.html', {'form': form})

А в шаблоні ти можеш використовувати переданий об'єкт:

<h1>Стаття "{{ article.title }}" успішно створена!</h1>
<p>Номер статті: {{ article.id }}</p>

Помилки, які ви, можливо, зустрінете

Одна з найпоширеніших помилок — забувати про параметр commit=False. Якщо ти намагаєшся щось змінити в об'єкті перед збереженням у базу, але забув вказати commit=False, отримаєш неприємний сюрприз: об'єкт вже буде записаний у базу.

Інша проблема пов'язана з тим, що ти можеш не передати всі необхідні дані у форму. Якщо модель очікує обов'язкове поле, але ти не додав його у форму, база даних поскаржиться. Завжди перевіряй налаштування моделі, щоб уникнути сюрпризів.

І останнє: не забувай валідовувати дані перед збереженням, навіть якщо ти впевнений, що "все має працювати". Користувачі (або зловмисники) можуть бути дуже винахідливими.

На цьому все! Тепер ти знаєш, як зберігати дані із ModelForm у базу даних, включаючи обробку помилок і додавання додаткових даних. Наступним кроком буде кастомізація полів форми, щоб твої форми виглядали привабливо і були зручними у використанні.