def test_filter(self):
evens = query.filter_(lambda a: query.equals(query.modulo(a, 2), 0), [1, 2, 3, 4])
self.assertEqual(self._q(evens), [2, 4])
# Works on page too
ref = self._create(n=11, m=12)["ref"]
self._create(n=11)
self._create(n=11)
page = query.paginate(query.match(self.n_index_ref, 11))
refs_with_m = query.filter_(lambda a: query.contains_path(["data", "m"], query.get(a)), page)
self.assertEqual(self._q(refs_with_m), {"data": [ref]})
def read_customers_between(client, min_cust_id, max_cust_id):
#
# Extending the previous example to show getting a range between two values.
#
res = client.query(
q.map_(
lambda x: q.select("data", q.get(q.select(1, x))),
q.filter_(
lambda y: q.lte(min_cust_id, q.select(0, y)),
q.paginate(q.match(q.index("customer_id_filter")),
before=[max_cust_id]))))
print('Query for id\'s > {0} and < {1} : {2}'.format(
min_cust_id, max_cust_id, res))
def unsubscribe_from_anime(self, anime_doc_id: str):
try:
anime = client.query(
q.get(q.ref(q.collection(animes), anime_doc_id)))
client.query(
q.let(
{
'anime_ref':
q.ref(q.collection(animes), anime_doc_id),
'bot_user':
q.ref(q.collection(users), self.chat_id),
'followers':
q.select(['data', 'followers'],
q.get(q.var('anime_ref'))),
},
q.do(
q.update(
q.var('anime_ref'), {
'data': {
'followers': q.subtract(
q.var('followers'), 1)
}
}),
q.update(
q.var('bot_user'), {
'data': {
'animes_watching':
q.filter_(
q.lambda_(
'watched_anime_ref',
q.not_(
q.equals(
q.var('watched_anime_ref'),
q.var('anime_ref')))),
q.select(['data', 'animes_watching'],
q.get(q.var('bot_user'))))
}
}),
q.if_(q.equals(q.var('followers'), 1),
q.delete(q.var('anime_ref')), 'successful!'))))
updater.bot.send_message(chat_id=self.chat_id,
text='You have stopped following ' +
anime['data']['title'])
except errors.NotFound:
logger.info(
'Somehow, a user {0} almost unsubscribed from an anime that did not exist'
.format(self.chat_id))
except Exception as err:
log_error(err)
def get_converted_documents(self) -> Iterator[Document]:
results = self.client.query(
q.filter_(
q.lambda_(
['original_name', 'converted', 'ref'],
q.var('converted')
),
q.paginate(q.match(q.index('documents')))
)
)
documents = []
for data in results['data']:
documents.append(
Document(
original_name=data[0],
converted=data[1],
ref=data[2]
)
)
return documents
def wordPartsGenerator(word):
return q.let(
{
"indexes":
q.map_(
# Reduce this array if you want less ngrams per word.
# Setting it to [ 0 ] would only create the word itself, Setting it to [0, 1] would result in the word itself
# and all ngrams that are one character shorter, etc..
lambda index: q.subtract(q.length(word), index),
maxNgrams),
"indexesFiltered":
q.filter_(
# left min parts length 3
lambda l: q.gte(l, 3),
q.var('indexes')),
"ngramsArray":
q.distinct(
q.union(
q.map_(lambda l: q.ngram(q.lowercase(word), l, l),
q.var('indexesFiltered'))))
},
q.var('ngramsArray'))
def test_filter_expr(self):
self.assertJson(
query.filter_(lambda a: a, [True, False, True]),
'{"collection":[true,false,true],"filter":{"expr":{"var":"a"},"lambda":"a"}}'
)
def test_typecheckfns(self):
coll = query.collection("typecheck_coll")
db = query.database("typecheck_db")
fn = query.function("typecheck_fn")
index = query.index("typecheck_index")
self.admin_client.query(query.create_collection({"name": "typecheck_coll"}))
self.admin_client.query(query.create_index(
{"name": "typecheck_index", "source": coll, "active": True}))
doc = self.admin_client.query(query.create(
coll, {"data": {}, "credentials": {"password": "******"}}))
self.admin_client.query(query.create_database({"name": "typecheck_db"}))
function = self._q(query.create_function(
{"name": "typecheck_fn", "body": query.query(query.lambda_("x", query.now()))}))
key = self.admin_client.query(
query.create_key({"database": db, "role": "admin"}))
token = self._q(query.login(doc["ref"], {"password": "******"}))
credentials = self._q(query.select(['data', 0], query.paginate(query.credentials())))
role = self.admin_client.query(query.create_role(
{"name": "typecheck_role", "membership": [], "privileges": []}))
values = [
None,
bytearray([12,3,4,5]),
credentials,
90,
3.14,
True,
query.to_date(query.now()),
query.date("1970-01-01"),
query.now(),
query.epoch(1, "second"),
query.time("1970-01-01T00:00:00Z"),
{"x": 10},
query.get(doc["ref"]),
query.paginate(query.collections()),
[1, 2, 3],
"a string",
coll,
query.collections(),
query.match(index),
query.union(query.match(index)),
doc["ref"],
query.get(doc["ref"]),
index,
db,
coll,
token["ref"],
role["ref"],
key["ref"],
function["ref"],
query.get(function["ref"]),
query.query(query.lambda_("x", query.var("x"))),
]
pairs = [
["array", query.is_array],
["object", query.is_object],
["string", query.is_string],
["null", query.is_null],
["number", query.is_number],
["bytes", query.is_bytes],
["date", query.is_date],
["timestamp", query.is_timestamp],
["set", query.is_set],
["ref", query.is_ref],
["boolean", query.is_boolean],
["double", query.is_double],
["integer", query.is_integer],
["database", query.is_database],
["index", query.is_index],
["collection", query.is_collection],
["token", query.is_token],
["function", query.is_function],
["collection", query.is_collection],
["role", query.is_role],
["credentials", query.is_credentials],
["key", query.is_key],
]
expected = {
"array": 1,
"boolean": 1,
"bytes": 1,
"collection": 3,
"credentials": 1,
"database": 1,
"date": 2,
"double": 1,
"function": 2,
"integer": 1,
"index": 1,
"key": 1,
"null": 1,
"number": 2,
"object": 5,
"ref": 11,
"role": 1,
"set": 3,
"string": 1,
"timestamp": 3,
"token": 1,
}
q = []
for p in pairs:
d = dict()
d[p[0]] = query.count(query.filter_(query.lambda_("v", p[1](query.var("v"))), query.var("vals")))
q.append(d)
actual = self._q(query.let({"vals": values}, query.merge({}, q)))
self.assertEqual(actual, expected)
