fix: support same options in model.query as query (#407)

cguardia · web-flow · commit ce9945bed606 · 2020-04-28T20:58:23.000-05:00
For maintaining consistency.
diff --git a/packages/google-cloud-ndb/google/cloud/ndb/model.py b/packages/google-cloud-ndb/google/cloud/ndb/model.py
@@ -5265,6 +5265,7 @@ def _prepare_for_put(self):
         projection=None,
         distinct_on=None,
         group_by=None,
+        default_options=None,
     )
     def _query(cls, *filters, **kwargs):
         """Generate a query for this class.
@@ -5290,6 +5291,7 @@ def _query(cls, *filters, **kwargs):
             distinct_on (list[str]): The field names used to group query
                 results.
             group_by (list[str]): Deprecated. Synonym for distinct_on.
+            default_options (QueryOptions): QueryOptions object.
         """
         # Validating distinct
         if kwargs["distinct"]:
@@ -5322,6 +5324,7 @@ def _query(cls, *filters, **kwargs):
             projection=kwargs["projection"],
             distinct_on=kwargs["distinct_on"],
             group_by=kwargs["group_by"],
+            default_options=kwargs["default_options"],
         )
         query = query.filter(*cls._default_filters())
         query = query.filter(*filters)
