Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix pylint
  • Loading branch information
annie-mac committed Oct 2, 2024
commit 41074040f8bac57ad05018299606dc92b1956d6c
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import base64
import json
from abc import ABC, abstractmethod
from typing import Union, List, Dict, Any
from typing import Union, List, Dict, Any, Optional

from azure.cosmos._routing.routing_range import Range
from azure.cosmos.partition_key import _Undefined, _Empty
Expand All @@ -42,7 +42,7 @@ def to_dict(self) -> Dict[str, Any]:
pass

def _to_base64_encoded_string(self) -> str:
data_json = json.dumps(self._to_dict())
data_json = json.dumps(self.to_dict())
json_bytes = data_json.encode('utf-8')
# Encode the bytes to a Base64 string
base64_bytes = base64.b64encode(json_bytes)
Expand Down Expand Up @@ -102,7 +102,7 @@ def __init__(self, feed_range: Range) -> None:
raise ValueError("feed_range cannot be None")

self._range = feed_range
self._base64_encoded_string = None
self._base64_encoded_string: Optional[str] = None

def get_normalized_range(self) -> Range:
return self._range.to_normalized_range()
Expand All @@ -129,4 +129,4 @@ def __str__(self) -> str:
if self._base64_encoded_string is None:
self._base64_encoded_string = self._to_base64_encoded_string()

return self._base64_encoded_string
return self._base64_encoded_string
2 changes: 1 addition & 1 deletion sdk/cosmos/azure-cosmos/azure/cosmos/aio/_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -630,10 +630,10 @@ def query_items_change_feed( # pylint: disable=unused-argument
request. Once the user has reached their provisioned throughput, low priority requests are throttled
before high priority requests start getting throttled. Feature must first be enabled at the account level.
:keyword Callable response_hook: A callable invoked with the response metadata.
:param Any args: args
:returns: An Iterable of items (dicts).
:rtype: Iterable[Dict[str, Any]]
"""

# pylint: disable=too-many-statements
if kwargs.get("priority") is not None:
kwargs['priority'] = kwargs['priority']
Expand Down
1 change: 1 addition & 0 deletions sdk/cosmos/azure-cosmos/azure/cosmos/container.py
Original file line number Diff line number Diff line change
Expand Up @@ -456,6 +456,7 @@ def query_items_change_feed(
request. Once the user has reached their provisioned throughput, low priority requests are throttled
before high priority requests start getting throttled. Feature must first be enabled at the account level.
:keyword Callable response_hook: A callable invoked with the response metadata.
:param Any args: args
:returns: An Iterable of items (dicts).
:rtype: Iterable[Dict[str, Any]]
"""
Expand Down
21 changes: 20 additions & 1 deletion sdk/cosmos/azure-cosmos/samples/examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,4 +255,23 @@
query='SELECT * FROM products p WHERE p.state = "GA"'
):
container.delete_item(item, partition_key=["GA", "Atlanta", 30363])
# [END delete_items]
# [END delete_items]

# Get the feed ranges list from container.
# [START read_feed_ranges]
container.read_feed_ranges()
# [END read_feed_ranges]

# Query a sorted list of items that were changed for one feed range
# [START query_items_change_feed]
feed_ranges = container.read_feed_ranges()
for item in container.query_items_change_feed(feed_range=feed_ranges[0]):
print(json.dumps(item, indent=True))
# [END query_items_change_feed]

# Query a sorted list of items that were changed for one feed range
# [START query_items_change_feed_from_beginning]
feed_ranges = container.read_feed_ranges()
for item in container.query_items_change_feed(feed_range=feed_ranges[0], start_time="Beginning"):
print(json.dumps(item, indent=True))
# [END query_items_change_feed_from_beginning]
23 changes: 23 additions & 0 deletions sdk/cosmos/azure-cosmos/samples/examples_async.py
Original file line number Diff line number Diff line change
Expand Up @@ -263,6 +263,29 @@ async def examples_async():
await container.delete_item(item, partition_key=["GA", "Atlanta", 30363])
# [END delete_items]

# Get the feed ranges list from container.
# [START read_feed_ranges]
await container.read_feed_ranges()
# [END read_feed_ranges]

# Query a sorted list of items that were changed for one feed range.
# The asynchronous client returns asynchronous iterators for its query methods;
# as such, we iterate over it by using an async for loop
# [START query_items_change_feed]
feed_ranges = await container.read_feed_ranges()
async for item in container.query_items_change_feed(feed_range=feed_ranges[0]):
print(json.dumps(item, indent=True))
# [END query_items_change_feed]

# Query a sorted list of items that were changed for one feed range from beginning.
# The asynchronous client returns asynchronous iterators for its query methods;
# as such, we iterate over it by using an async for loop
# [START query_items_change_feed_from_beginning]
feed_ranges = await container.read_feed_ranges()
async for item in container.query_items_change_feed(feed_range=feed_ranges[0], start_time="Beginning"):
print(json.dumps(item, indent=True))
# [END query_items_change_feed_from_beginning]

await client.delete_database(database_name)
print("Sample done running!")

Expand Down