adds directory upload.

Author: filiplajszczak

pythonanywhere_core/files.py

@@ -1,4 +1,5 @@
 import getpass
+from pathlib import Path
 from typing import Tuple, Union
 from urllib.parse import urljoin
 
@@ -174,3 +175,23 @@ def tree_get(self, path: str) -> dict:
             return result.json()
 
         raise PythonAnywhereApiException(f"GET to {url} failed, got {result}{self._error_msg(result)}")
+
+    def tree_post(self, local_dir_path: str, remote_dir_path: str) -> None:
+        """Uploads contents of a local directory to remote path on
+        PythonAnywhere.  Walks `local_dir_path` recursively and uploads
+        each file using :meth:`path_post`, preserving directory structure.
+
+        Raises :exc:`PythonAnywhereApiException` on first upload failure."""
+
+        local_dir = Path(local_dir_path)
+        if not local_dir.is_dir():
+            raise ValueError(f"{local_dir_path} is not a directory")
+        for path in sorted(local_dir.rglob("*")):
+            if path.is_file():
+                relative = path.relative_to(local_dir)
+                remote_path = f"{remote_dir_path}/{relative}"
+                self.path_post(remote_path, path.read_bytes())
+            elif path.is_dir() and not any(path.iterdir()):
+                placeholder = f"{remote_dir_path}/{path.relative_to(local_dir)}/.empty"
+                self.path_post(placeholder, b"")
+                self.path_delete(placeholder)

tests/test_files.py

@@ -399,3 +399,88 @@ def test_raises_when_path_does_not_exist(
         f"GET to {url} failed, got <Response [400]>: {invalid_path} does not exist"
     )
     assert str(e.value) == expected_error_msg
+
+
+def test_tree_post_uploads_single_file(
+        api_token, api_responses, base_url, home_dir_path, tmp_path
+):
+    (tmp_path / "index.html").write_bytes(b"<h1>hello</h1>")
+    remote_dir = f"{home_dir_path}/myapp"
+    remote_file_url = f"{base_url}path{remote_dir}/index.html"
+    api_responses.add(responses.POST, url=remote_file_url, status=201)
+
+    Files().tree_post(str(tmp_path), remote_dir)
+
+    assert len(api_responses.calls) == 1
+    assert api_responses.calls[0].request.url == remote_file_url
+
+
+def test_tree_post_preserves_nested_structure(
+        api_token, api_responses, base_url, home_dir_path, tmp_path
+):
+    (tmp_path / "index.html").write_bytes(b"<h1>hello</h1>")
+    (tmp_path / "static" / "css").mkdir(parents=True)
+    (tmp_path / "static" / "css" / "style.css").write_bytes(b"body {}")
+    (tmp_path / "static" / "app.js").write_bytes(b"console.log('hi')")
+
+    remote_dir = f"{home_dir_path}/myapp"
+    api_responses.add(responses.POST, url=f"{base_url}path{remote_dir}/index.html", status=201)
+    api_responses.add(responses.POST, url=f"{base_url}path{remote_dir}/static/app.js", status=201)
+    api_responses.add(responses.POST, url=f"{base_url}path{remote_dir}/static/css/style.css", status=201)
+
+    Files().tree_post(str(tmp_path), remote_dir)
+
+    assert len(api_responses.calls) == 3
+    uploaded_urls = {call.request.url for call in api_responses.calls}
+    assert uploaded_urls == {
+        f"{base_url}path{remote_dir}/index.html",
+        f"{base_url}path{remote_dir}/static/app.js",
+        f"{base_url}path{remote_dir}/static/css/style.css",
+    }
+
+
+def test_tree_post_creates_empty_directories(
+        api_token, api_responses, base_url, home_dir_path, tmp_path
+):
+    (tmp_path / "empty_dir").mkdir()
+    remote_dir = f"{home_dir_path}/myapp"
+    empty_file_url = f"{base_url}path{remote_dir}/empty_dir/.empty"
+    api_responses.add(responses.POST, url=empty_file_url, status=201)
+    api_responses.add(responses.DELETE, url=empty_file_url, status=204)
+
+    Files().tree_post(str(tmp_path), remote_dir)
+
+    assert len(api_responses.calls) == 2
+    assert api_responses.calls[0].request.url == empty_file_url
+    assert api_responses.calls[0].request.method == "POST"
+    assert api_responses.calls[1].request.url == empty_file_url
+    assert api_responses.calls[1].request.method == "DELETE"
+
+
+def test_tree_post_raises_when_path_is_not_a_directory(
+        api_token, tmp_path
+):
+    file_path = tmp_path / "somefile.txt"
+    file_path.write_bytes(b"content")
+
+    with pytest.raises(ValueError) as e:
+        Files().tree_post(str(file_path), "/home/user/remote")
+
+    assert "is not a directory" in str(e.value)
+
+
+def test_tree_post_fails_fast_on_upload_error(
+        api_token, api_responses, base_url, home_dir_path, tmp_path
+):
+    (tmp_path / "a.txt").write_bytes(b"aaa")
+    (tmp_path / "b.txt").write_bytes(b"bbb")
+    (tmp_path / "c.txt").write_bytes(b"ccc")
+
+    remote_dir = f"{home_dir_path}/myapp"
+    api_responses.add(responses.POST, url=f"{base_url}path{remote_dir}/a.txt", status=201)
+    api_responses.add(responses.POST, url=f"{base_url}path{remote_dir}/b.txt", status=403)
+
+    with pytest.raises(PythonAnywhereApiException):
+        Files().tree_post(str(tmp_path), remote_dir)
+
+    assert len(api_responses.calls) == 2