cornflex 1.1.5__tar.gz → 1.2.0__tar.gz

{cornflex-1.1.5 → cornflex-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: cornflex
-Version: 1.1.5
+Version: 1.2.0
 Summary: simple sftp reader
 Project-URL: Homepage, https://github.com/ebremstedt/cornflex
 Project-URL: Issues, https://github.com/ebremstedt/cornflex/issues

{cornflex-1.1.5 → cornflex-1.2.0}/cornflex/reader.py

@@ -5,7 +5,7 @@ import shutil
 import tempfile
 import zipfile
 from dataclasses import dataclass
-from typing import Any, Generator, List, Optional
+from typing import Any, Callable, Generator, List, Optional
 
 import chardet
 import paramiko
@@ -218,3 +218,56 @@ class SFTPReader:
         except Exception as e:
             print(f"Error getting {file_name}: {e}")
             return None
+
+    def _upload_and_finalize(
+        self,
+        write_to_temp: Callable[[str], None],
+        remote_path: str,
+        file_name: str,
+    ) -> paramiko.SFTPAttributes:
+        if not self._sftp:
+            raise RuntimeError("Not connected. Call connect() first.")
+
+        final_path = f"{remote_path.rstrip('/')}/{file_name}"
+        tmp_path = f"{final_path}.part"
+        write_to_temp(tmp_path)
+        self._sftp.rename(tmp_path, final_path)
+        return self._sftp.stat(final_path)
+
+    def put_bytes(
+        self,
+        content: bytes,
+        file_name: str,
+        remote_path: str = ".",
+    ) -> paramiko.SFTPAttributes:
+        def _write(tmp_path: str) -> None:
+            with io.BytesIO(content) as buffer:
+                self._sftp.putfo(buffer, tmp_path)
+
+        return self._upload_and_finalize(_write, remote_path, file_name)
+
+    def put_string(
+        self,
+        content: str,
+        file_name: str,
+        remote_path: str = ".",
+        encoding: str = "utf-8",
+    ) -> paramiko.SFTPAttributes:
+        return self.put_bytes(
+            content=content.encode(encoding),
+            file_name=file_name,
+            remote_path=remote_path,
+        )
+
+    def put_file(
+        self,
+        local_path: str,
+        remote_path: str = ".",
+        file_name: Optional[str] = None,
+    ) -> paramiko.SFTPAttributes:
+        remote_file_name = file_name or os.path.basename(local_path)
+
+        def _write(tmp_path: str) -> None:
+            self._sftp.put(local_path, tmp_path)
+
+        return self._upload_and_finalize(_write, remote_path, remote_file_name)

{cornflex-1.1.5 → cornflex-1.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cornflex"
-version = "1.1.5"
+version = "1.2.0"
 authors = [
   { name="Erik Bremstedt", email="erik.bremstedt@gmail.com" },
 ]

{cornflex-1.1.5 → cornflex-1.2.0}/tests/test_reader.py

@@ -223,3 +223,97 @@ def test_file_to_string_returns_none_on_error(connected_reader: SFTPReader) -> N
     connected_reader._sftp.file.side_effect = Exception("boom")
     result = connected_reader.file_to_string("bad.txt")
     assert result is None
+
+
+# --- put_bytes ---
+
+
+def test_put_bytes_raises_when_not_connected(reader_password: SFTPReader) -> None:
+    with pytest.raises(RuntimeError, match="Not connected"):
+        reader_password.put_bytes(b"hello", "file.txt")
+
+
+def test_put_bytes_writes_to_temp_then_renames(connected_reader: SFTPReader) -> None:
+    captured: dict = {}
+
+    def _capture_putfo(buffer, tmp_path) -> None:
+        captured["content"] = buffer.read()
+        captured["tmp_path"] = tmp_path
+
+    connected_reader._sftp.putfo.side_effect = _capture_putfo
+    connected_reader._sftp.stat.return_value = "final-attrs"
+
+    result = connected_reader.put_bytes(b"hello", "orders.csv", remote_path="/out")
+
+    assert captured == {"content": b"hello", "tmp_path": "/out/orders.csv.part"}
+    connected_reader._sftp.rename.assert_called_once_with(
+        "/out/orders.csv.part", "/out/orders.csv"
+    )
+    connected_reader._sftp.stat.assert_called_once_with("/out/orders.csv")
+    assert result == "final-attrs"
+
+
+# --- put_string ---
+
+
+def test_put_string_encodes_and_delegates_to_put_bytes(
+    connected_reader: SFTPReader,
+) -> None:
+    connected_reader.put_bytes = MagicMock(return_value="attrs")
+
+    result = connected_reader.put_string("héllo", "orders.csv", remote_path="/out")
+
+    connected_reader.put_bytes.assert_called_once_with(
+        content="héllo".encode("utf-8"),
+        file_name="orders.csv",
+        remote_path="/out",
+    )
+    assert result == "attrs"
+
+
+def test_put_string_respects_custom_encoding(connected_reader: SFTPReader) -> None:
+    connected_reader.put_bytes = MagicMock(return_value="attrs")
+
+    connected_reader.put_string("héllo", "orders.csv", encoding="latin-1")
+
+    connected_reader.put_bytes.assert_called_once_with(
+        content="héllo".encode("latin-1"),
+        file_name="orders.csv",
+        remote_path=".",
+    )
+
+
+# --- put_file ---
+
+
+def test_put_file_raises_when_not_connected(reader_password: SFTPReader) -> None:
+    with pytest.raises(RuntimeError, match="Not connected"):
+        reader_password.put_file("/local/orders.csv")
+
+
+def test_put_file_streams_from_disk_then_renames(connected_reader: SFTPReader) -> None:
+    connected_reader._sftp.stat.return_value = "final-attrs"
+
+    result = connected_reader.put_file("/local/orders.csv", remote_path="/out")
+
+    connected_reader._sftp.put.assert_called_once_with(
+        "/local/orders.csv", "/out/orders.csv.part"
+    )
+    connected_reader._sftp.rename.assert_called_once_with(
+        "/out/orders.csv.part", "/out/orders.csv"
+    )
+    connected_reader._sftp.stat.assert_called_once_with("/out/orders.csv")
+    assert result == "final-attrs"
+
+
+def test_put_file_uses_explicit_file_name(connected_reader: SFTPReader) -> None:
+    connected_reader.put_file(
+        "/local/orders.csv", remote_path="/out", file_name="custom.csv"
+    )
+
+    connected_reader._sftp.put.assert_called_once_with(
+        "/local/orders.csv", "/out/custom.csv.part"
+    )
+    connected_reader._sftp.rename.assert_called_once_with(
+        "/out/custom.csv.part", "/out/custom.csv"
+    )

{cornflex-1.1.5 → cornflex-1.2.0}/uv.lock

@@ -219,7 +219,7 @@ wheels = [
 
 [[package]]
 name = "cornflex"
-version = "1.1.3"
+version = "1.1.5"
 source = { editable = "." }
 dependencies = [
     { name = "chardet" },