Bussy-boy/client.py

149 lines
3.9 KiB
Python

import requests
import sys
import os
import math
import time
import shelve
from requests.adapters import HTTPAdapter, Retry
from config import config
class DramaClient:
BASE_URL = "https://rdrama.net"
def __init__(self):
self.db = shelve.open(f"{config['data_dir']}/client_state.p", writeback=True)
self.db.setdefault("processed_replies", set())
self.session = requests.Session()
retries = Retry(
total=5, backoff_factor=5, status_forcelist=[500, 502, 503, 504, 521]
)
self.session.mount("https://", HTTPAdapter(max_retries=retries))
self.chud_phrase = self.get("/@me").get("chud_phrase", "")
def get(self, endpoint):
print("GET", endpoint)
time.sleep(5)
while True:
r = self.session.get(
f"{self.BASE_URL}{endpoint}",
headers={"Authorization": config["api_token"]},
)
if "502 Bad Gateway" in r.text:
print("Received 502")
time.sleep(10)
continue
break
# Return None for country club and chudrama posts.
if r.status_code == 403:
return None
if r.status_code != 200:
print("Error!", r, r.status_code, r.content)
sys.exit(1)
return r.json()
def post(self, endpoint, payload=None, files=[]):
print("POST", endpoint, f"Payload:\n{payload}")
time.sleep(5)
while True:
r = self.session.post(
f"{self.BASE_URL}{endpoint}",
payload,
headers={"Authorization": config["api_token"]},
files=files,
)
if "502 Bad Gateway" in r.text:
print("Received 502")
time.sleep(10)
continue
break
if r.status_code != 200:
print("Error!", r, r.status_code, r.content)
sys.exit(1)
return r.json()
def fetch_new_comments(self, limit=0):
comments = []
last_processed_id = self.db.get("last_processed_id", -1)
earliest_id = math.inf
page = 1
# Fetch comments until we find the last one processed.
while earliest_id > last_processed_id:
page_comments = self.fetch_page(page)
if len(page_comments) == 0:
break
earliest_id = min([c["id"] for c in page_comments])
comments += [c for c in page_comments if c["id"] > last_processed_id]
if limit > 0 and len(comments) >= limit:
break
page += 1
if not comments:
return []
self.db["last_processed_id"] = max(c["id"] for c in comments)
self.db.sync()
# New comments may have pushed others to page n+1 while fetching.
deduped_comments = {c["id"]: c for c in comments}.values()
# Oldest first.
comments.reverse()
return comments
def fetch_new_replies(self):
notifs = self.get("/unread")["data"]
notifs = [n for n in notifs if n["body"]]
return notifs
def fetch_page(self, page):
return self.get(f"/comments?page={page}")["data"]
def fetch_context(self, comment):
post = self.get(f"/post/{comment['post_id']}")
if not post:
return None, None
comments = [comment]
while parent_id := comments[-1].get("parent_comment_id", None):
parent = self.get(f"/comment/{parent_id}")
comments.append(parent)
comments.reverse()
return post, comments
def reply(self, body, comment):
if self.chud_phrase and self.chud_phrase not in body:
body += f"\n{self.chud_phrase}"
payload = {
"parent_fullname": f"c_{comment['id']}",
"body": body,
}
self.post("/comment", payload)