這是我們關於如何使用 HTMX 和 Django 建立 Todo 應用程式系列的第二篇文章。按此處查看第 1 部分。
在第 2 部分中，我們將建立 Todo 模型並透過單元測試實現其基本功能。
在 models.py 中，我們建立 Todo 模型及其基本屬性。我們希望將 Todo 專案與 UserProfile 關聯，以便使用者只能看到自己的專案。待辦事項還將有一個標題和一個布林屬性 is_completed。我們對 Todo 模型有很多未來的想法，例如除了已完成或未開始之外還可以將任務設置為“正在進行”，以及截止日期，但那是以後的事了。現在讓我們保持簡單，盡快在螢幕上顯示一些內容。
注意：在現實世界的應用程式中，我們可能應該考慮使用 UUID 作為 UserProfile 和 Todo 模型的主鍵，但我們現在會保持簡單。
# core/models.py
from django.contrib.auth.models import AbstractUser
from django.db import models # <-- NEW
class UserProfile(AbstractUser):
pass
# NEW
class Todo(models.Model):
title = models.CharField(max_length=255)
is_completed = models.BooleanField(default=False)
user = models.ForeignKey(
UserProfile,
related_name="todos",
on_delete=models.CASCADE,
)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
def __str__(self):
return self.title
讓我們運行新模型的遷移：
❯ uv run python manage.py makemigrations
Migrations for 'core':
core/migrations/0002_todo.py
+ Create model Todo
❯ uv run python manage.py migrate
Operations to perform:
Apply all migrations: admin, auth, contenttypes, core, sessions
Running migrations:
Applying core.0002_todo... OK
讓我們為我們的專案寫第一個測驗。我們希望確保使用者只能看到自己的待辦事項，而看不到其他使用者的項目。
為了幫助我們編寫測試，我們將向我們的專案 model-bakery 新增新的開發依賴項，這簡化了建立虛擬 Django 模型實例的過程。我們還將添加 pytest-django。
❯ uv add model-bakery pytest-django --dev Resolved 27 packages in 425ms Installed 2 packagez in 12ms + model-bakery==1.20.0 + pytest-django==4.9.0
在 pyproject.toml 中，我們需要透過在檔案末尾新增一些行來設定 pytest：
# pyproject.toml # NEW [tool.pytest.ini_options] DJANGO_SETTINGS_MODULE = "todomx.settings" python_files = ["test_*.py", "*_test.py", "testing/python/*.py"]
現在讓我們來寫第一個測試，以確保使用者只能存取自己的待辦事項。
# core/tests/test_todo_model.py
import pytest
@pytest.mark.django_db
class TestTodoModel:
def test_todo_items_are_associated_to_users(self, make_todo, make_user):
[user1, user2] = make_user(_quantity=2)
for i in range(3):
make_todo(user=user1, title=f"user1 todo {i}")
make_todo(user=user2, title="user2 todo")
assert {todo.title for todo in user1.todos.all()} == {
"user1 todo 0",
"user1 todo 1",
"user1 todo 2",
}
assert {todo.title for todo in user2.todos.all()} == {"user2 todo"}
我們使用 pytest 中的 conftest.py 檔案來存放我們計劃在測試中使用的所有裝置。 model_bakery 函式庫讓使用最少的樣板檔案建立 UserProfile 和 Todo 的實例變得簡單。
#core/tests/conftest.py
import pytest
from model_bakery import baker
@pytest.fixture
def make_user(django_user_model):
def _make_user(**kwargs):
return baker.make("core.UserProfile", **kwargs)
return _make_user
@pytest.fixture
def make_todo(make_user):
def _make_todo(user=None, **kwargs):
return baker.make("core.Todo", user=user or make_user(), **kwargs)
return _make_todo
讓我們執行測試！
❯ uv run pytest
Test session starts (platform: darwin, Python 3.12.8, pytest 8.3.4, pytest-sugar 1.0.0)
django: version: 5.1.4, settings: todomx.settings (from ini)
configfile: pyproject.toml
plugins: sugar-1.0.0, django-4.9.0
collected 1 item
core/tests/test_todo_model.py ✓ 100% ██████████
Results (0.25s):
1 passed
最後，我們可以為其註冊一個管理頁面：
# core/admin.py
from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import Todo, UserProfile # <-- NEW
# .. previous code
# NEW
@admin.register(Todo)
class TodoAdmin(admin.ModelAdmin):
model = Todo
list_display = ["title", "is_completed", "user"]
list_filter = ["is_completed"]
search_fields = ["title"]
list_per_page = 10
ordering = ["title"]
我們現在可以從管理員添加一些待辦事項！
如果你想查看第 2 部分結束之前的整個程式碼，可以在 Github 的第 02 部分分支上查看。
以上是使用 Django 和 HTMX 建立 To-Do 應用程式 - 使用 TDD 新增 Todo 模型部分的詳細內容。更多資訊請關注PHP中文網其他相關文章！
