@@ -487,7 +487,7 @@ def compound_query_single_clause():
487
487
488
488
489
489
def compound_query_valid_multi_clause ():
490
- db = firestore .Client ()
490
+ db = firestore .Client (add_unique_string = False ) # Flag for testing purposes, needs index to be precreated
491
491
# [START firestore_query_filter_compound_multi_eq]
492
492
cities_ref = db .collection ("cities" )
493
493
@@ -500,6 +500,7 @@ def compound_query_valid_multi_clause():
500
500
# [END firestore_query_filter_compound_multi_eq]
501
501
print (denver_query )
502
502
print (large_us_cities_query )
503
+ return denver_query , large_us_cities_query
503
504
504
505
505
506
def compound_query_valid_single_field ():
@@ -525,8 +526,9 @@ def compound_query_invalid_multi_field():
525
526
def order_simple_limit ():
526
527
db = firestore .Client ()
527
528
# [START firestore_order_simple_limit]
528
- db .collection ("cities" ).order_by ("name" ).limit (3 ).stream ()
529
+ query = db .collection ("cities" ).order_by ("name" ).limit (3 ).stream ()
529
530
# [END firestore_order_simple_limit]
531
+ return query
530
532
531
533
532
534
def order_simple_limit_desc ():
@@ -537,16 +539,18 @@ def order_simple_limit_desc():
537
539
results = query .stream ()
538
540
# [END firestore_query_order_desc_limit]
539
541
print (results )
542
+ return results
540
543
541
544
542
545
def order_multiple ():
543
- db = firestore .Client ()
546
+ db = firestore .Client (add_unique_string = False ) # Flag for testing purposes, needs index to be precreated
544
547
# [START firestore_query_order_multi]
545
548
cities_ref = db .collection ("cities" )
546
- cities_ref .order_by ("state" ).order_by (
549
+ ordered_city_ref = cities_ref .order_by ("state" ).order_by (
547
550
"population" , direction = firestore .Query .DESCENDING
548
551
)
549
552
# [END firestore_query_order_multi]
553
+ return ordered_city_ref
550
554
551
555
552
556
def order_where_limit ():
@@ -561,6 +565,7 @@ def order_where_limit():
561
565
results = query .stream ()
562
566
# [END firestore_query_order_limit_field_valid]
563
567
print (results )
568
+ return results
564
569
565
570
566
571
def order_limit_to_last ():
@@ -571,6 +576,7 @@ def order_limit_to_last():
571
576
results = query .get ()
572
577
# [END firestore_query_order_limit]
573
578
print (results )
579
+ return results
574
580
575
581
576
582
def order_where_valid ():
@@ -583,10 +589,11 @@ def order_where_valid():
583
589
results = query .stream ()
584
590
# [END firestore_query_order_with_filter]
585
591
print (results )
592
+ return results
586
593
587
594
588
595
def order_where_invalid ():
589
- db = firestore .Client ()
596
10000
+ db = firestore .Client (add_unique_string = False ) # Flag for testing purposes, needs index to be precreated
590
597
# [START firestore_query_order_field_invalid]
591
598
cities_ref = db .collection ("cities" )
592
599
query = cities_ref .where (filter = FieldFilter ("population" , ">" , 2500000 )).order_by (
@@ -595,6 +602,7 @@ def order_where_invalid():
595
602
results = query .stream ()
596
603
# [END firestore_query_order_field_invalid]
597
604
print (results )
605
+ return results
598
606
599
607
600
608
def cursor_simple_start_at ():
@@ -628,9 +636,6 @@ def snapshot_cursors():
628
636
)
629
637
# [END firestore_query_cursor_start_at_document]
630
638
results = start_at_snapshot .limit (10 ).stream ()
631
- for doc in results :
632
- print (f"{ doc .id } " )
633
-
634
639
return results
635
640
636
641
@@ -790,7 +795,7 @@ def on_snapshot(col_snapshot, changes, read_time):
790
795
791
796
792
797
def cursor_multiple_conditions ():
793
- db = firestore .Client ()
798
+ db = firestore .Client (add_unique_string = False ) # Flag for testing purposes, needs index to be precreated
794
799
# [START firestore_query_cursor_start_at_field_value_multi]
795
800
start_at_name = (
796
801
db .collection ("cities" ).order_by ("name" ).start_at ({"name" : "Springfield" })
@@ -932,7 +937,8 @@ def update_document_increment(db):
932
937
def list_document_subcollections ():
933
938
db = firestore .Client ()
934
939
# [START firestore_data_get_sub_collections]
935
- collections = db .collection ("cities" ).document ("SF" ).collections ()
940
+ city_ref = db .collection ("cities" ).document ("SF" )
941
+ collections = city_ref .collections ()
936
942
for collection in collections :
937
943
for doc in collection .stream ():
938
944
print (f"{ doc .id } => { doc .to_dict ()} " )
@@ -983,3 +989,4 @@ def regional_endpoint():
983
989
for r in cities_query :
984
990
print (r )
985
991
# [END firestore_regional_endpoint]
992
+ return cities_query
0 commit comments