first commit

functions/__init__.py

@@ -0,0 +1,32 @@
+from . import get_set_Book
+from . import set_Variables
+from . import backupType
+from . import backupLogic
+from . import get_confirmation
+import bookstack
+
+def main(version):
+    print(f"Functions module running version: {version}")
+    env = set_Variables.set_variables()
+    print(f"You are currently running on Version: {version}")
+
+    api = bookstack.BookStack(env['base_url'], token_id=env['token_id'], token_secret=env['token_secret'])
+    api.generate_api_methods()
+
+    try:
+        book_list = api.get_books_list()['data']
+    except Exception as e:
+        print(f"Error loading books: {e}")
+        return
+
+    get_set_Book.get_Books(book_list)
+    book_to_backup = get_set_Book.set_Book(book_list)
+    backup_extension = backupType.get_backup_extension()
+    backup_extension_fileType = backupType.set_backup_extension_filetype(backup_extension)
+    book_to_backup = next(book for book in book_list if book['id'] == book_to_backup)
+
+    if get_confirmation.get_confirmation(backup_extension, book_to_backup):
+        backupLogic.backup_book(api, env, book_to_backup, backup_extension, backup_extension_fileType)
+        print("Backup completed.")
+    else:
+        return

functions/backupLogic.py

@@ -0,0 +1,31 @@
+import os
+import requests
+
+
+def backup_book(api, env, selected_book, backup_extension, backup_extension_fileType):
+    base_url = env['base_url']
+    backup_dir = env['backup_dir']
+    headers = {
+        "Authorization": f"Token {env['token_id']}:{env['token_secret']}"
+    }
+
+    pages_list = api.get_pages_list()['data']
+    chapter_list = api.get_chapters_list()['data']
+
+    print(f"Backup type: {backup_extension}")
+
+    for chapter in chapter_list:
+        for page in pages_list:
+            if page['book_id'] == selected_book['id']:
+                if chapter['id'] == page['chapter_id']:
+                    folder = os.path.join(backup_dir, chapter['name'])
+                    filename = os.path.join(folder, page['name'] + backup_extension_fileType)
+                    print(f"Backing up: {filename}")
+                    if not os.path.exists(folder):
+                        os.makedirs(folder, exist_ok=True)
+                    response = requests.get(
+                        f'{base_url}/api/pages/{page["id"]}/export/{backup_extension}',
+                        headers=headers
+                    )
+                    with open(filename, 'wb') as file:
+                        file.write(response.content)

functions/backupType.py

@@ -0,0 +1,24 @@
+def get_backup_extension():
+    backup_type = input("Enter backup type (md, pdf, html, text): ").strip().lower()
+    if backup_type in ["md", "markdown"]:
+        return "markdown"
+    elif backup_type == "pdf":
+        return "PDF"
+    elif backup_type == "html":
+        return "HTML"
+    elif backup_type in ["plain", "text", "txt", "plain_text"]:
+        return "Plaintext"
+    else:
+        raise ValueError("Unsupported backup type")
+
+def set_backup_extension_filetype(backup_type):
+    if backup_type == "markdown":
+        return ".md"
+    elif backup_type == "PDF":
+        return ".pdf"
+    elif backup_type == "HTML":
+        return ".html"
+    elif backup_type == "Plaintext":
+        return ".txt"
+    else:
+        print("Error file type not found")

functions/get_confirmation.py

@@ -0,0 +1,14 @@
+def get_confirmation(backup_extension, selected_book):
+    print(f"Backup type: {backup_extension}")
+    while True:
+        confirm = input("Are these settings correct? (y/n): ").strip().lower()
+        if confirm == "y":
+            print(f"Starting backup for book: {selected_book['name']} with filetype {backup_extension}.")
+            print("Starting now.......")
+            return True
+        elif confirm == "n":
+            print("Stopping now.....")
+            return False
+        else:
+            print("Please enter 'y' or 'n'.")
+

functions/get_set_Book.py

@@ -0,0 +1,16 @@
+def get_Books(book_list):
+    print("Available books:")
+    print("Book_ID: ", "Book_Name")
+    for book in book_list:
+        print(f"{book['id']}: {book['name']}")
+
+def set_Book(book_list):
+    while True:
+        try:
+            book_to_backup = int(input("Enter the book ID to backup: "))
+            if any(book['id'] == book_to_backup for book in book_list):
+                return book_to_backup
+            else:
+                print("Invalid book ID, please try again.")
+        except ValueError:
+            print("Please enter a valid number.")

functions/set_Variables.py

@@ -0,0 +1,11 @@
+from dotenv import load_dotenv
+import os
+
+def set_variables():
+    load_dotenv()
+    return {
+        "base_url": os.getenv('BASE_URL'),
+        "token_id": os.getenv('TOKEN_ID'),
+        "token_secret": os.getenv('TOKEN_SECRET'),
+        "backup_dir": os.getenv('BACKUP_DIR')
+    }