notifier-draft-2

services/core.py

@@ -10,7 +10,7 @@ async def get_all_authors() -> List[ChatMember]:
     query_name = "authorsAll"
     query_type = "query"
     operation = "AuthorsAll"
-    query_fields = "id slug userpic name"
+    query_fields = "id slug pic name"
 
     gql = {
         "query": query_type + " " + operation + " { " + query_name + " { " + query_fields + " } " + " }",
@@ -38,10 +38,10 @@ async def get_my_followings() -> List[ChatMember]:
     query_name = "loadMySubscriptions"
     query_type = "query"
     operation = "LoadMySubscriptions"
-    query_fields = "id slug userpic name"
+    query_fields = "id slug pic name"
 
     gql = {
-        "query": query_type + " " + operation + " { " + query_name + " { authors {" + query_fields + "} } " + " }",
+        "query": query_type + " " + operation + " { " + query_name + " { " + query_fields + " } " + " }",
         "operationName": operation,
         "variables": None,
     }
@@ -60,3 +60,32 @@ async def get_my_followings() -> List[ChatMember]:
 
             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)
+            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, {})
+            else:
+                raise Exception("json response error")
+        except Exception:
+            import traceback
+
+            traceback.print_exc()