mongo-pipebuilder 0.2.1__tar.gz → 0.2.2__tar.gz

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/LICENSE

@@ -20,3 +20,4 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 
+

{mongo_pipebuilder-0.2.1/src/mongo_pipebuilder.egg-info → mongo_pipebuilder-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mongo-pipebuilder
-Version: 0.2.1
+Version: 0.2.2
 Summary: Type-safe, fluent MongoDB aggregation pipeline builder
 Author-email: seligoroff <seligoroff@gmail.com>
 License: MIT
@@ -373,3 +373,4 @@ See [DEVELOPMENT.md](DEVELOPMENT.md) for development guidelines.
 
 MIT License - see [LICENSE](LICENSE) file for details.
 
+

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/README.md

@@ -345,3 +345,4 @@ See [DEVELOPMENT.md](DEVELOPMENT.md) for development guidelines.
 
 MIT License - see [LICENSE](LICENSE) file for details.
 
+

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mongo-pipebuilder"
-version = "0.2.1"
+version = "0.2.2"
 description = "Type-safe, fluent MongoDB aggregation pipeline builder"
 readme = "README.md"
 requires-python = ">=3.8"

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/src/mongo_pipebuilder/__init__.py

@@ -9,6 +9,6 @@ Author: seligoroff
 
 from mongo_pipebuilder.builder import PipelineBuilder
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 __all__ = ["PipelineBuilder"]
 

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/src/mongo_pipebuilder/builder.py

@@ -158,36 +158,47 @@ class PipelineBuilder:
             self._stages.append({"$project": fields})
         return self
 
-    def group(self, group_by: Dict[str, Any], accumulators: Dict[str, Any]) -> Self:
+    def group(self, group_by: Union[str, Dict[str, Any], Any], accumulators: Dict[str, Any]) -> Self:
         """
         Add a $group stage for grouping documents.
         
         Args:
-            group_by: Expression for grouping (becomes _id)
+            group_by: Expression for grouping (becomes _id). Can be:
+                     - A string (field path, e.g., "$category")
+                     - A dict (composite key, e.g., {"category": "$category"})
+                     - Any other value (null, number, etc.)
             accumulators: Dictionary with accumulators (sum, avg, count, etc.)
             
         Returns:
             Self for method chaining
             
         Raises:
-            TypeError: If arguments are not dictionaries
-            ValueError: If both group_by and accumulators are empty
+            TypeError: If accumulators is not a dictionary
+            ValueError: If both group_by and accumulators are empty (when group_by is dict/str)
             
         Example:
             >>> builder.group(
-            ...     group_by={"category": "$category"},
+            ...     group_by="$category",  # String field path
+            ...     accumulators={"total": {"$sum": "$amount"}}
+            ... )
+            >>> builder.group(
+            ...     group_by={"category": "$category"},  # Composite key
             ...     accumulators={"total": {"$sum": "$amount"}}
             ... )
         """
-        if not isinstance(group_by, dict):
-            raise TypeError(f"group_by must be a dict, got {type(group_by)}")
         if not isinstance(accumulators, dict):
             raise TypeError(f"accumulators must be a dict, got {type(accumulators)}")
         
-        # Empty group_by is technically valid in MongoDB (groups all into one document)
-        # But if both are empty, it's likely an error
-        if not group_by and not accumulators:
-            raise ValueError("group_by and accumulators cannot both be empty")
+        # Validate empty cases
+        # group_by can be None, empty string, empty dict, etc. - all are valid in MongoDB
+        # But if it's a string and empty, or dict and empty, and accumulators is also empty,
+        # it's likely an error
+        if isinstance(group_by, dict):
+            if not group_by and not accumulators:
+                raise ValueError("group_by and accumulators cannot both be empty")
+        elif isinstance(group_by, str):
+            if not group_by and not accumulators:
+                raise ValueError("group_by and accumulators cannot both be empty")
         
         group_stage = {"_id": group_by, **accumulators}
         self._stages.append({"$group": group_stage})

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2/src/mongo_pipebuilder.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mongo-pipebuilder
-Version: 0.2.1
+Version: 0.2.2
 Summary: Type-safe, fluent MongoDB aggregation pipeline builder
 Author-email: seligoroff <seligoroff@gmail.com>
 License: MIT
@@ -373,3 +373,4 @@ See [DEVELOPMENT.md](DEVELOPMENT.md) for development guidelines.
 
 MIT License - see [LICENSE](LICENSE) file for details.
 
+

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/tests/test_builder.py

@@ -78,7 +78,7 @@ class TestPipelineBuilder:
         assert pipeline == [{"$project": {"name": 1, "email": 1, "_id": 0}}]
 
     def test_group_stage(self):
-        """Test adding $group stage."""
+        """Test adding $group stage with dict."""
         builder = PipelineBuilder()
         pipeline = builder.group(
             group_by={"category": "$category"},
@@ -92,6 +92,21 @@ class TestPipelineBuilder:
             }
         }]
 
+    def test_group_stage_with_string(self):
+        """Test adding $group stage with string field path."""
+        builder = PipelineBuilder()
+        pipeline = builder.group(
+            group_by="$categoryType",
+            accumulators={"total": {"$sum": "$amount"}}
+        ).build()
+        
+        assert pipeline == [{
+            "$group": {
+                "_id": "$categoryType",
+                "total": {"$sum": "$amount"}
+            }
+        }]
+
     def test_unwind_stage(self):
         """Test adding $unwind stage."""
         builder = PipelineBuilder()

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/tests/test_builder_validation.py

@@ -171,3 +171,4 @@ class TestPipelineValidation:
         assert "$sort" in pipeline[5]
         assert "$limit" in pipeline[6]
 
+

{mongo_pipebuilder-0.2.1 → mongo_pipebuilder-0.2.2}/tests/test_builder_validation_existing.py

@@ -72,11 +72,18 @@ class TestGroupValidation:
         pipeline = builder.group({}, {"count": {"$sum": 1}}).build()
         assert pipeline == [{"$group": {"_id": {}, "count": {"$sum": 1}}}]
 
-    def test_group_invalid_group_by_type_raises_error(self):
-        """Test that group(123, {}) raises TypeError."""
+    def test_group_with_string_group_by(self):
+        """Test that group() accepts string for group_by (field path)."""
         builder = PipelineBuilder()
-        with pytest.raises(TypeError, match="group_by must be a dict"):
-            builder.group(123, {})
+        # String field path is valid in MongoDB
+        pipeline = builder.group("$categoryType", {"total": {"$sum": "$amount"}}).build()
+        assert pipeline == [{"$group": {"_id": "$categoryType", "total": {"$sum": "$amount"}}}]
+        
+    def test_group_empty_string_with_empty_accumulators_raises_error(self):
+        """Test that group('', {}) raises ValueError."""
+        builder = PipelineBuilder()
+        with pytest.raises(ValueError, match="group_by and accumulators cannot both be empty"):
+            builder.group("", {})
 
     def test_group_invalid_accumulators_type_raises_error(self):
         """Test that group({}, 123) raises TypeError."""