124 lines
2.9 KiB
Python
124 lines
2.9 KiB
Python
|
import io
|
||
|
|
from math import inf
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class Image(BaseModel):
|
||
|
|
bytes: bytes
|
||
|
|
width: int
|
||
|
|
height: int
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_pil(cls, pil_image):
|
||
|
|
buffer = io.BytesIO()
|
||
|
|
pil_image.save(buffer, format="PNG")
|
||
|
|
buffer.seek(0)
|
||
|
|
|
||
|
|
return cls(
|
||
|
|
bytes=buffer.read(),
|
||
|
|
width=pil_image.width,
|
||
|
|
height=pil_image.height,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class Dimension(BaseModel):
|
||
|
|
mm: float
|
||
|
|
__EPSILON = 0.0001
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_inch(cls, inch: float) -> "Dimension":
|
||
|
|
return cls(mm=inch * 25.4)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_points(cls, points: float, dpi: int) -> "Dimension":
|
||
|
|
return cls.from_inch(points / dpi)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def inch(self) -> float:
|
||
|
|
return self.mm / 25.4
|
||
|
|
|
||
|
|
def in_pixels(self, dpi: int) -> int:
|
||
|
|
return int(self.inch * dpi)
|
||
|
|
|
||
|
|
def __ensure_same_type(self, other):
|
||
|
|
if type(other) != Dimension:
|
||
|
|
raise TypeError(f"Cannot use {other} to {self}")
|
||
|
|
|
||
|
|
def __add__(self, other):
|
||
|
|
self.__ensure_same_type(other)
|
||
|
|
return Dimension(mm=self.mm + other.mm)
|
||
|
|
|
||
|
|
def __sub__(self, other):
|
||
|
|
self.__ensure_same_type(other)
|
||
|
|
return Dimension(mm=self.mm - other.mm)
|
||
|
|
|
||
|
|
def __mul__(self, other):
|
||
|
|
if type(other) not in (int, float):
|
||
|
|
raise TypeError(f"Cannot multiply {self} by {other}")
|
||
|
|
return Dimension(mm=self.mm * other)
|
||
|
|
|
||
|
|
def __rmul__(self, other):
|
||
|
|
return self.__mul__(other)
|
||
|
|
|
||
|
|
def __truediv__(self, other):
|
||
|
|
if type(other) not in (int, float):
|
||
|
|
raise TypeError(f"Cannot divide {self} by {other}")
|
||
|
|
|
||
|
|
return Dimension(mm=self.mm / other)
|
||
|
|
|
||
|
|
def __eq__(self, other):
|
||
|
|
self.__ensure_same_type(other)
|
||
|
|
if self.mm == inf and other.mm == inf:
|
||
|
|
return True
|
||
|
|
return abs(self.mm - other.mm) < self.__EPSILON
|
||
|
|
|
||
|
|
def __lt__(self, other):
|
||
|
|
self.__ensure_same_type(other)
|
||
|
|
return self.mm < other.mm
|
||
|
|
|
||
|
|
def __gt__(self, other):
|
||
|
|
self.__ensure_same_type(other)
|
||
|
|
return self.mm > other.mm
|
||
|
|
|
||
|
|
|
||
|
|
class LabelRequest(BaseModel):
|
||
|
|
text: str
|
||
|
|
length: Dimension | None
|
||
|
|
|
||
|
|
|
||
|
|
class LabelDefinition(BaseModel):
|
||
|
|
text: str
|
||
|
|
length: Dimension | None = None
|
||
|
|
width: Dimension
|
||
|
|
dpi: int
|
||
|
|
|
||
|
|
@property
|
||
|
|
def pixel_width(self):
|
||
|
|
return self.width.in_pixels(self.dpi)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def pixel_length(self):
|
||
|
|
return self.length.in_pixels(self.dpi)
|
||
|
|
|
||
|
|
|
||
|
|
class MediaDefinition(BaseModel):
|
||
|
|
width: Dimension
|
||
|
|
length: Dimension
|
||
|
|
minimal_margin_vertical: Dimension
|
||
|
|
minimal_margin_horizontal: Dimension
|
||
|
|
dpi: int
|
||
|
|
|
||
|
|
@property
|
||
|
|
def printable_width(self) -> Dimension:
|
||
|
|
return self.width - 2 * self.minimal_margin_horizontal
|
||
|
|
|
||
|
|
@property
|
||
|
|
def printable_length(self) -> Dimension:
|
||
|
|
return self.length - 2 * self.minimal_margin_vertical
|
||
|
|
|
||
|
|
|
||
|
|
class Label(BaseModel):
|
||
|
|
dpi: str
|
||
|
|
image: Image
|