init-strawberry

services/core.py

@@ -1,82 +1,18 @@
-from httpx import AsyncClient
-from settings import API_BASE
 from typing import List
-from models.member import ChatMember
+
+from httpx import AsyncClient
+
+from orm.author import Author
+from orm.shout import Shout
+from settings import API_BASE
 
 headers = {"Content-Type": "application/json"}
 
 
-async def get_all_authors() -> List[ChatMember]:
-    query_name = "authorsAll"
-    query_type = "query"
-    operation = "AuthorsAll"
-    query_fields = "id slug pic name"
-
-    gql = {
-        "query": query_type + " " + operation + " { " + query_name + " { " + query_fields + " } " + " }",
-        "operationName": operation,
-        "variables": None,
-    }
-
+async def _request_endpoint(query_name, body):
     async with AsyncClient() as client:
         try:
-            response = await client.post(API_BASE, headers=headers, json=gql)
-            print(f"[services.core] {query_name}: [{response.status_code}] {len(response.text)} bytes")
-            if response.status_code != 200:
-                return []
-            r = response.json()
-            if r:
-                return r.get("data", {}).get(query_name, [])
-        except Exception:
-            import traceback
-
-            traceback.print_exc()
-    return []
-
-
-async def get_my_followings() -> List[ChatMember]:
-    query_name = "loadMySubscriptions"
-    query_type = "query"
-    operation = "LoadMySubscriptions"
-    query_fields = "id slug pic name"
-
-    gql = {
-        "query": query_type + " " + operation + " { " + query_name + " { " + query_fields + " } " + " }",
-        "operationName": operation,
-        "variables": None,
-    }
-
-    async with AsyncClient() as client:
-        try:
-            response = await client.post(API_BASE, headers=headers, json=gql)
-            print(f"[services.core] {query_name}: [{response.status_code}] {len(response.text)} bytes")
-            if response.status_code != 200:
-                return []
-            r = response.json()
-            if r:
-                return r.get("data", {}).get(query_name, {}).get("authors", [])
-        except Exception:
-            import traceback
-
-            traceback.print_exc()
-    return []
-
-
-async def get_author(author_id) -> Author:
-    query_name = "getAuthor"
-    query_type = "query"
-    operation = "GetAuthor"
-    query_fields = "id slug pic name"
-
-    gql = {
-        "query": query_type + " " + operation + " { " + query_name + " { authors {" + query_fields + "} } " + " }",
-        "operationName": operation,
-        "variables": None,
-    }
-
-    async with AsyncClient() as client:
-        try:
-            response = await client.post(API_BASE, headers=headers, json=gql)
+            response = await client.post(API_BASE, headers=headers, json=body)
             print(f"[services.core] {query_name}: [{response.status_code}] {len(response.text)} bytes")
             if response.status_code != 200:
                 return []
@@ -89,3 +25,31 @@ async def get_author(author_id) -> Author:
             import traceback
 
             traceback.print_exc()
+
+
+async def get_author(author_id) -> Author:
+    query_name = "getAuthor"
+    query_type = "query"
+    operation = "GetAuthor"
+    query_fields = "id slug pic name"
+
+    gql = {
+        "query": query_type + " " + operation + " { " + query_name + " { " + query_fields + "м} " + " }",
+        "operationName": operation,
+        "variables": None,
+    }
+
+    return await _request_endpoint(query_name, gql)
+
+
+async def get_followed_shouts(author_id: int) -> List[Shout]:
+    query_name = "getFollowedShouts"
+    query_type = "query"
+    operation = "GetFollowedShouts"
+    query_fields = "id slug title"
+
+    query = f"{query_type} {operation}($author_id: Int!) {{ {query_name}(author_id: $author_id) {{ {query_fields} }} }}"
+
+    body = {"query": query, "operationName": operation, "variables": {"author_id": author_id}}
+
+    return await _request_endpoint(query_name, body)