inbox/services/core.py

from typing import Any, List

import aiohttp
from aiohttp import ClientResponse, ClientSession
from models.member import ChatMember
from settings import API_BASE

headers = {"Content-Type": "application/json"}


async def _request_endpoint(query_name, body) -> Any:
    async with aiohttp.ClientSession() as session:
        async with session.post(API_BASE, headers=headers, json=body) as response:
            print(f"[services.core] {query_name} response: <{response.status}> {await response.text()}")
            if response.status == 200:
                r = await response.json()
                if r:
                    return r.get("data", {}).get(query_name, {})
            return []


async def get_all_authors() -> List[ChatMember]:
    query_name = "authorsAll"

    gql = {
        "query": "query { " + query_name + " { id slug pic name } }",
        "variables": None,
    }

    return await _request_endpoint(query_name, gql)


async def get_my_followed() -> List[ChatMember]:
    query_name = "get_my_followed"

    gql = {
        "query": "query { " + query_name + " { authors { id slug pic name } } }",
        "variables": None,
    }

    result = await _request_endpoint(query_name, gql)

    return result.get("authors", [])


async def get_author(user: str = ""):
    query_name = "get_author_id"
    operation = "GetAuthorId"

    gql = {
        "query": f"query {operation}($user: String!) {{ {query_name}(user: $user) {{ id slug pic name }} }}",
        "operationName": operation,
        "variables": {"user": user},
    }

    r = await _request_endpoint(query_name, gql)
    if r and isinstance(r, list):
        r = r.pop()
    return r or None