60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
from httpx import AsyncClient
|
|
from settings import API_BASE
|
|
|
|
headers = {"Content-Type": "application/json"}
|
|
INTERNAL_AUTH_SERVER = 'v2.' in API_BASE
|
|
|
|
async def get_all_authors():
|
|
query_name = "authorsAll"
|
|
query_type = "query"
|
|
operation = "AuthorsAll"
|
|
query_fields = "id slug userpic name"
|
|
headers = {"Content-Type": "application/json"} # "Bearer " + removed
|
|
|
|
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)
|
|
except Exception:
|
|
import traceback
|
|
traceback.print_exc()
|
|
print(f"[services.core] {query_name}: {response.status_code} {response.text}")
|
|
if response.status_code != 200:
|
|
return None
|
|
r = response.json()
|
|
return r.get("data", {}).get(query_name)
|
|
|
|
|
|
async def get_my_followings():
|
|
query_name = "loadMySubscriptions"
|
|
query_type = "query"
|
|
operation = "LoadMySubscriptions"
|
|
query_fields = "id slug userpic name"
|
|
headers = {"Content-Type": "application/json"} # "Bearer " + removed
|
|
|
|
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)
|
|
except Exception:
|
|
import traceback
|
|
traceback.print_exc()
|
|
print(f"[services.core] {query_name}: {response.status_code} {response.text}")
|
|
if response.status_code != 200:
|
|
return None
|
|
r = response.json()
|
|
authors = r.get("data", {}).get(query_name).get("authors", [])
|
|
return authors
|
|
|
|
members = get_all_authors() # [ { id slug name userpic }, ..]
|
|
print(f"[services.core] cache members amount: {len(members)}")
|
|
members_by_slug = {member['slug']: member for member in members}
|
|
members_by_id = {member['id']: member for member in members} |