52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
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)
|