CropFilter

Complexity

Total Complexity

13

Size/Duplication

Total Lines	52
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
B	apply()	0	31	6
B	__init__()	0	13	7

"""
This module implement a Crop filter.
"""
from .interface import ImagineFilterInterface
from PIL import Image


class CropFilter(ImagineFilterInterface):
    """
    Crop filter
    """
    start = None
    size = None

    def __init__(self, **kwargs):
        """
        :param kwargs: dict
        """
        if 'start' in kwargs and isinstance(kwargs['start'], list) and len(kwargs['start']) == 2:
            self.start = kwargs.get('start')
        else:
            raise ValueError('Unknown start position.')

        if 'size' in kwargs and isinstance(kwargs['size'], list) and len(kwargs['size']) == 2:
            self.size = kwargs.get('size')
        else:
            raise ValueError('Unknown image size.')

    def apply(self, resource):
        """
        Apply filter to resource
        :param resource: Image
        :return: Image
        """
        if not isinstance(resource, Image.Image):
            raise ValueError('Unknown resource format')

        original_width, original_height = resource.size

        if self.start[0] < original_width or self.start[1] < original_height:
            left = self.start[0]
            upper = self.start[1]
        else:
            left = 0
            upper = 0

        right = self.start[0] + self.size[0]
        lower = self.start[1] + self.size[1]

        resource = resource.crop(
            (
                left,
                upper,
                right if right < original_width else original_width,
                lower if lower < original_height else original_height
            )
        )

        return resource