notifier/services/core.py

from typing import Any

import aiohttp

from settings import API_BASE


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


# TODO: rewrite to orm usage?


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} HTTP 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_followed_shouts(author_id: int):
    query_name = 'load_shouts_followed'
    operation = 'GetFollowedShouts'

    query = f"""query {operation}($author_id: Int!, limit: Int, offset: Int) {{
        {query_name}(author_id: $author_id, limit: $limit, offset: $offset) {{ id slug title }}
    }}"""

    gql = {
        'query': query,
        'operationName': operation,
        'variables': {'author_id': author_id, 'limit': 1000, 'offset': 0},  # FIXME: too big limit
    }

    return await _request_endpoint(query_name, gql)


async def get_shout(shout_id):
    query_name = 'get_shout'
    operation = 'GetShout'

    query = f"""query {operation}($slug: String, $shout_id: Int) {{
        {query_name}(slug: $slug, shout_id: $shout_id) {{ id slug title authors {{ id slug name pic }} }}
    }}"""

    gql = {'query': query, 'operationName': operation, 'variables': {'slug': None, 'shout_id': shout_id}}

    return await _request_endpoint(query_name, gql)