-
Notifications
You must be signed in to change notification settings - Fork 2.7k
/
sample_indexers_operations.py
123 lines (98 loc) · 3.79 KB
/
sample_indexers_operations.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
"""
FILE: sample_indexer_operations.py
DESCRIPTION:
This sample demonstrates how to get, create, update, or delete a Indexer.
USAGE:
python sample_indexer_operations.py
Set the environment variables with your own values before running the sample:
1) AZURE_SEARCH_SERVICE_ENDPOINT - the endpoint of your Azure Cognitive Search service
2) AZURE_SEARCH_API_KEY - your search API key
"""
import os
service_endpoint = os.environ["AZURE_SEARCH_SERVICE_ENDPOINT"]
key = os.environ["AZURE_SEARCH_API_KEY"]
connection_string = os.environ["AZURE_STORAGE_CONNECTION_STRING"]
from azure.core.credentials import AzureKeyCredential
from azure.search.documents.indexes.models import (
SearchIndexerDataContainer,
SearchIndexerDataSourceConnection,
SearchIndex,
SearchIndexer,
SimpleField,
SearchFieldDataType,
)
from azure.search.documents.indexes import SearchIndexClient, SearchIndexerClient
indexers_client = SearchIndexerClient(service_endpoint, AzureKeyCredential(key))
def create_indexer():
# create an index
index_name = "indexer-hotels"
fields = [
SimpleField(name="hotelId", type=SearchFieldDataType.String, key=True),
SimpleField(name="baseRate", type=SearchFieldDataType.Double),
]
index = SearchIndex(name=index_name, fields=fields)
ind_client = SearchIndexClient(service_endpoint, AzureKeyCredential(key))
ind_client.create_index(index)
# [START create_indexer]
# create a datasource
container = SearchIndexerDataContainer(name="searchcontainer")
data_source_connection = SearchIndexerDataSourceConnection(
name="indexer-datasource", type="azureblob", connection_string=connection_string, container=container
)
data_source = indexers_client.create_data_source_connection(data_source_connection)
# create an indexer
indexer = SearchIndexer(
name="sample-indexer", data_source_name="indexer-datasource", target_index_name="indexer-hotels"
)
result = indexers_client.create_indexer(indexer)
print("Create new Indexer - sample-indexer")
# [END create_indexer]
def list_indexers():
# [START list_indexer]
result = indexers_client.get_indexers()
names = [x.name for x in result]
print("Found {} Indexers in the service: {}".format(len(result), ", ".join(names)))
# [END list_indexer]
def get_indexer():
# [START get_indexer]
result = indexers_client.get_indexer("sample-indexer")
print("Retrived Indexer 'sample-indexer'")
return result
# [END get_indexer]
def get_indexer_status():
# [START get_indexer_status]
result = indexers_client.get_indexer_status("sample-indexer")
print("Retrived Indexer status for 'sample-indexer'")
return result
# [END get_indexer_status]
def run_indexer():
# [START run_indexer]
result = indexers_client.run_indexer("sample-indexer")
print("Ran the Indexer 'sample-indexer'")
return result
# [END run_indexer]
def reset_indexer():
# [START reset_indexer]
result = indexers_client.reset_indexer("sample-indexer")
print("Reset the Indexer 'sample-indexer'")
return result
# [END reset_indexer]
def delete_indexer():
# [START delete_indexer]
indexers_client.delete_indexer("sample-indexer")
print("Indexer 'sample-indexer' successfully deleted")
# [END delete_indexer]
if __name__ == "__main__":
create_indexer()
list_indexers()
get_indexer()
get_indexer_status()
run_indexer()
reset_indexer()
delete_indexer()