From a18fb8aa8733d50bd90db636bbc4065314892af2 Mon Sep 17 00:00:00 2001
From: Sean C <sean@corrgs.ca>
Date: Sun, 1 Jan 2023 01:17:16 -0500
Subject: [PATCH] Init commit with preliminary func

---
 main.py | 65 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..1d2194e
--- /dev/null
+++ b/main.py
@@ -0,0 +1,65 @@
+#
+# Pulling and pushing md files to/from bookstack
+#
+
+import requests, json, os, creds
+
+## Tweakables
+bookstackURL = "https://bs.corrgs.ca/api"
+keyToken = creds.keyToken
+secretToken = creds.secretToken
+
+exportFolder = "export"
+importFolder = "import"
+
+
+## Advanced
+debug = True
+
+headers = {}
+authHeader = { "Authorization" : f"Token {keyToken}:{secretToken}" }
+headers.update(authHeader)
+stars = "*"*25
+
+
+######
+# Book formats
+# Book > Chapter > Page
+######
+
+def pullAllBooks():
+    req = requests.get(f'{bookstackURL}/books', headers=headers)
+    books = []
+    for book in json.loads(req.content.decode())['data']:
+       books.append(book)
+    if debug:
+        print(f"{stars}\nPulling books status code: {req.status_code}\n{stars}")
+        print(books)
+    return books
+
+def pullAllChapters(bookID):
+    req = requests.get(f'{bookstackURL}/books', headers=headers)
+    chapters = []
+
+def backupAllBooks(): ## This exports each book as a single page... it works but isn't a true backup solution
+    jsonTests = []
+    for book in pullAllBooks():
+        req = requests.get(f"{bookstackURL}/books/{book['id']}/export/markdown", headers=headers)
+        with open(f"{exportFolder}/{book['name']}", 'w+') as file:
+            file.write(req.content.decode())
+
+def importBooks(): ##### Incomplete
+    importFiles = os.listdir(importFolder)
+    for file in importFiles:
+        with open(f'{importFolder}/{file}', 'r') as file:
+            filedata = file.read()
+
+if debug:
+    ## Pull all books
+    pullAllBooks()
+
+    ## Download allbooks
+    backupAllBooks()
+
+    ## Import test books
+    importBooks()
\ No newline at end of file