動態解析和構建TinyDB查詢

Question

是否可以在TinyDB中動態構建查詢？它的邏輯查詢操作是這樣的：

>>> from tinydb import TinyDB, where 
>>> db = TinyDB('db.json') 
>>> # Logical AND: 
>>> db.search((where('int') == 1) & (where('char') == 'b')) 
[{'int': 1, 'char': 'b'}] 

但我需要從用戶的輸入條件動態構建查詢。我能想出的唯一辦法是連接的條件爲一個字符串和exec這樣說：

#!/usr/bin/env python3 
import shlex 
from tinydb import TinyDB, where 

# create db sample 
db = TinyDB('test.json') 
db.insert({'id': '1', 'name': 'Tom', 'age': '10', 'grade': '4'}) 
db.insert({'id': '2', 'name': 'Alice', 'age': '9', 'grade': '3'}) 
db.insert({'id': '3', 'name': 'John', 'age': '11', 'grade': '5'}) 
db.close() 

# query test 
db = TinyDB('test.json') 
q = input("query for name/age/grade: ") 
# name='Tom' grade='4' 
qdict = dict(token.split('=') for token in shlex.split(q)) 

result = [] 
query = "result = db.search(" 
qlen = len(qdict) 
count = 0 
for key, value in qdict.items(): 
    query += "(where('%s') == '%s')" % (key, value) 
    count += 1 
    if count < qlen: 
     query += " & " 

query += ')' 
exec(query) 
print(result) 
# [{'age': '10', 'id': '1', 'grade': '4', 'name': 'Tom'}] 

是否還有更好的和優雅的方式來做到這一點？非常感謝。

Answer 1

這裏是一個最小解決方案，支持以下運算符：

==，!=，>=，<-，>，<

查詢的語法是：

<key> <operator> <value> 

你必須將每個令牌分隔一個溫泉CE。

代碼：

#!/usr/bin/env python3 


from __future__ import print_function 


try: 
    import readline # noqa 
except ImportError: 
    print("Warning: No readline support available!") 


try: 
    input = raw_input 
except NameError: 
    pass 


import sys 
from os import path 
from operator import eq, ge, gt, le, lt, ne 


from tinydb import TinyDB, where 


ops = { 
    "==": eq, 
    "!=": ne, 
    "<=": le, 
    ">=": ge, 
    "<": lt, 
    ">": gt, 
} 


def isint(s): 
    return all(map(str.isdigit, s)) 


def isfloat(s): 
    return "." in s and isint(s.replace(".", "")) 


def createdb(filename): 
    db = TinyDB(filename) 
    db.insert({"id": 1, "name": "Tom", "age": 10, "grade": 4}) 
    db.insert({"id": 2, "name": "Alice", "age": 9, "grade": 3}) 
    db.insert({"id": 3, "name": "John", "age": 11, "grade": 5}) 
    db.close() 


def opendb(filename): 
    return TinyDB(filename) 


def parse_query(s): 
    qs = [] 

    tokens = s.split("&") 
    tokens = map(str.strip, tokens) 

    for token in tokens: 
     try: 
      k, op, v = token.split(" ", 3) 
     except Exception as e: 
      print("Syntax Error with {0:s}: {1:s}".format(repr(s), e)) 
      return where(None) 

     opf = ops.get(op, None) 
     if opf is None: 
      print("Unknown operator: {0:s}".format(op)) 
      return where(None) 

     if isfloat(v): 
      v = float(v) 
     elif isint(v): 
      v = int(v) 

     qs.append(opf(where(k), v)) 

    return reduce(lambda a, b: a & b, qs) 


def main(): 
    if not path.exists(sys.argv[1]): 
     createdb(sys.argv[1]) 

    db = opendb(sys.argv[1]) 

    while True: 
     try: 
      s = input("Query: ") 
      q = parse_query(s) 
      print(repr(db.search(q))) 
     except (EOFError, KeyboardInterrupt): 
      break 

    db.close() 


if __name__ == "__main__": 
    main() 

演示：

$ python foo.py test.json 
Query: name == Tom 
[{u'grade': 4, u'age': 10, u'id': 1, u'name': u'Tom'}] 
Query: grade >= 3 
[{u'grade': 4, u'age': 10, u'id': 1, u'name': u'Tom'}, {u'grade': 3, u'age': 9, u'id': 2, u'name': u'Alice'}, {u'grade': 5, u'age': 11, u'id': 3, u'name': u'John'}] 
Query: grade == 3 
[{u'grade': 3, u'age': 9, u'id': 2, u'name': u'Alice'}] 
Query: age <= 13 
[{u'grade': 4, u'age': 10, u'id': 1, u'name': u'Tom'}, {u'grade': 3, u'age': 9, u'id': 2, u'name': u'Alice'}, {u'grade': 5, u'age': 11, u'id': 3, u'name': u'John'}] 
Query: 

---

注：

• 我只測試了這個關於Python 2.7
• 我用最近tinydb庫
• 我改變你的「測試數據」，包括「真實」數據類型

但最重要的;這不以任何方式使用eval()或exec和嘗試解析輸入並建立查詢對象。