-
Notifications
You must be signed in to change notification settings - Fork 10
Expand file tree
/
Copy pathimage_qa.py
More file actions
105 lines (87 loc) · 5.5 KB
/
image_qa.py
File metadata and controls
105 lines (87 loc) · 5.5 KB
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# coding: utf-8
"""
Generated by: https://openapi-generator.tech
"""
from __future__ import annotations
import pprint
import re # noqa: F401
import json
from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt
from typing import Any, ClassVar, Dict, List, Optional, Union
from regula.documentreader.webclient.gen.models.glares_check_params import GlaresCheckParams
from regula.documentreader.webclient.gen.models.input_image_quality_checks import InputImageQualityChecks
from typing import Optional, Set
from typing_extensions import Self
from pydantic import SkipValidation, Field
class ImageQA(BaseModel):
"""
ImageQA
""" # noqa: E501
brightness_threshold: SkipValidation[Optional[float]] = Field(alias="brightnessThreshold", default=None, description="Set the threshold for an actual document brightness below which the check fails")
dpi_threshold: SkipValidation[Optional[int]] = Field(alias="dpiThreshold", default=None, description="This parameter sets threshold for Image QA check of the presented document physical dpi. If actual document dpi is below this threshold, check will fail.")
angle_threshold: SkipValidation[Optional[int]] = Field(alias="angleThreshold", default=None, description="This parameter sets threshold for Image QA check of the presented document perspective angle in degrees. If actual document perspective angle is above this threshold, check will fail.")
focus_check: SkipValidation[Optional[bool]] = Field(alias="focusCheck", default=None, description="This option enables focus check while performing image quality validation.")
glares_check: SkipValidation[Optional[bool]] = Field(alias="glaresCheck", default=None, description="This option enables glares check while performing image quality validation.")
colorness_check: SkipValidation[Optional[bool]] = Field(alias="colornessCheck", default=None, description="This option enables colorness check while performing image quality validation.")
moire_check: SkipValidation[Optional[bool]] = Field(alias="moireCheck", default=None, description="This option enables screen capture (moire patterns) check while performing image quality validation.")
document_position_indent: SkipValidation[Optional[int]] = Field(alias="documentPositionIndent", default=None, description="This parameter specifies the necessary margin. Default 0.")
expected_pass: SkipValidation[Optional[List[InputImageQualityChecks]]] = Field(alias="expectedPass", default=None, description="This parameter controls the quality checks that the image should pass to be considered a valid input during the scanning process.")
glares_check_params: SkipValidation[Optional[GlaresCheckParams]] = Field(alias="glaresCheckParams", default=None)
__properties: ClassVar[List[str]] = ["brightnessThreshold", "dpiThreshold", "angleThreshold", "focusCheck", "glaresCheck", "colornessCheck", "moireCheck", "documentPositionIndent", "expectedPass", "glaresCheckParams"]
model_config = ConfigDict(
populate_by_name=True,
validate_assignment=True,
protected_namespaces=(),
arbitrary_types_allowed=True,
use_enum_values=True
)
def to_str(self) -> str:
"""Returns the string representation of the model using alias"""
return pprint.pformat(self.model_dump(by_alias=True))
def to_json(self) -> str:
"""Returns the JSON representation of the model using alias"""
# TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
return json.dumps(self.to_dict())
@classmethod
def from_json(cls, json_str: str) -> Optional[Self]:
"""Create an instance of ImageQA from a JSON string"""
return cls.from_dict(json.loads(json_str))
def to_dict(self) -> Dict[str, Any]:
"""Return the dictionary representation of the model using alias.
This has the following differences from calling pydantic's
`self.model_dump(by_alias=True)`:
* `None` is only added to the output dict for nullable fields that
were set at model initialization. Other fields with value `None`
are ignored.
"""
excluded_fields: Set[str] = set([
])
_dict = self.model_dump(
by_alias=True,
exclude=excluded_fields,
exclude_none=True,
)
# override the default output from pydantic by calling `to_dict()` of glares_check_params
if self.glares_check_params:
_dict['glaresCheckParams'] = self.glares_check_params.to_dict()
return _dict
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of ImageQA from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"brightnessThreshold": obj.get("brightnessThreshold"),
"dpiThreshold": obj.get("dpiThreshold"),
"angleThreshold": obj.get("angleThreshold"),
"focusCheck": obj.get("focusCheck"),
"glaresCheck": obj.get("glaresCheck"),
"colornessCheck": obj.get("colornessCheck"),
"moireCheck": obj.get("moireCheck"),
"documentPositionIndent": obj.get("documentPositionIndent"),
"expectedPass": obj.get("expectedPass"),
"glaresCheckParams": GlaresCheckParams.from_dict(obj["glaresCheckParams"]) if obj.get("glaresCheckParams") is not None else None
})
return _obj