mongo-pipebuilder 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

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/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.dist-info → mongo_pipebuilder-0.2.2.dist-info}/METADATA

@@ -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.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+mongo_pipebuilder/__init__.py,sha256=VF9G-Gp0kabZMoEmeQnSnSqHT3hLqxlEHY-FBeXfnVA,336
+mongo_pipebuilder/builder.py,sha256=qvyQd1k9YbaIV0tRixQnsNr7yuE-_SSGN5tBSwDAk5E,27199
+mongo_pipebuilder-0.2.2.dist-info/licenses/LICENSE,sha256=xAHmf48PmIziXYIdaJzRYeYpXFUPIb70SsSPhAHdggY,1089
+mongo_pipebuilder-0.2.2.dist-info/METADATA,sha256=QZJAl4akVEiT9ucIaviwAwJuvTWFL9XyBEyKb_LI6jc,9127
+mongo_pipebuilder-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mongo_pipebuilder-0.2.2.dist-info/top_level.txt,sha256=wLn7H_v-qaNIws5FeBbKPZBCmYFYgFEhPaLjoCWcisc,18
+mongo_pipebuilder-0.2.2.dist-info/RECORD,,

{mongo_pipebuilder-0.2.1.dist-info → mongo_pipebuilder-0.2.2.dist-info}/licenses/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.dist-info/RECORD

@@ -1,7 +0,0 @@
-mongo_pipebuilder/__init__.py,sha256=eOj_NuBMA9YbVFPmm13UV25RR5_QK2ctrx8QXV3yGTU,336
-mongo_pipebuilder/builder.py,sha256=kurxcQ5IXErefUQjqQ5XAzkfZEv9siN-PufmWoFA0aE,26545
-mongo_pipebuilder-0.2.1.dist-info/licenses/LICENSE,sha256=GLx_6hrvLsyIL34dpRYvjCSIXyYD8PzhBR09opTrixI,1088
-mongo_pipebuilder-0.2.1.dist-info/METADATA,sha256=Kjav6gd0U9MREeLM3tcLz2b5RfBlCyjnJP0H9gG97XM,9126
-mongo_pipebuilder-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mongo_pipebuilder-0.2.1.dist-info/top_level.txt,sha256=wLn7H_v-qaNIws5FeBbKPZBCmYFYgFEhPaLjoCWcisc,18
-mongo_pipebuilder-0.2.1.dist-info/RECORD,,