pincer.utils.api_data

Complexity

Total Complexity

6

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	APIDataGen.__async()	0	3	1
A	APIDataGen.__await__()	0	2	1
A	APIDataGen.__init__()	0	8	2
A	APIDataGen.__aiter__()	0	3	2

# Copyright Pincer 2021-Present

Missing module docstring

# Full MIT License can be found in `LICENSE` at the project root.

from __future__ import annotations

from typing import Generic, TypeVar, TYPE_CHECKING

from pincer.utils import APIObject

if TYPE_CHECKING:
    from typing import Any, Coroutine, List, Type, Generator, AsyncIterator

T = TypeVar('T')


class APIDataGen(Generic[T]):

Missing class docstring

    def __init__(
        self,

Wrong hanging indentation before block (add 4 spaces).

        factory: Type[T],

Wrong hanging indentation before block (add 4 spaces).

        request_func: Coroutine[Any, None, Any]

Wrong hanging indentation before block (add 4 spaces).

    ):

        self.fac = factory if isinstance(factory, APIObject) else factory.from_dict
        self.request_func = request_func

    async def __async(self) -> List[T]:
        data = await self.request_func
        return [self.fac(i) for i in data]

    def __await__(self) -> Generator[Any, None, Any]:
        return self.__async().__await__()

    async def __aiter__(self) -> AsyncIterator[List[T]]:
        for item in await self:
            yield self.fac(item)