added english localization
[meabook] / database / SQLite.py
index 0b2447e..66b25a1 100644 (file)
@@ -136,7 +136,6 @@ class SQLite:
         # update RELATION table
         execute("INSERT INTO relation values(?,?)", (_id, child_id))
 
-    
     def get_folders(self, parent=None):
         """
         Returns list of all folders (items with folders and files)
@@ -149,59 +148,37 @@ class SQLite:
         else:
             return self.conn.execute("SELECT DISTINCT name, id FROM struct \
                 WHERE parent=? ORDER BY name ASC", (parent,)).fetchall()
-    
-    def _get_files(self, fields, parent=0):
-        
+
+    def get_files(self, fields, parent=0):
+        """Finds all files with selected parent"""
+
         items_dict = {}
         execute = self.conn.execute
         fields = dict(execute("SELECT id, name FROM fields WHERE name IN (%s)" \
             % ','.join('%r' % (field,) for field in fields)).fetchall())
-        """ 
+        """
         query = "SELECT id, field_id, value FROM data \
             left join relation on relation.data_id=data.id \
             where relation.struct_id=? and field_id in (%s)"
-        """ 
-        query = "select id, field_id, value from data where id in (select data_id from relation where struct_id=?) and field_id in (%s)"
-        
-        data = execute(query % ','.join('%r' % f for f in fields.keys()), (parent,)).fetchall()
-                
-        for id, field_id, value in data:
-            if not items_dict.has_key(id):
-                items_dict[id] = {}
-            items_dict[id][fields[int(field_id)]] = value
-        return items_dict
-
-
-    
-    def get_files(self, fields, parent=0, pattern=''):
-        """Returns list of all files from DATA table."""
-
-        items_dict = {}
-        execute = self.conn.execute
-        fields = execute("SELECT id, name FROM fields WHERE name IN (%s)" \
-            % ','.join('%r' % (field,) for field in fields)).fetchall()
-        if parent == 0: # get all files
-            #data_ids = [_id[0] for _id in execute("SELECT DISTINCT id FROM \
-            #    data").fetchall()]
-            data_ids = [_id[0] for _id in execute("SELECT DISTINCT id FROM \
-                data WHERE value LIKE '%s%%' LIMIT 50" % pattern.capitalize()).fetchall()]
-        else: # get files for selected parent
-            data_ids = [_id[0] for _id in execute("SELECT data_id FROM \
-                relation WHERE struct_id=?", (parent,)).fetchall()]
-        for data_id in data_ids:
-            items_dict[data_id] = {}
-            for field_id, field_name in fields:
-                field_value = execute("SELECT value FROM data WHERE id=? \
-                    AND field_id=?", (data_id, field_id)).fetchone()[0]
-                items_dict[data_id][field_name] = field_value
+        """
+        query = "SELECT id, field_id, value FROM data \
+            WHERE id IN (SELECT data_id FROM relation WHERE struct_id=?) \
+            AND field_id IN (%s)"
+        data = execute(query % ','.join('%r' % f for f in fields.keys()), \
+            (parent,)).fetchall()
+        for data_id, field_id, value in data:
+            if not items_dict.has_key(data_id):
+                items_dict[data_id] = {}
+            items_dict[data_id][fields[int(field_id)]] = value
         return items_dict
 
-    def _get_files_by_pattern(self, fields, key='cn', pattern='', search_from_start=False):
+    def get_files_by_pattern(self, fields, key='cn', pattern='', \
+        search_from_start=False):
         """Returns list of all files from DATA table."""
 
         items_dict = {}
         if key not in fields:
-            fields = fields + (key,)
+            fields.append(str(key))
         execute = self.conn.execute
         fields = dict(execute("SELECT id, name FROM fields WHERE name IN (%s)" \
             % ','.join('%r' % (field,) for field in fields)).fetchall())
@@ -214,60 +191,33 @@ class SQLite:
             s = '%s%%' % pattern.capitalize()
         else:
             s = '%%%s%%' % pattern
-
-        data = execute("SELECT id, field_id, value FROM data WHERE id in \
-            (select id from data where value LIKE '%s' and field_id=? LIMIT 50) \
-            and field_id in (%s)" % (s, ','.join('%r' % f for f in fields.keys())), (key_id,)).fetchall()
-        
-        for id, field_id, value in data:
-            if not items_dict.has_key(id):
-                items_dict[id] = {}
-            items_dict[id][fields[int(field_id)]] = value
+        data = execute("SELECT id, field_id, value FROM data WHERE id IN \
+            (SELECT id FROM data WHERE value LIKE '%s' AND field_id=? LIMIT \
+            50) AND field_id in (%s)" % (s, ','.join('%r' % f for f in \
+            fields.keys())), (key_id,)).fetchall()
+        for data_id, field_id, value in data:
+            if not items_dict.has_key(data_id):
+                items_dict[data_id] = {}
+            items_dict[data_id][fields[int(field_id)]] = value
         return items_dict
 
-
-    def get_files_by_pattern(self, pattern):
-        """Returns list of all files, matched pattern string."""
-
-        if not pattern:
-            return []
-        execute = self.conn.execute
-        field_id = execute("SELECT id FROM fields WHERE name=?", \
-            ('cn',)).fetchone()[0]
-        return execute("SELECT id, value FROM data WHERE value LIKE '%s%%' \
-            AND field_id=? LIMIT 50" % pattern.capitalize(), (field_id,)).fetchall()
-
-    def _get_entry(self, _id):
+    def get_entry(self, _id):
         """Returns full entry by it id."""
 
         execute = self.conn.execute
         entry_dict = {}
-        entry_dict = dict(execute("select fields.name, value from data \
-            left join fields on fields.id=data.field_id \
-            where data.id=?", (_id,)).fetchall())
-        sql_res = execute('select photo from photo where data_id=?', \
+        entry_dict = dict(execute("SELECT fields.name, value FROM data \
+            LEFT JOIN fields ON fields.id=data.field_id \
+            WHERE data.id=?", (_id,)).fetchall())
+        sql_res = execute('SELECT photo FROM photo WHERE data_id=?', \
             (_id,)).fetchone()
         if sql_res is not None:
             entry_dict['image'] = sql_res[0]
         return entry_dict
 
-    def get_entry(self, _id):
-        """Returns full entry by it id."""
 
-        execute = self.conn.execute
-        entry_dict = {}
-        for field_id, value in execute("SELECT field_id, value FROM data WHERE \
-            id=?", (_id,)).fetchall():
-            field_name = execute("SELECT name FROM fields WHERE id=?", \
-                (field_id,)).fetchone()[0]
-            entry_dict[field_name] = value
-        sql_res = execute('select photo from photo where data_id=?', \
-            (_id,)).fetchone()
-        if sql_res is not None:
-            entry_dict['image'] = sql_res[0]
-        return entry_dict
 
 if __name__ == "__main__":
     db = SQLite('/tmp')
-    #print db.get_files(('cn', 'o', 'ou'), 0)
-    print db._get_files_by_pattern(('cn', 'o', 'ou'), 'cn', 'Ра', True)
+    # put test code here
+    db.close()