From 58caa3566e828ab4605b566c84dc632af326d6dd Mon Sep 17 00:00:00 2001 From: Karl Hobley Date: Sun, 26 Oct 2014 14:02:20 +0000 Subject: [PATCH] Added get_rect method to image --- wagtail/wagtailimages/models.py | 3 +++ wagtail/wagtailimages/tests/test_models.py | 4 ++++ 2 files changed, 7 insertions(+) diff --git a/wagtail/wagtailimages/models.py b/wagtail/wagtailimages/models.py index 4e89a3ad66..185ad534ee 100644 --- a/wagtail/wagtailimages/models.py +++ b/wagtail/wagtailimages/models.py @@ -73,6 +73,9 @@ class AbstractImage(models.Model, TagSearchable): def __str__(self): return self.title + def get_rect(self): + return Rect(0, 0, self.width, self.height) + def get_focal_point(self): if self.focal_point_x is not None and \ self.focal_point_y is not None and \ diff --git a/wagtail/wagtailimages/tests/test_models.py b/wagtail/wagtailimages/tests/test_models.py index 382a18ecec..5c14958d4c 100644 --- a/wagtail/wagtailimages/tests/test_models.py +++ b/wagtail/wagtailimages/tests/test_models.py @@ -13,6 +13,7 @@ from wagtail.tests.models import EventPage, EventPageCarouselItem from wagtail.wagtailimages.models import Rendition from wagtail.wagtailimages.backends import get_image_backend from wagtail.wagtailimages.backends.pillow import PillowBackend +from wagtail.wagtailimages.utils.rect import Rect from .utils import Image, get_test_image_file @@ -31,6 +32,9 @@ class TestImage(TestCase): def test_is_landscape(self): self.assertTrue(self.image.is_landscape()) + def test_get_rect(self): + self.assertTrue(self.image.get_rect(), Rect(0, 0, 640, 480)) + class TestImagePermissions(TestCase): def setUp(self):