doccano/backend/data_import/pipeline/data.py

import abc
import uuid
from typing import Any, Dict

from pydantic import UUID4, BaseModel, validator

from examples.models import Example
from projects.models import Project


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

    def __init__(self, **data):
        data["uuid"] = uuid.uuid4()
        super().__init__(**data)

    @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=self.uuid,
            project=project,
            filename=self.filename,
            upload_name=self.upload_name,
            text=self.text,
            meta=meta,
        )


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