23 lines
744 B
Python
23 lines
744 B
Python
from DesignEntity import DesignEntity
|
|
from PIL import ImageDraw, ImageOps
|
|
|
|
|
|
class BoxDesign (DesignEntity):
|
|
"""Redefinition of ImageDraw.Draw.Rectangle"""
|
|
|
|
def __init__(self, size, fill=None, outline=None, width=0):
|
|
super(BoxDesign, self).__init__((size[0]+1, size[1]+1), mask=True)
|
|
self.size = size
|
|
self.__define_corners__()
|
|
self.fill = fill
|
|
self.outline = outline
|
|
self.width = width
|
|
|
|
def __define_corners__(self):
|
|
topleft = (0, 0)
|
|
bottomright = self.size
|
|
self.corners = [topleft, bottomright]
|
|
|
|
def __finish_image__(self):
|
|
ImageDraw.Draw(self.__image__).rectangle(
|
|
self.corners, fill=self.fill, outline=self.outline, width=self.width)
|