Skip to content

Commit 1c8796b

Browse files
committed
extend query service example with parameters
1 parent 3735084 commit 1c8796b

File tree

1 file changed

+36
-7
lines changed

1 file changed

+36
-7
lines changed

examples/query-service/basic_example.py

Lines changed: 36 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def main():
1818

1919
print("=" * 50)
2020
print("DELETE TABLE IF EXISTS")
21-
pool.execute_with_retries("drop table if exists example")
21+
pool.execute_with_retries("DROP TABLE IF EXISTS example")
2222

2323
print("=" * 50)
2424
print("CREATE TABLE")
@@ -82,22 +82,51 @@ def callee(session):
8282

8383
pool.retry_operation_sync(callee)
8484

85-
query_print = """
86-
select $a
87-
"""
88-
8985
def callee(session: ydb.QuerySessionSync):
86+
query_print = """select $a"""
87+
9088
print("=" * 50)
91-
print("Check typed parameters")
89+
print("Check implicit typed parameters")
9290

93-
values = [1, 1.0, True, "text"]
91+
values = [
92+
1,
93+
1.0,
94+
True,
95+
"text",
96+
{"4": 8, "15": 16, "23": 42},
97+
[{"name": "Michael"}, {"surname": "Scott"}],
98+
]
9499

95100
for value in values:
96101
print(f"value: {value}")
97102
with session.transaction().execute(query=query_print, parameters={"$a": value}, commit_tx=True) as results:
98103
for result_set in results:
99104
print(f"rows: {str(result_set.rows)}")
100105

106+
print("=" * 50)
107+
print("Check typed parameters as tuple pair")
108+
109+
typed_value = ([1, 2, 3], ydb.ListType(ydb.PrimitiveType.Int64))
110+
print(f"value: {typed_value}")
111+
112+
with session.transaction().execute(
113+
query=query_print, parameters={"$a": typed_value}, commit_tx=True,
114+
) as results:
115+
for result_set in results:
116+
print(f"rows: {str(result_set.rows)}")
117+
118+
print("=" * 50)
119+
print("Check typed parameters as ydb.TypedValue")
120+
121+
typed_value = ydb.TypedValue(111, ydb.PrimitiveType.Int64)
122+
print(f"value: {typed_value}")
123+
124+
with session.transaction().execute(
125+
query=query_print, parameters={"$a": typed_value}, commit_tx=True,
126+
) as results:
127+
for result_set in results:
128+
print(f"rows: {str(result_set.rows)}")
129+
101130
pool.retry_operation_sync(callee)
102131

103132

0 commit comments

Comments
 (0)