text.py

from typing import Optional

from regula.documentreader.webclient.ext.models.text_field import TextField
from regula.documentreader.webclient.gen.models import LCID, Text as GenText


class Text(GenText):

    def get_field(self, field_type: int, lcid: int = None) -> Optional[TextField]:
        result = None
        for field in self.field_list:
            if field.field_type == field_type:
                if lcid is not None and field.lcid == lcid:
                    return TextField.from_json(field.to_json())
                elif lcid is None and field.lcid == LCID.LATIN:
                    return TextField.from_json(field.to_json())
                elif lcid is None and result is None:
                    result = field
        return TextField.from_json(result.to_json())

    def get_field_value(self, field_type: int, lcid: int = None) -> Optional[str]:
        field = self.get_field(field_type, lcid)
        return field.value if field else None

    def get_field_by_name(self, field_name: str, lcid: int = None) -> Optional[TextField]:
        result = None
        for field in self.field_list:
            if field.field_name == field_name:
                if lcid is not None and field.lcid == lcid:
                    return TextField.from_json(field.to_json())
                elif lcid is None and field.lcid == LCID.LATIN:
                    return TextField.from_json(field.to_json())
                elif lcid is None and result is None:
                    result = field
        return TextField.from_json(result.to_json())

    def get_field_value_by_name(self, field_name: str, lcid: int = None) -> Optional[str]:
        field = self.get_field_by_name(field_name, lcid)
        return field.value if field else None