Detect supportness.

README.md

@@ -38,7 +38,6 @@
 ## TODOs
 
 - As a Flask plug-in
-- Aware of underlying SQLite FTS query syntax (flag for detection, at least)
 - PyPI (?
 - Benchmark
 - Further tags analytics (e.g. top-k ~ KNN classification)

simpletag.py

@@ -21,12 +21,22 @@ class __CONSTS__(object):
     SQL_STATS = 'SELECT term, documents, occurrences FROM {}_terms WHERE col=0;'
     pass
 
-
 class ns(object):
 
     dbfile = 'simpletag.db'
     table = None
     conn = None
+    using_parenthesis_query = False
+
+    def resolve_supported_level(self):
+        sql = 'PRAGMA compile_options;'
+        csr = self.conn.cursor()
+        opts = [row[0] for row in csr.execute(sql)]
+        if 'ENABLE_FTS3' not in opts:
+            raise RuntimeError('SQLite''s FTS is not enabled')
+        if 'ENABLE_FTS3_PARENTHESIS' in opts:
+            self.using_parenthesis_query = True
+        pass
 
     def get_existing_tbl_type(self, name):
         sql = 'select name from sqlite_master where type = "table";'
@@ -40,6 +50,7 @@ def __priv_init__(self):
         conn = sqlite3.connect(self.dbfile)
         conn.row_factory = sqlite3.Row
         self.conn = conn
+        self.resolve_supported_level()
         pass
 
     def open_table_(self, name):

tests.py

@@ -13,10 +13,6 @@ def setUpClass(cls):
         print '===='
         print 'sqlite ver', sqlite3.sqlite_version
         print '===='
-        conn = sqlite3.connect(':memory:')
-        csr = conn.cursor()
-        cls.SQLITE_COMPILE_OPTS = [row[0] for row in csr.execute('PRAGMA compile_options;')]
-        conn.close()
         pass
 
     def setUp(self):
@@ -69,14 +65,21 @@ def test_update_list_then_query_ids(self):
         self.ns.update(789, [u'中文', u'行不行'])
         self.assertEqual([123, 456], [i for i in self.ns.query_ids('is')])
         self.assertIn(789, [i for i in self.ns.query_ids(u'行不行')])
+
+    def test_update_list_then_query_str_ids(self):
+        self.ns_str.update('/a/b', ['simpletag', 'is', 'awsome'])
+        self.ns_str.update('/', ['test', 'is', 'a', 'MUST'])
+        self.ns_str.update('/b/a', [u'中文', u'行不行'])
+        self.assertEqual(['/a/b', '/'], [i for i in self.ns_str.query_ids('is')])
+        self.assertIn('/b/a', [i for i in self.ns_str.query_ids(u'行不行')])
         pass
 
     def test_set_query(self):
         self.ns.update(123, ['simpletag', 'is', 'awsome'])
         self.ns.update(456, ['test', 'is', 'a', 'MUST'])
 
         query = None
-        if 'ENABLE_FTS3_PARENTHESIS' in self.SQLITE_COMPILE_OPTS:
+        if self.ns.using_parenthesis_query:
             query = 'is NOT awsome'
         else:
             query = 'is -awsome'