@@ -42,7 +42,6 @@ class Storage(object):
42
42
"""
43
43
44
44
# Public
45
-
46
45
def __init__ (self , es = None ):
47
46
# Use the passed `es` or create a new Elasticsearch instance
48
47
self .__es = es if es is not None else Elasticsearch ()
@@ -55,7 +54,7 @@ def __repr__(self):
55
54
56
55
@property
57
56
def buckets (self ):
58
- indexes = self .__es .indices .get_alias ('*' )
57
+ indexes = self .__es .indices .get_alias (index = '*' )
59
58
for index_name , index in indexes .items ():
60
59
for alias_name in index .get ('aliases' , {}).keys ():
61
60
yield alias_name
@@ -72,15 +71,15 @@ def create_index(self, bucket, index_settings=None):
72
71
body = dict (
73
72
settings = index_settings
74
73
)
75
- self .__es .indices .create (index_name , body = body )
76
- self .__es .indices .put_alias (index_name , bucket )
74
+ self .__es .indices .create (index = index_name , body = body )
75
+ self .__es .indices .put_alias (index = index_name , name = bucket )
77
76
return index_name
78
77
79
78
def put_mapping (self , bucket , descriptor , index_name , mapping_generator_cls ):
80
79
mapping = mappers .descriptor_to_mapping (
81
80
descriptor , mapping_generator_cls = mapping_generator_cls
82
81
)
83
- self .__es .indices .put_mapping (mapping , index = index_name )
82
+ self .__es .indices .put_mapping (index = index_name , ** mapping )
84
83
85
84
def generate_doc_id (self , row , primary_key ):
86
85
return '/' .join ([str (row .get (k )) for k in primary_key ])
@@ -108,7 +107,7 @@ def create(self, bucket, descriptor,
108
107
"""
109
108
existing_index_names = []
110
109
if self .__es .indices .exists_alias (name = bucket ):
111
- existing_index_names = self .__es .indices .get_alias (bucket )
110
+ existing_index_names = self .__es .indices .get_alias (name = bucket )
112
111
existing_index_names = sorted (existing_index_names .keys ())
113
112
114
113
if len (existing_index_names ) == 0 or always_recreate :
@@ -138,11 +137,11 @@ def create(self, bucket, descriptor,
138
137
version_type = 'external'
139
138
)
140
139
)
141
- self .__es .reindex (reindex_body )
140
+ self .__es .reindex (body = reindex_body )
142
141
self .__es .indices .flush ()
143
142
144
143
for existing_index_name in existing_index_names :
145
- self .__es .indices .delete (existing_index_name )
144
+ self .__es .indices .delete (index = existing_index_name )
146
145
147
146
def delete (self , bucket = None ):
148
147
"""Delete index with mapping by schema.
@@ -153,10 +152,10 @@ def delete(self, bucket=None):
153
152
"""
154
153
def internal_delete (bucket ):
155
154
if self .__es .indices .exists_alias (name = bucket ):
156
- existing_index_names = self .__es .indices .get_alias (bucket )
155
+ existing_index_names = self .__es .indices .get_alias (name = bucket )
157
156
existing_index_names = list (existing_index_names .keys ())
158
157
for existing_index_name in existing_index_names :
159
- self .__es .indices .delete (existing_index_name )
158
+ self .__es .indices .delete (index = existing_index_name )
160
159
161
160
if bucket is None :
162
161
for bucket in self .buckets :
@@ -226,4 +225,4 @@ def actions(rows_, primary_key_, update_):
226
225
else :
227
226
collections .deque (iter , maxlen = 0 )
228
227
229
- self .__es .indices .flush (bucket )
228
+ self .__es .indices .flush (index = bucket )
0 commit comments