Sync widget

desktop_client/file.py

@@ -0,0 +1,124 @@
+from __future__ import annotations
+
+import base64
+import datetime
+import hashlib
+import os
+import uuid
+
+import file_widgets
+import pydantic
+from PyQt6.QtGui import QIcon
+from PyQt6.QtWidgets import QFileDialog, QLabel, QMessageBox, QWidget
+from request_client import RequestClient
+
+
+class File(pydantic.BaseModel):
+    file_id: uuid.UUID
+    file_name: str
+    file_size: int
+    sha512: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+
+    def name(self) -> str:
+        return self.file_name
+
+    def delete(self) -> None:
+        RequestClient().client.delete(
+            "/files", params={"file_id": self.file_id}
+        )
+
+    def details(self, list: file_widgets.FileListWidget) -> QWidget:
+        del list
+        file_size = self._format_bytes(self.file_size)
+        file_size_text = f"{file_size[0]:.2f} {file_size[1]}"
+        details = (
+            f"file id: {self.file_id}\nfile_name: {self.file_name}\n"
+            + f"file_size: {file_size_text}\n"
+            + f"created at: {self.created_at}\nupdated at: {self.updated_at}"
+        )
+        label = QLabel()
+        label.setWindowTitle("File info")
+        label.setText(details)
+        return label
+
+    @staticmethod
+    def _format_bytes(size: int):
+        power = 2**10
+        n = 0
+        power_labels = {0: "", 1: "kibi", 2: "mebi", 3: "gibi", 4: "tebi"}
+        while size > power and n < 4:
+            size /= power
+            n += 1
+        return size, power_labels[n] + "bytes"
+
+    def icon(self) -> QIcon:
+        return QIcon("assets/file.png")
+
+    def double_click(self, list: file_widgets.FileListWidget) -> None:
+        location = QFileDialog.getExistingDirectory(
+            list, caption="Select save location"
+        )
+        if not location:
+            return
+        with open(
+            os.path.join(location, self.file_name), "wb"
+        ) as f, RequestClient().client.stream(
+            "GET", "/files", params={"file_id": self.file_id}
+        ) as stream:
+            if not stream.is_success:
+                QMessageBox.warning(list, "Error downloading the file")
+                return
+            for data in stream.iter_bytes():
+                f.write(data)
+
+    def create(path: str, parent_id: uuid.UUID):
+        """Upload the file"""
+        file_name = os.path.basename(path)
+        try:
+            with open(path, "rb") as f:
+                files = {"file": (file_name, f)}
+                response = RequestClient().client.post(
+                    "/files",
+                    files=files,
+                    params={"parent_folder": parent_id},
+                )
+                if not response.is_success:
+                    QMessageBox.warning(
+                        None, "Error", f"Upload failed: {response.text}"
+                    )
+                print(response.text)
+        except Exception as e:
+            QMessageBox.critical(None, "HTTP Error", str(e))
+
+    def download(self, path: str):
+        with open(path, "wb") as f, RequestClient().client.stream(
+            "GET", "/files", params={"file_id": self.file_id}
+        ) as stream:
+            if not stream.is_success:
+                QMessageBox.warning(None, "Error downloading the file")
+                return
+            for data in stream.iter_bytes():
+                f.write(data)
+
+    def modify(self, path: str):
+        """Upload the file"""
+        file_name = os.path.basename(path)
+        hash = hashlib.file_digest(open(path, "rb"), "sha512").digest()
+        if base64.b64decode(self.sha512) == hash:
+            return
+        try:
+            with open(path, "rb") as f:
+                files = {"file": (file_name, f)}
+                response = RequestClient().client.patch(
+                    "/files",
+                    files=files,
+                    params={"file_id": self.file_id},
+                )
+                if not response.is_success:
+                    QMessageBox.warning(
+                        None, "Error", f"Upload failed: {response.text}"
+                    )
+        except Exception as e:
+            QMessageBox.critical(None, "HTTP Error", str(e))