Query match where a field is in a list

eshober

I'd like a query to match if the field value is in a list of values. I can't figure out the syntax for this. The field is not a list.

ie. Given elements:
{'name': 'Fred', 'status': 'attached', 'age': 32}
{'name': 'George', 'status': 'loner', 'age': 45}
{'name': 'Anna', 'status': 'married', 'age': 22}

I'd like a query to match a list of ages, something like this:
db.search(Query().age.any([x for x in range(25, 35)])

But it seems this only works for fields that are lists? Is that right?
Is there some other way to do this?

Thanks,
Ed

eshober

eshober By the way, the result I want from the desired query above is:
[{'name': 'Fred', 'status': 'attached', 'age': 32}]

msiemens

Hey eshober

you're right, the Query().any(value) query is for checking a list of values if it contains the specified item. I think you could use a custom test function like this:

def in_range(value, low, high):
    return low <= value <= high

db.search(Query().age.test(in_range, 25, 35))

Does that work for you?

Cheers!

eshober

Thanks for your response but that won't work for me. This example is just a toy example. I really need to check for membership in a list like this:

def in_list(value, list):
return value in list

db.search(Query().test_id.test(in_list, [2, 4, 251])

but when I try this I get "TypeError: unhashable type: 'list'"

msiemens

Hmm... That's another not really well-documented thing: the arguments have to be hashable. So you could pass a tuple instead of a list like this:

def in_list(value, l):
    return value in l

db.search(Query().test_id.test(in_list, tuple([2, 4, 251]))

That should work ?

eshober

Yes, that works fine, thank you! TinyDB has really worked well for me in a current project and this solution completes all of my requirements.

Thanks again!

msiemens

That's nice to hear! Thank you ?