LucaCappelletti94/italian_csv_type_prediction

from .set_type_predictor import SetTypePredictor
from .string_type import StringType
from ..datasets import load_codice_catasto


class CadastreCodeType(StringType):

    def __init__(self):
        super().__init__()
        self._predictor = SetTypePredictor(
            load_codice_catasto(), normalize_values=True)

    def validate(self, candidate, **kwargs) -> bool:
        """Return boolean representing if given candidate is a valid country code."""
        return super().validate(candidate, **kwargs) and self._predictor.validate(candidate)