Source code for arxiv.submission.domain.annotation

"""
Provides quality-assurance annotations for the submission & moderation system.
"""

from typing import Optional, Union, List
from datetime import datetime
import hashlib
from enum import Enum
from mypy_extensions import TypedDict

from dataclasses import dataclass, asdict, field

from arxiv.taxonomy import Category

from .util import get_tzaware_utc_now
from .agent import Agent, agent_factory



[docs]@dataclass
class Comment:
    """A freeform textual annotation."""

    event_id: str
    creator: Agent
    created: datetime
    proxy: Optional[Agent] = field(default=None)
    body: str = field(default_factory=str)

    def __post_init__(self):
        """Check our agents."""
        if self.creator and type(self.creator) is dict:
            self.creator = agent_factory(**self.creator)
        if self.proxy and type(self.proxy) is dict:
            self.proxy = agent_factory(**self.proxy)



ClassifierResult = TypedDict('ClassifierResult',
                             {'category': Category, 'probability': float})



[docs]@dataclass
class Annotation:
    def __post_init__(self):
        """Check our agents."""
        if self.creator and type(self.creator) is dict:
            self.creator = agent_factory(**self.creator)
        if self.proxy and type(self.proxy) is dict:
            self.proxy = agent_factory(**self.proxy)




[docs]@dataclass
class ClassifierResults(Annotation):
    """Represents suggested classifications from an auto-classifier."""


[docs]    class Classifiers(Enum):
        """Supported classifiers."""

        CLASSIC = "classic"


    event_id: str
    creator: Agent
    created: datetime
    proxy: Optional[Agent] = field(default=None)
    classifier: Classifiers = field(default=Classifiers.CLASSIC)
    results: List[ClassifierResult] = field(default_factory=list)
    annotation_type: str = field(default='ClassifierResults')

    def __post_init__(self):
        """Check our enums."""
        super(ClassifierResults, self).__post_init__()
        self.classifier = self.Classifiers(self.classifier)




[docs]@dataclass
class Feature(Annotation):
    """Represents features drawn from the content of the submission."""


[docs]    class Type(Enum):
        """Supported features."""

        CHARACTER_COUNT = "chars"
        PAGE_COUNT = "pages"
        STOPWORD_COUNT = "stops"
        STOPWORD_PERCENT = "%stop"
        WORD_COUNT = "words"


    event_id: str
    created: datetime
    creator: Agent
    feature_type: Type
    proxy: Optional[Agent] = field(default=None)
    feature_value: Union[int, float] = field(default=0)
    annotation_type: str = field(default='Feature')

    def __post_init__(self):
        """Check our enums."""
        super(Feature, self).__post_init__()
        self.feature_type = self.Type(self.feature_type)



annotation_types = {
    'Feature': Feature,
    'ClassifierResults': ClassifierResults
}



[docs]def annotation_factory(**data) -> Annotation:
    return annotation_types[data.pop('annotation_type')](**data)