doccano/backend/data_import/pipeline/data.py

import abc
import uuid
from typing import Any, Dict

from pydantic import BaseModel, validator

from api.models import Example, Project


class BaseData(BaseModel, abc.ABC):
    filename: str

    @classmethod
    def parse(cls, **kwargs):
        return cls.parse_obj(kwargs)

    def __hash__(self):
        return hash(tuple(self.dict()))

    @abc.abstractmethod
    def create(self, project: Project, meta: Dict[Any, Any]) -> Example:
        raise NotImplementedError('Please implement this method in the subclass.')


class TextData(BaseData):
    text: str

    @validator('text')
    def text_is_not_empty(cls, value: str):
        if value:
            return value
        else:
            raise ValueError('is not empty.')

    def create(self, project: Project, meta: Dict[Any, Any]) -> Example:
        return Example(
            uuid=uuid.uuid4(),
            project=project,
            filename=self.filename,
            text=self.text,
            meta=meta
        )


class FileData(BaseData):

    def create(self, project: Project, meta: Dict[Any, Any]) -> Example:
        return Example(
            uuid=uuid.uuid4(),
            project=project,
            filename=self.filename,
            meta=meta
        )