Skip to content

Commit

Permalink
use custom fixture for cover
Browse files Browse the repository at this point in the history
  • Loading branch information
sunForest committed May 14, 2015
1 parent 2e1689c commit e499d9e
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 4 deletions.
5 changes: 2 additions & 3 deletions avipost/avipost/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,8 @@
from django.conf.urls.static import static

from rest_framework import routers

from postcards.views import PostcardViewSet



router = routers.DefaultRouter()
router.register(r'postcards', PostcardViewSet)

Expand All @@ -16,4 +13,6 @@
)

if settings.DEBUG:
import autofixture
autofixture.autodiscover()
urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)
19 changes: 19 additions & 0 deletions avipost/postcards/autofixtures.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
from .models import Postcard
from autofixture import generators, register, AutoFixture


class PostcardAutoFixture(AutoFixture):

default_cover_sizes = (
(800, 600),
(600, 800),
(1200, 600),
(600, 1200),
(1000, 1000),
)

field_values = {
'cover': generators.ImageGenerator(sizes=default_cover_sizes),
}

register(Postcard, PostcardAutoFixture)
3 changes: 2 additions & 1 deletion avipost/postcards/models.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
from django.db import models


class Postcard(models.Model):
# didn't use TextField here because Django doesn't check max length for textField?
message = models.CharField('Message', max_length=140, default="Hello!")
cover = models.ImageField('Cover')
sender = models.CharField('Sender', max_length=20, default="demo")
sender = models.CharField('Sender', max_length=20, default="demo")

0 comments on commit e499d9e

Please sign in to comment.