chore: improve devcontainer setup

Author: Matthias Zimmermann

.devcontainer/devcontainer.json

@@ -39,5 +39,5 @@
             ]
         }
     },
-    "postCreateCommand": "uv sync --all-groups && git config --global core.editor nano"
+    "postCreateCommand": "bash .devcontainer/post-create.sh"
 }

.devcontainer/post-create.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+set -e
+
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "🔧 Setting up Arkiv SDK development environment..."
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+
+echo ""
+echo "📦 Installing Python dependencies with uv..."
+uv sync --all-groups
+
+echo ""
+echo "⚙️  Configuring git editor..."
+git config --global core.editor nano
+
+echo ""
+echo "✅ Verifying installation..."
+echo "   Python: $(python --version)"
+echo "   UV: $(uv --version)"
+echo "   Pytest: $(uv run pytest --version)"
+echo "   Ruff: $(uv run ruff --version)"
+echo "   Mypy: $(uv run mypy --version)"
+
+echo ""
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "🎉 Dev container is ready!"
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo ""
+echo "Quick start commands:"
+echo "  • ./scripts/check-all.sh     # Run all quality checks"
+echo "  • uv run pytest -n auto      # Run all tests (parallel mode)"
+echo "  • uv run ruff check .        # Lint code"
+echo "  • uv run mypy src/ tests/    # Type check"
+echo ""

src/arkiv/module.py

@@ -420,17 +420,18 @@ def query_entities(
             raise ValueError("Must provide either query or cursor")
 
         if query is not None and len(query.strip()) > 0:
-            logger.info(f"Query: '{query}', limit={limit}, at_block={at_block}, cursor={cursor}")
+            logger.info(
+                f"Query: '{query}', limit={limit}, at_block={at_block}, cursor={cursor}"
+            )
 
             # Fetch raw results from RPC
             raw_results = self.client.eth.query_entities(query)  # type: ignore[attr-defined]
-            
+
             # Transform and log each result
             entities: list[Entity] = []
             for result in raw_results:
                 entity_result = QueryEntitiesResult(
-                    entity_key=result.key,
-                    storage_value=base64.b64decode(result.value)
+                    entity_key=result.key, storage_value=base64.b64decode(result.value)
                 )
                 logger.info(f"Query result item: {entity_result}")
                 entities.append(to_entity(entity_result))

tests/test_entity_query.py

@@ -1,6 +1,5 @@
 """Tests for entity query functionality."""
 
-from unittest import result
 import uuid
 
 import pytest
@@ -30,7 +29,9 @@ def test_query_entities_accepts_query_only(self, arkiv_client_http: Arkiv) -> No
         """Test that query_entities accepts query without cursor."""
         # Should not raise ValueError for missing cursor
         # Query will execute (returns empty result since no matching entities exist)
-        result = arkiv_client_http.arkiv.query_entities(query='owner = "0x0000000000000000000000000000000000000000"')
+        result = arkiv_client_http.arkiv.query_entities(
+            query='owner = "0x0000000000000000000000000000000000000000"'
+        )
         assert not result  # check for falsy result
         assert len(result) == 0  # No entities match this owner
 
@@ -41,7 +42,9 @@ def test_query_entities_accepts_cursor_only(self, arkiv_client_http: Arkiv) -> N
 
         # Should not raise ValueError for missing query
         # Will raise NotImplementedError since cursor-based pagination is not yet implemented
-        with pytest.raises(NotImplementedError, match="not yet implemented for cursors"):
+        with pytest.raises(
+            NotImplementedError, match="not yet implemented for cursors"
+        ):
             arkiv_client_http.arkiv.query_entities(cursor=cursor)
 
     def test_query_entities_both_query_and_cursor_not_allowed(
@@ -52,7 +55,10 @@ def test_query_entities_both_query_and_cursor_not_allowed(
 
         # Should raise ValueError when both are provided
         with pytest.raises(ValueError, match="Cannot provide both query and cursor"):
-            arkiv_client_http.arkiv.query_entities(query='owner = "0x0000000000000000000000000000000000000000"', cursor=cursor)
+            arkiv_client_http.arkiv.query_entities(
+                query='owner = "0x0000000000000000000000000000000000000000"',
+                cursor=cursor,
+            )
 
     def test_query_entities_with_all_parameters(self, arkiv_client_http: Arkiv) -> None:
         """Test that query_entities accepts all parameters."""
@@ -73,7 +79,9 @@ def test_query_entities_with_cursor_and_parameters(
 
         # Should not raise ValueError
         # Will raise NotImplementedError since cursor-based pagination is not yet implemented
-        with pytest.raises(NotImplementedError, match="not yet implemented for cursors"):
+        with pytest.raises(
+            NotImplementedError, match="not yet implemented for cursors"
+        ):
             arkiv_client_http.arkiv.query_entities(
                 cursor=cursor,
                 limit=100,  # Ignored when cursor is provided
@@ -106,12 +114,12 @@ def test_query_entities_by_annotation(self, arkiv_client_http: Arkiv) -> None:
 
         # Verify result basics
         assert result  # Check __bool__()
+        assert result.block_number > 0
         assert result.has_more() is False
         assert result.next_cursor is None  # only 3 results, no pagination needed
 
         # Verify we got back all 3 entities
         assert len(result.entities) == 3
-        assert result.block_number > 0
 
         # Verify the entity keys match (order may differ)
         result_keys = {entity.entity_key for entity in result.entities}