33 lines
1,012 B
Python
33 lines
1,012 B
Python
|
'''
|
||
|
|
Class to generate a csv file based on data fetched via Discourse REST API
|
||
|
|
'''
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
class Discourse():
|
||
|
|
def __init__(self, url):
|
||
|
|
self.url = url
|
||
|
|
|
||
|
|
def get_category_data(self, category_id: str) -> requests.Response:
|
||
|
|
"""Get data related to a Discourse category"""
|
||
|
|
headers = {
|
||
|
|
"content-type": "application/json",
|
||
|
|
}
|
||
|
|
url = f"{self.url}/c/{category_id}.json"
|
||
|
|
res = requests.get(url, headers)
|
||
|
|
res.raise_for_status()
|
||
|
|
|
||
|
|
res_json = json.loads(res.text)
|
||
|
|
return res_json
|
||
|
|
|
||
|
|
def get_category_topics(self, category_id: str) -> requests.Response:
|
||
|
|
"""Get topics from a Discourse category"""
|
||
|
|
cat_data = self.get_category_data(category_id)
|
||
|
|
for topic in cat_data["topic_list"]["topics"]:
|
||
|
|
print(f'{topic["id"]}, {topic["title"]}, {topic["tags"]}, {self.url}t/{topic["id"]}')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__=="__main__":
|
||
|
|
dis = Discourse("https://kb.hs3.pl/")
|
||
|
|
dis.get_category_topics(9)
|