bookstack/Bookstack.py

97 lines
2.9 KiB
Python

import base64
import json
import requests
def get_auth_header(token_id, token_secret):
return {"Authorization": "Token {}:{}".format(token_id, token_secret)}
def get_page_html(wiki_url, token_id, token_secret, page_id):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
r = requests.get(wiki_url + "/api/pages/{}/export/html".format(page_id), headers=headers)
return r.text
def get_attachments(wiki_url, token_id, token_secret):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
r = requests.get(wiki_url + "/api/attachments", headers=headers)
attachment_data = json.loads(r.text)['data']
return attachment_data
def get_attachment_id(wiki_url, token_id, token_secret, attachment_name, page_id):
attachment_data = get_attachments(wiki_url, token_id, token_secret)
for attachment in attachment_data:
if attachment['name'] == attachment_name and attachment['uploaded_to'] == page_id:
return attachment['id']
return None
def post_attachment(wiki_url, token_id, token_secret, filename, filedata, page_id):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
data = {}
data["name"] = filename
data["uploaded_to"] = page_id
files = {}
files["file"] = filedata
requests.post(wiki_url + "/api/attachments", data=data, files=files, headers=headers)
def get_attachment(wiki_url, token_id, token_secret, attachment_id):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
r = requests.get(wiki_url + "/api/attachments/{}".format(attachment_id), headers=headers)
attachment_data_encoded = json.loads(r.text)['content']
attachment_data_decoded = base64.b64decode(attachment_data_encoded)
return attachment_data_decoded
def update_attachment(wiki_url, token_id, token_secret, attachment_id, file_name, file_data, page_id):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
data = {}
# Workaround for updating files: https://github.com/BookStackApp/BookStack/issues/3666
data["_method"] = "PUT"
data["name"] = file_name
data["uploaded_to"] = page_id
files = {}
files["file"] = file_data
requests.post(wiki_url + "/api/attachments/{}".format(attachment_id), headers=headers, data=data, files=files)
def delete_attachment(wiki_url, token_id, token_secret, attachment_id):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
requests.delete(wiki_url + "/api/attachments/{}".format(attachment_id), headers=headers)
def update_page(wiki_url, token_id, token_secret, page_id, markdown=None, html=None):
headers = {}
headers.update(get_auth_header(token_id, token_secret))
data = {}
data['markdown'] = markdown
data['html'] = html
requests.put(wiki_url + "/api/pages/{}".format(page_id), data=data, headers=headers)