Skip to content

Commit a62c454

Browse files
authored
Improve Spanner delete_data sample coverage [(#3922)](GoogleCloudPlatform/python-docs-samples#3922)
* improve Spanner delete_data sample coverage * specify end to stay consistent with other languages Co-authored-by: larkee <larkee@users.noreply.github.com>
1 parent 2bec1d2 commit a62c454

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

samples/samples/snippets.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,14 +106,23 @@ def delete_data(instance_id, database_id):
106106
instance = spanner_client.instance(instance_id)
107107
database = instance.database(database_id)
108108

109-
singers_to_delete = spanner.KeySet(
110-
keys=[[1], [2], [3], [4], [5]])
109+
# Delete individual rows
111110
albums_to_delete = spanner.KeySet(
112-
keys=[[1, 1], [1, 2], [2, 1], [2, 2], [2, 3]])
111+
keys=[[2, 1], [2, 3]])
112+
113+
# Delete a range of rows where the column key is >=3 and <5
114+
singers_range = spanner.KeyRange(start_closed=[3], end_open=[5])
115+
singers_to_delete = spanner.KeySet(
116+
ranges=[singers_range])
117+
118+
# Delete remaining Singers rows, which will also delete the remaining
119+
# Albums rows because Albums was defined with ON DELETE CASCADE
120+
remaining_singers = spanner.KeySet(all_=True)
113121

114122
with database.batch() as batch:
115123
batch.delete('Albums', albums_to_delete)
116124
batch.delete('Singers', singers_to_delete)
125+
batch.delete('Singers', remaining_singers)
117126

118127
print('Deleted data.')
119128
# [END spanner_delete_data]

0 commit comments

Comments
 (0)