feat: add token support (#24)

* Add token related fields to melding model

* Add abstract token generator

* Let melding create action generate token

* Set token expiry moment

* Pass token generator mock and token duration to action in test

meldingen_core/actions/melding.py

@@ -1,4 +1,5 @@
 from abc import ABCMeta, abstractmethod
+from datetime import datetime, timedelta
 from typing import Generic, TypeVar, override
 
 from meldingen_core.actions.base import BaseCreateAction, BaseListAction, BaseRetrieveAction
@@ -7,6 +8,7 @@
 from meldingen_core.models import Melding
 from meldingen_core.repositories import BaseMeldingRepository, BaseRepository
 from meldingen_core.statemachine import BaseMeldingStateMachine, MeldingTransitions
+from meldingen_core.token import BaseTokenGenerator
 
 T = TypeVar("T", bound=Melding)
 T_co = TypeVar("T_co", covariant=True, bound=Melding)
@@ -17,21 +19,30 @@ class MeldingCreateAction(BaseCreateAction[T, T_co]):
 
     _classify: Classifier
     _state_machine: BaseMeldingStateMachine[T]
+    _generate_token: BaseTokenGenerator
+    _token_duration: timedelta
 
     def __init__(
         self,
         repository: BaseRepository[T, T_co],
         classifier: Classifier,
         state_machine: BaseMeldingStateMachine[T],
+        token_generator: BaseTokenGenerator,
+        token_duration: timedelta,
     ):
         super().__init__(repository)
         self._classify = classifier
         self._state_machine = state_machine
+        self._generate_token = token_generator
+        self._token_duration = token_duration
 
     @override
     async def __call__(self, obj: T) -> None:
         await super().__call__(obj)
 
+        obj.token = await self._generate_token()
+        obj.token_expires = datetime.now() + self._token_duration
+
         classification = await self._classify(obj.text)
         obj.classification = classification
 

meldingen_core/models.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from datetime import datetime
 
 
 @dataclass
@@ -14,6 +15,8 @@ class Melding:
 
     text: str
     classification: Classification | None = None
+    token: str | None = None
+    token_expires: datetime | None = None
 
 
 @dataclass

meldingen_core/repositories.py

@@ -2,7 +2,7 @@
 from collections.abc import Collection
 from typing import Generic, TypeVar
 
-from meldingen_core.models import Answer, Classification, Melding, Question, User, Form
+from meldingen_core.models import Answer, Classification, Form, Melding, Question, User
 
 T = TypeVar("T")
 T_co = TypeVar("T_co", covariant=True)

meldingen_core/token.py

@@ -0,0 +1,7 @@
+from abc import ABCMeta, abstractmethod
+
+
+class BaseTokenGenerator(metaclass=ABCMeta):
+    @abstractmethod
+    async def __call__(self) -> str:
+        """Generates and returns token"""

tests/test_actions/test_melding_actions.py

@@ -1,3 +1,4 @@
+from datetime import timedelta
 from unittest.mock import AsyncMock, Mock
 
 import pytest
@@ -14,6 +15,7 @@
 from meldingen_core.models import Classification, Melding
 from meldingen_core.repositories import BaseMeldingRepository
 from meldingen_core.statemachine import BaseMeldingStateMachine, MeldingTransitions
+from meldingen_core.token import BaseTokenGenerator
 
 
 @pytest.mark.asyncio
@@ -22,7 +24,9 @@ async def test_melding_create_action() -> None:
     classifier = AsyncMock(Classifier, return_value=classification)
     state_machine = Mock(BaseMeldingStateMachine)
     repository = Mock(BaseMeldingRepository)
-    action: MeldingCreateAction[Melding, Melding] = MeldingCreateAction(repository, classifier, state_machine)
+    action: MeldingCreateAction[Melding, Melding] = MeldingCreateAction(
+        repository, classifier, state_machine, AsyncMock(BaseTokenGenerator), timedelta(days=3)
+    )
     melding = Melding("text")
 
     await action(melding)