<filename>nexus/ingest/jobs/crossref_api.py
from datetime import (
    datetime,
    timedelta,
)
from typing import (
    Any,
    AsyncIterable,
    Iterable,
    Optional,
)

from aiocrossref import CrossrefClient
from nexus.ingest.jobs.base import BaseJob


class CrossrefApiJob(BaseJob):
    name = 'crossref-api'

    def __init__(
        self,
        base_url: str,
        max_retries: int,
        retry_delay: int,
        actions: Iterable[dict],
        sinks: Iterable[dict],
        from_date: Optional[str] = None,
    ):
        super().__init__(actions=actions, sinks=sinks)
        self.crossref_client = CrossrefClient(base_url=base_url, max_retries=max_retries, retry_delay=retry_delay)
        self.from_date = from_date or str(datetime.date(datetime.now()) - timedelta(days=1))
        self.starts.append(self.crossref_client)

    async def iterator(self) -> AsyncIterable[Any]:
        async for chunk in self.crossref_client.works_cursor(
            filter=f'from-index-date:{self.from_date}',
            rows=1000,
            select='DOI',
        ):
            for item in chunk['items']:
                yield item
