Skip to content

Commit

Permalink
google.cloud.datastore add_filter warnings
Browse files Browse the repository at this point in the history
refs googleapis/python-datastore#504

To get rid of:
UserWarning: Detected filter using positional arguments. Prefer using the 'filter' keyword argument instead.
  • Loading branch information
dhimmel committed Mar 13, 2024
1 parent 1fd6a0e commit 5451991
Showing 1 changed file with 16 additions and 5 deletions.
21 changes: 16 additions & 5 deletions articat/catalog_datastore.py
Expand Up @@ -7,6 +7,7 @@

from google.cloud import datastore
from google.cloud.datastore import Client, Entity, Key
from google.cloud.datastore.query import PropertyFilter

from articat.artifact import ID, Artifact, Metadata, Partition, Version, not_supplied
from articat.catalog import Catalog
Expand Down Expand Up @@ -91,16 +92,22 @@ def _lookup(
and partition_dt_start == partition_dt_end
):
query.add_filter(
"partition", "=", convert_to_datetime(partition_dt_start)
filter=PropertyFilter(
"partition", "=", convert_to_datetime(partition_dt_start)
)
)
else:
if partition_dt_start is not None:
query.add_filter(
"partition", ">=", convert_to_datetime(partition_dt_start)
filter=PropertyFilter(
"partition", ">=", convert_to_datetime(partition_dt_start)
)
)
if partition_dt_end is not None:
query.add_filter(
"partition", "<", convert_to_datetime(partition_dt_end)
filter=PropertyFilter(
"partition", "<", convert_to_datetime(partition_dt_end)
)
)
query.order = ["-partition"]
if version is not not_supplied:
Expand All @@ -109,10 +116,14 @@ def _lookup(
if metadata is not None:
if metadata.schema_fields:
for f in metadata.schema_fields:
query.add_filter("metadata.schema_fields", "=", f)
query.add_filter(
filter=PropertyFilter("metadata.schema_fields", "=", f)
)
if metadata.arbitrary is not None:
for k, v in metadata.arbitrary.items():
query.add_filter(f"metadata.arbitrary.{k}", "=", v)
query.add_filter(
filter=PropertyFilter(f"metadata.arbitrary.{k}", "=", v)
)
yield from query.fetch(limit)

@classmethod
Expand Down

0 comments on commit 5451991

Please sign in to comment.