次の演算子をサポートする最小限のソリューションを次に示します。
==
、、、、、、!=
_ >=
_ <-
_ >
_<
クエリの構文は次のとおりです。
<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ライブラリを使用しました
- 「実際の」データ型を含めるように「テスト データ」を変更しました
ただし、最も重要なことです。これはorをまったく使用せず、入力を解析してクエリ オブジェクトを構築しようとします。eval()
exec