Dr. Transmisia's changes

master
Chuck Sneed 2022-07-22 19:50:12 -05:00
parent 7939a8c10d
commit fae35e819f
1 changed files with 10 additions and 3 deletions

View File

@ -45,24 +45,27 @@ class RDramaAPIInterface:
''' '''
Gets "all" comments. Gets "all" comments.
''' '''
def get_comments(self, number_of_pages=1, user=None): def get_comments(self, number_of_pages=1, user=None, sort="new", upper_bound = 0, lower_bound = 0):
if (user == None): if (user == None):
url=f"{self.protocol}://{self.site}/comments" url=f"{self.protocol}://{self.site}/comments"
else: else:
url=f"{self.protocol}://{self.site}/@{user}/comments" url=f"{self.protocol}://{self.site}/@{user}/comments"
params = f"?sort={sort}&t=all&before={upper_bound}&after={lower_bound}"
url+=params
if number_of_pages == 1: if number_of_pages == 1:
return self.get(url) return self.get(url)
else: else:
results = [] results = []
for i_ in range(number_of_pages): for i_ in range(number_of_pages):
i = i_ + 1 i = i_ + 1
full_url=f"{url}?page={i}&sort=new&t=all" full_url = f"{url}&page={i}"
results += self.get(full_url)['data'] results += self.get(full_url)['data']
return { return {
'data': results 'data': results
} }
''' '''
Calls the notifications endpoint Calls the notifications endpoint
''' '''
@ -85,6 +88,10 @@ class RDramaAPIInterface:
url=f"{self.protocol}://{self.site}" url=f"{self.protocol}://{self.site}"
return self.get(url) return self.get(url)
def get_hole(self, hole: str):
url = f"{self.protocol}://{self.site}/h/{hole}"
return self.get(url)
def has_url_been_posted(self, the_url): def has_url_been_posted(self, the_url):
url=f"{self.protocol}://{self.site}/is_repost" url=f"{self.protocol}://{self.site}/is_repost"
return self.post(url, {'url': the_url})['permalink'] != '' return self.post(url, {'url': the_url})['permalink'] != ''