TinyDB获取列表查询中的所有ID

时间:2016-12-26 02:09:03

标签: python json tinydb nosql

我在我的python项目中使用TinyDB。但是,为了使逻辑有效,我需要能够在查询中获得多个对象。我的偏好是使用列表

listOfIDs = ['123', '456']

我正在使用最新的Python版本和TinyDB实现。

我有以下实现:

from tinydb import TinyDB, Query
db = TinyDB('db.json')
myDBQuery= Query()
db.insert({'id': '123', 'name': 'bob'})
db.insert({'id': '456', 'name': 'mary'})
result = db.search(myDBQuery.id.all(['123', '456']))
print(result)

但是,我一直得到以下结果:

[]

我知道我的实施正在发挥作用,因为当我执行以下操作时,我得到了预期的结果:

result = db.search((myDBQuery.id == '123') | myDBQuery.id == '456'))

任何人都知道如何查询TinyDB中的列表?

任何人都知道如何查询TinyDB中的列表?

修改

对于那些你想知道的人,目前的实现没有办法做到这一点。我唯一的解决方法是创建类似下面的内容

def getAllObjectsMatchingId(listOfIds):

tempList = []

for idMember in tempList:
    result = db.search(myDBQuery.id == idMember)
    tempList.append(result)

return tempList

3 个答案:

答案 0 :(得分:3)

仅当列表的所有元素与查询

匹配时,

.all才返回true

.any可能会改为

result = db.search(myDBQuery.id.any(['123', '456']))

来自问题OP的编辑:请参阅上面的编辑

答案 1 :(得分:0)

这似乎是我最好的解决方案:

db.search(myDBQuery.id.test(lambda x: x in listOfIDs))

test() Matches any document for which the function returns True

答案 2 :(得分:0)

如果仍然有人遇到这种情况,可以选择Reinaldo的实现方式

listOfIds = ['1','2','3']
q = Query()
db.search(q.id.one_of(listOfIds))

取自Query's API

one_of(items: List[Any]) → tinydb.queries.QueryInstance
    Check if the value is contained in a list or generator.

您还可以否定查询(这正是我所需要的)以获取不在列表中的项目:

listOfIds = ['4','5']
q = Query()
db.search(~q.id.one_of(listOfIds))