def test_no_index(client):
client.ft().create_index((
TextField("field"),
TextField("text", no_index=True, sortable=True),
NumericField("numeric", no_index=True, sortable=True),
GeoField("geo", no_index=True, sortable=True),
TagField("tag", no_index=True, sortable=True),
))
client.ft().add_document("doc1",
field="aaa",
text="1",
numeric="1",
geo="1,1",
tag="1")
client.ft().add_document("doc2",
field="aab",
text="2",
numeric="2",
geo="2,2",
tag="2")
waitForIndex(client, "idx")
res = client.ft().search(Query("@text:aa*"))
assert 0 == res.total
res = client.ft().search(Query("@field:aa*"))
assert 2 == res.total
res = client.ft().search(Query("*").sort_by("text", asc=False))
assert 2 == res.total
assert "doc2" == res.docs[0].id
res = client.ft().search(Query("*").sort_by("text", asc=True))
assert "doc1" == res.docs[0].id
res = client.ft().search(Query("*").sort_by("numeric", asc=True))
assert "doc1" == res.docs[0].id
res = client.ft().search(Query("*").sort_by("geo", asc=True))
assert "doc1" == res.docs[0].id
res = client.ft().search(Query("*").sort_by("tag", asc=True))
assert "doc1" == res.docs[0].id
# Ensure exception is raised for non-indexable, non-sortable fields
with pytest.raises(Exception):
TextField("name", no_index=True, sortable=False)
with pytest.raises(Exception):
NumericField("name", no_index=True, sortable=False)
with pytest.raises(Exception):
GeoField("name", no_index=True, sortable=False)
with pytest.raises(Exception):
TagField("name", no_index=True, sortable=False)
def test_tags(client):
client.ft().create_index((TextField("txt"), TagField("tags")))
tags = "foo,foo bar,hello;world"
tags2 = "soba,ramen"
client.ft().add_document("doc1", txt="fooz barz", tags=tags)
client.ft().add_document("doc2", txt="noodles", tags=tags2)
waitForIndex(client, "idx")
q = Query("@tags:{foo}")
res = client.ft().search(q)
assert 1 == res.total
q = Query("@tags:{foo bar}")
res = client.ft().search(q)
assert 1 == res.total
q = Query("@tags:{foo\\ bar}")
res = client.ft().search(q)
assert 1 == res.total
q = Query("@tags:{hello\\;world}")
res = client.ft().search(q)
assert 1 == res.total
q2 = client.ft().tagvals("tags")
assert (tags.split(",") + tags2.split(",")).sort() == q2.sort()
def test_json_with_multipath(client):
"""
Create definition with IndexType.JSON as index type (ON JSON),
and use json client to test it.
"""
definition = IndexDefinition(prefix=["king:"], index_type=IndexType.JSON)
client.ft().create_index((TagField("$..name", as_name="name")),
definition=definition)
client.json().set("king:1", Path.rootPath(), {
"name": "henry",
"country": {
"name": "england"
}
})
res = client.ft().search("@name:{henry}")
assert res.docs[0].id == "king:1"
assert res.docs[0].json == '{"name":"henry","country":{"name":"england"}}'
assert res.total == 1
res = client.ft().search("@name:{england}")
assert res.docs[0].id == "king:1"
assert res.docs[0].json == '{"name":"henry","country":{"name":"england"}}'
assert res.total == 1
def ftadd_beers(r, rsclient):
# create beer index
ftidxfields = [
TextField('name', weight=5.0),
TextField('brewery'),
NumericField('breweryid', sortable=True),
TextField('category'),
NumericField('categoryid'),
TextField('style'),
NumericField('styleid'),
TextField('description'),
NumericField('abv', sortable=True),
NumericField('ibu', sortable=True),
TagField('favorite')
]
rsclient.create_index([*ftidxfields])
header = []
dontadd = 0
with open(beerfile) as csvfile:
beers = csv.reader(csvfile)
for row in beers:
docid = ''
docscore = 1.0
ftaddfields = {}
if beers.line_num == 1:
header = row
continue
for idx, field in enumerate(row):
if idx == 0:
docid = "{}:{}".format(beer, field)
continue
# idx 1 is brewery name
if idx == 1:
if field == "":
# something is wrong with the csv, skip this line.
print("\tEJECTING: {}".format(row))
dontadd = 1
break
bkey = "{}:{}".format(brewery, field)
ftaddfields['brewery'] = r.hget(bkey, 'name')
ftaddfields['breweryid'] = field
# idx 2 is beer name
elif idx == 2:
ftaddfields['name'] = field
# idx 3 is category ID
elif idx == 3:
catname = 'None'
if int(field) != -1:
# get the category key and hget the name of the category
ckey = "{}:{}".format(category, field)
catname = r.hget(ckey, 'cat_name')
ftaddfields['category'] = catname
ftaddfields['categoryid'] = field
# idx 4 is style ID
elif idx == 4:
stylename = 'None'
if int(field) != -1:
skey = "{}:{}".format(style, field)
stylename = r.hget(skey, 'style_name')
ftaddfields['style'] = stylename
ftaddfields['styleid'] = field
# idx 5 is ABV
elif idx == 5:
ftaddfields['abv'] = field
# update the document score based on ABV
docscore = get_beer_doc_score(field)
# idx 6 is IBU
elif idx == 6:
ftaddfields['ibu'] = field
if dontadd:
dontadd = 0
continue
# add beer document
rsclient.add_document(docid, score=docscore, **ftaddfields)