openrelik-api-client 0.2.1__tar.gz → 0.2.3__tar.gz

{openrelik_api_client-0.2.1 → openrelik_api_client-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openrelik-api-client
-Version: 0.2.1
+Version: 0.2.3
 Summary: API client that automatically handles token refresh
 Author: Johan Berggren
 Author-email: jberggren@gmail.com

{openrelik_api_client-0.2.1 → openrelik_api_client-0.2.3}/openrelik_api_client/api_client.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import tempfile
 from requests_toolbelt import MultipartEncoder
 from requests.exceptions import RequestException
 import requests
@@ -71,6 +72,26 @@ class APIClient:
         url = f"{self.base_url}{endpoint}"
         return self.session.delete(url, **kwargs)
 
+    def download_file(self, file_id: int, filename: str) -> str | None:
+        """Downloads a file from OpenRelik.
+
+        Args:
+            file_id: The ID of the file to download.
+            filename: The name of the file to download.
+
+        Returns:
+            str: The path to the downloaded file.
+        """
+        endpoint = f"{self.base_url}/files/{file_id}/download"
+        response = self.session.get(endpoint)
+        filename_prefix, extension = os.path.splitext(filename)
+        file = tempfile.NamedTemporaryFile(
+            mode="wb", prefix=f"{filename_prefix}", suffix=extension, delete=False
+        )
+        file.write(response.content)
+        file.close()
+        return file.name
+
     def upload_file(
         self, file_path: str, folder_id: int) -> int | None:
         """Uploads a file to the server.
@@ -108,11 +129,15 @@ class APIClient:
             while chunk := fh.read(chunk_size):
                 resumableChunkNumber += 1
                 params = {
-                    "resumableChunkNumber": str(resumableChunkNumber),
-                    "resumableTotalChunks": str(resumableTotalChunks),
+                    "resumableRelativePath": resumableFilename,
+                    "resumableTotalSize": total_size,
+                    "resumableCurrentChunkSize": chunk_size,
+                    "resumableChunkSize": chunk_size,
+                    "resumableChunkNumber": resumableChunkNumber,
+                    "resumableTotalChunks": resumableTotalChunks,
                     "resumableIdentifier": resumableIdentifier,
                     "resumableFilename": resumableFilename,
-                    "folder_id": str(folder_id),
+                    "folder_id": folder_id
                 }
                 encoder = MultipartEncoder(
                     {"file": (file_path.name, chunk,

{openrelik_api_client-0.2.1 → openrelik_api_client-0.2.3}/openrelik_api_client/folders.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Any
+
 from openrelik_api_client.api_client import APIClient
 
 
@@ -81,3 +83,26 @@ class FoldersAPI:
         response = self.api_client.session.get(endpoint)
         response.raise_for_status()
         return response.status_code == 200
+
+    def update_folder(
+        self, folder_id: int, folder_data: dict[str, Any]
+    ):
+        """Updates an existing folder.
+
+        Args:
+            folder_id: The ID of the folder to update.
+            folder_data: The updated folder data.
+
+        Returns:
+            The updated folder data, or None.
+
+        Raises:
+            HTTPError: If the API request failed.
+        """
+        endpoint = f"{self.api_client.base_url}/folders/{folder_id}"
+        response = self.api_client.session.patch(
+            endpoint,
+            json=folder_data
+        )
+        response.raise_for_status()
+        return response.json()

{openrelik_api_client-0.2.1 → openrelik_api_client-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openrelik-api-client"
-version = "0.2.1"
+version = "0.2.3"
 description = "API client that automatically handles token refresh"
 authors = ["Johan Berggren <jberggren@gmail.com>"]
 readme = "README.md"

openrelik_api_client-0.2.1/openrelik_api_client/test.py

@@ -1,11 +0,0 @@
-from folders import FoldersAPI
-
-
-def main():
-    c = FoldersAPI("http://172.19.0.6:8710", api_key="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJmZThmMjkyYWViZDM0MjA3YjkyODYxOGQ5MmRmZThkMSIsImlhdCI6MTczNDQ0NzcyOSwibmJmIjoxNzM0NDQ3NzI5LCJleHAiOjE3MzUwNTI1MjksImlzcyI6Imh0dHA6Ly9sb2NhbGhvc3Q6ODcxMCIsImF1ZCI6ImFwaS1jbGllbnQiLCJqdGkiOiI2ZWZhNzBiYWU0ZDE0MjJlOGM4ODY5OGY1NDhmOTBjYiIsInRva2VuX3R5cGUiOiJyZWZyZXNoIn0.MJA48D570agD49oE4HXfTWCQRpyh5JCy7YCIsHN5YJs")
-    # r = c.create_root_folder("myfolder")
-    # r = c.create_subfolder(33, 'myfolder_sub')
-    print(c.folder_exists(1))
-
-
-main()