Динамический анализ и сборка запросов TinyDB

Можно ли будет динамически создавать запросы в 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'}]

Есть ли лучший и элегантный способ сделать это? Большое спасибо.

Источник

2 ответа

Решение

Вот минимальное решение, которое поддерживает следующие операторы:

==, !=, >=, <-, >, <

Синтаксис запросов:

<key> <operator> <value>

Вы должны разделить каждый токен пробелом.

Код:

#!/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 любым способом и пытается проанализировать входные данные и создать объект запроса.

Источник

Да, похоже, что динамически создавать запросы TinyDB довольно легко. Я сам столкнулся с той же проблемой и нашел это рабочее решение:

      from tinydb import TinyDB, Query

T = TinyDB('storage.tinydb').table('data')
Q = Query()


# creating some sample data..

T.truncate()
T.insert_multiple(
  [
    {'name': 'John', 'age': 22},
    {'name': 'John', 'age': 37},
    {'name': 'John', 'age': 45},
    {'name': 'Mike', 'age': 33},
    {'name': 'Paul', 'age': 40},
  ]
)


# here you can create you query units..

query1 = Q.name == 'John'

min_age = 30
query2 = Q.age >= min_age

person_name = 'Paul'
query3 = Q.name == person_name


# then those query units can be combined as you like..

combined_query = query1 & query2


print( T.search(query1) )

print( T.search(query1 & query2) )

print( T.search(combined_query) )

print( T.search(query1 | query3) )

print( T.search((query1 | query3) & query2) )

Я делаю это, используя TinyDB версии 4.7.1.

Ваше здоровье! :)

Источник
Другие вопросы по тегам