datashare-python 0.11.1__tar.gz → 0.11.3__tar.gz

{datashare_python-0.11.1 → datashare_python-0.11.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datashare-python
-Version: 0.11.1
+Version: 0.11.3
 Summary: Manage Python tasks and local resources in Datashare
 Project-URL: Homepage, https://icij.github.io/datashare-python/
 Project-URL: Documentation, https://icij.github.io/datashare-python/

{datashare_python-0.11.1 → datashare_python-0.11.3}/datashare_python/objects.py

@@ -309,8 +309,9 @@ class Document(DatashareModel):
                 resource_name = self.path.name
             else:
                 msg = (
-                    f"doc {self} can't turn ES document into a {ProcessedFile.__name__} "
-                    f"without tika metadata resources name in metadata: {self.metadata}"
+                    f"doc {self} can't turn ES document into a"
+                    f" {ProcessedFile.__name__} without tika metadata resources name"
+                    f" in metadata: {self.metadata}"
                 )
                 raise ValueError(msg)
         resource_name = cast(str, resource_name)
@@ -365,7 +366,17 @@ class ManifestEntryStatus(StrEnum):
     PARTIAL = "partial"
 
 
+class User(DatashareModel):
+    id: str
+    name: str | None = None
+    email: str | None = None
+    provider: str | None = None
+    details: dict[str, Any] = Field(default_factory=dict)
+
+
 class TaskArgs(DatashareModel, ABC):
+    user: User | None = None
+
     def as_manifest_task_input(self) -> dict[str, Any]:
         # This is a base implementation, if the input is too large to be dumped,
         # override this and pop large keys

{datashare_python-0.11.1 → datashare_python-0.11.3}/datashare_python/utils.py

@@ -512,13 +512,21 @@ def _write_artifact_bytes(path: Path, artifact: bytes | BytesIO | Path) -> None:
             with path.open("wb") as f:
                 f.write(artifact.read())
         case Path():
-            path.unlink(missing_ok=True)
+            _remove_previous_artifacts(path)
             shutil.move(artifact, path)
         case _:
             msg = f"unsupported artifact type: {artifact.__class__.__name__}"
             raise ValueError(msg)
 
 
+def _remove_previous_artifacts(path: Path) -> None:
+    if isinstance(path, Path) and path.exists():
+        with contextlib.suppress(FileNotFoundError):
+            shutil.rmtree(path)
+    else:
+        path.unlink(missing_ok=True)
+
+
 def debuggable_name(
     path: str, component_size_limit: int = 10, *, deterministic: bool = False
 ) -> str:

{datashare_python-0.11.1 → datashare_python-0.11.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datashare-python"
-version = "0.11.1"
+version = "0.11.3"
 description = "Manage Python tasks and local resources in Datashare"
 authors = [
   { name = "Clément Doumouro", email = "cdoumouro@icij.org" },