File size: 2,595 Bytes
a3d6c18 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import re
from typing import Optional
from pydantic import BaseModel, validator
from typing_extensions import Literal
class Parameters(BaseModel):
image_file_b64: Optional[str] = ""
image_url: Optional[str] = ""
size: Optional[Literal["preview", "full", "auto"]] = "preview"
type: Optional[
Literal["auto", "product", "person", "car"]
] = "auto" # Not supported at the moment
format: Optional[Literal["auto", "jpg", "png", "zip"]] = "auto"
roi: str = "0% 0% 100% 100%"
crop: bool = False
crop_margin: Optional[str] = "0px"
scale: Optional[str] = "original"
position: Optional[str] = "original"
channels: Optional[Literal["rgba", "alpha"]] = "rgba"
add_shadow: str = "false" # Not supported at the moment
semitransparency: str = "false" # Not supported at the moment
bg_color: Optional[str] = ""
bg_image_url: Optional[str] = ""
@validator("crop_margin")
def crop_margin_validator(cls, value):
if not re.match(r"[0-9]+(px|%)$", value):
raise ValueError(
"crop_margin paramter is not valid"
) # TODO: Add support of several values
if "%" in value and (int(value[:-1]) < 0 or int(value[:-1]) > 100):
raise ValueError("crop_margin mast be in range between 0% and 100%")
return value
@validator("scale")
def scale_validator(cls, value):
if value != "original" and (
not re.match(r"[0-9]+%$", value)
or not int(value[:-1]) <= 100
or not int(value[:-1]) >= 10
):
raise ValueError("scale must be original or in between of 10% and 100%")
if value == "original":
return 100
return int(value[:-1])
@validator("position")
def position_validator(cls, value, values):
if len(value.split(" ")) > 2:
raise ValueError(
"Position must be a value from 0 to 100 "
"for both vertical and horizontal axises or for both axises respectively"
)
if value == "original":
return "original"
elif len(value.split(" ")) == 1:
return [int(value[:-1]), int(value[:-1])]
else:
return [int(value.split(" ")[0][:-1]), int(value.split(" ")[1][:-1])]
@validator("bg_color")
def bg_color_validator(cls, value):
if not re.match(r"(#{0,1}[0-9a-f]{3}){0,2}$", value):
raise ValueError("bg_color is not in hex")
if len(value) and value[0] != "#":
value = "#" + value
return value
|