update

scripts/helper/paperless_api.py

@@ -0,0 +1,45 @@
+# TODO: The user can use anything in the standard library, installed for paperless
+# or use the custom startup scripts to install additional libraries via pip
+
+import requests
+
+
+class PaperlessAPI:
+    def __init__(self, api_url, auth_token, timeout=5) -> None:
+        self._base_api_url = api_url
+        self._auth_token = auth_token
+        self._timeout = timeout
+
+    def _get_item_by_id(self, item_type, item_id):
+        if item_id:
+            response = requests.get(f"{self._base_api_url}/{item_type}/{item_id}/",
+                                    headers = {"Authorization": f"Token {self._auth_token}"})
+        if response.ok:
+            return response.json()
+
+        return {}  
+
+    def get_document_by_id(self, document_id):
+        return self._get_item_by_id("documents", document_id)
+
+    def get_tag_id_by_name(self, tag_name):
+        response = requests.get(f"{self._base_api_url}/tags/",
+                            headers = {"Authorization": f"Token {self._auth_token}"})
+        if response.ok:
+            print("tags ok")
+            response_json = response.json()
+            tag_list = []
+            tag_list.extend(response_json.get("results"))
+            fetched_tag = [tag for tag in tag_list if tag.get("name") == tag_name]
+            if len(fetched_tag) > 0:
+                print(f'fetched = {fetched_tag}')
+                return fetched_tag[0].get("id")
+
+            print("not found, returning none")
+        return None
+
+    def patch_document(self, document_id, data):
+        return requests.patch(f"{self._base_api_url}/documents/{document_id}/",
+                            headers = {"Authorization": f"Token {self._auth_token}"},
+                            data = data,
+                            timeout = self._timeout)