Fix 90, 180 and 270 degrees rotation transformations.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-09-18 21:34:58 -04:00
parent 55284c4681
commit 618a5f0f59
2 changed files with 40 additions and 4 deletions

View File

@@ -6,8 +6,9 @@ from documents.tests import GenericDocumentTestCase
from ..models import Transformation
from ..transformations import (
BaseTransformation, TransformationCrop, TransformationResize,
TransformationRotate, TransformationZoom
BaseTransformation, TransformationCrop, TransformationLineArt,
TransformationResize, TransformationRotate, TransformationRotate90,
TransformationRotate180, TransformationRotate270, TransformationZoom
)
from .literals import (
@@ -154,3 +155,37 @@ class TransformationTestCase(GenericDocumentTestCase):
)
self.assertTrue(document_page.generate_image().startswith('page'))
def test_lineart_transformations(self):
document_page = self.document.pages.first()
Transformation.objects.add_for_model(
obj=document_page, transformation=TransformationLineArt,
arguments={}
)
self.assertTrue(document_page.generate_image().startswith('page'))
def test_rotate_transformations(self):
document_page = self.document.pages.first()
Transformation.objects.add_for_model(
obj=document_page, transformation=TransformationRotate90,
arguments={}
)
self.assertTrue(document_page.generate_image().startswith('page'))
Transformation.objects.add_for_model(
obj=document_page, transformation=TransformationRotate180,
arguments={}
)
self.assertTrue(document_page.generate_image().startswith('page'))
Transformation.objects.add_for_model(
obj=document_page, transformation=TransformationRotate270,
arguments={}
)
self.assertTrue(document_page.generate_image().startswith('page'))

View File

@@ -210,8 +210,9 @@ class TransformationRotate(BaseTransformation):
if self.degrees == 0:
return self.image
if self.fillcolor:
fillcolor = ImageColor.getrgb(self.fillcolor)
fillcolor_value = getattr(self, 'fillcolor', None)
if fillcolor_value:
fillcolor = ImageColor.getrgb(fillcolor_value)
else:
fillcolor = None