Skip to content

Commit b6ba4a6

Browse files
authored
add doctype method (#85)
1 parent eab4c69 commit b6ba4a6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

regula/documentreader/webclient/ext/models/recognition_response.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from typing import Optional, List
2-
from regula.documentreader.webclient import ImageQualityCheckList
2+
from regula.documentreader.webclient import ImageQualityCheckList, OneCandidate
33
from regula.documentreader.webclient.ext.models.authenticity.authenticity_check_list import AuthenticityCheckList
44
from regula.documentreader.webclient.ext.models.images import Images
55
from regula.documentreader.webclient.ext.models.text import Text
@@ -45,6 +45,10 @@ def image_quality_checks(self, page_idx=0) -> Optional[ImageQualityCheckList]:
4545
result = self.pageable_result_by_type(Result.IMAGE_QUALITY, page_idx)
4646
return result.image_quality_check_list if result is not None else None
4747

48+
def document_type(self, page_idx=0) -> Optional[OneCandidate]:
49+
result = self.pageable_result_by_type(Result.DOCUMENT_TYPE, page_idx)
50+
return result.one_candidate if result else None
51+
4852
def pageable_result_by_type(self, result_type: int, page_idx: int) -> Optional[ResultItem]:
4953
container_list = self.low_lvl_response.container_list.list
5054
for response in container_list:

0 commit comments

Comments
 (0)