field_id text,
value text
);
+ create index i_data on data (id);
create table fields(
id integer primary key,
name text
);
+ create index i_fields on fields (id);
+
+ create table relation(
+ data_id integer,
+ struct_id integer
+ );
+ create index i_relation_data on relation(data_id);
create table struct(
id integer primary key,
name text,
parent integer
);
-
- commit;
-
- create index i_fields on fields (id);
- create index i_data on data (id);
- create index i_struct on struct(parent);
-
+ create index i_struct_id on struct(parent);
commit;
"""
if not os.path.exists(self._path):
self.new()
else:
- self.conn = sqlite3.connect(self._path)
+ self.connect()
+
+ def connect(self):
+ """Connects to database."""
+
+ self.conn = sqlite3.connect(self._path, isolation_level="EXCLUSIVE")
def new(self):
"""Creates new databse."""
- self.conn = sqlite3.connect(self._path)
+ self.connect()
self.conn.executescript(SCHEMA)
- self.conn.commit()
def close(self):
"""Closes connection with database."""
- self.conn.commit()
self.conn.close()
def save(self):
execute("DELETE from data")
execute("DELETE from fields")
execute("DELETE from struct")
+ execute("DELETE from relation")
self.conn.commit()
def get_fields(self):
execute = self.conn.execute
- if not 'uid' in entry or not 'o' in entry:
- return
-
try:
_id = execute("SELECT MAX(id) FROM data").fetchone()[0] \
+ 1
for field, values in entry.items():
# update FIELDS table
- if execute("SELECT name FROM fields WHERE name=?", \
- (field,)).fetchone() is None:
+ field_id = execute("SELECT id FROM fields WHERE name=?", \
+ (field,)).fetchone()
+ if field_id is None:
execute("INSERT INTO fields values(NULL, ?)", (field,))
+ field_id = execute("SELECT last_insert_rowid()").fetchone()[0]
+ else:
+ field_id = field_id[0]
# update DATA table
for value in values:
- execute("INSERT INTO data values(?,?,?)", (_id, field, value))
+ execute("INSERT INTO data values(?,?,?)", \
+ (_id, field_id, value))
# update STRUCT table
name = entry[TOPLEVEL_FIELD][0]
- row_id = execute("SELECT id FROM struct WHERE name=? ", \
+ parent_id = execute("SELECT id FROM struct WHERE name=? ", \
(name,)).fetchone()
- if row_id is None: # update STRUCT table (TOPLEVEL_FIELD)
+ if parent_id is None: # update STRUCT table (TOPLEVEL_FIELD)
execute("INSERT INTO struct values(NULL,?,0)", (name,))
- row_id = execute("SELECT last_insert_rowid()").fetchone()[0]
+ parent_id = execute("SELECT last_insert_rowid()").fetchone()[0]
else:
- row_id = row_id[0]
+ parent_id = parent_id[0]
name = entry[MIDDLELEVEL_FIELD][0]
- sql_res = execute("SELECT id FROM struct WHERE name=? AND parent=?", \
- (name, row_id)).fetchone()
- if sql_res is None: # update STRUCT table (MIDDLELEVEL_FIELD)
- execute("INSERT INTO struct values(NULL,?,?)", (name, row_id))
-
- def get_entry(self, value):
- """Gets full entry imformation from DATA table."""
-
- _id = self.conn.execute("SELECT id FROM data WHERE value=?", \
- (value,)).fetchone()[0]
- return dict([(key, value) for key, value in self.conn.execute("\
- SELECT field_id, value FROM data WHERE id=?", (_id,)).fetchall()])
-
- def get_toplevel_entries(self):
- """Returns list of all toplevel entries from DATA table."""
+ child_id = execute("SELECT id FROM struct WHERE name=? AND parent=?", \
+ (name, parent_id)).fetchone()
+ if child_id is None: # update STRUCT table (MIDDLELEVEL_FIELD)
+ execute("INSERT INTO struct values(NULL,?,?)", (name, parent_id))
+ child_id = execute("SELECT last_insert_rowid()").fetchone()[0]
+ else:
+ child_id = child_id[0]
- return [item[0] for item in self.conn.execute("SELECT name \
- FROM struct WHERE parent=0 ORDER BY name ASC").fetchall()]
+ # update RELATION table
+ execute("INSERT INTO relation values(?,?)", (_id, child_id))
- def get_middlelevel_entries(self, parent=None):
- """Returns list of all middlelevel entries from DATA table."""
+ def get_folders(self, parent=None):
+ """
+ Returns list of all folders (items with folders and files)
+ from STRUCT table.
+ """
- if not parent:
- return [item[0] for item in self.conn.execute("SELECT DISTINCT \
- name FROM struct WHERE parent!=0 ORDER BY name ASC").fetchall()]
+ if parent is None: # return all folders on level2
+ return self.conn.execute("SELECT DISTINCT name, id FROM struct \
+ WHERE parent!=0 ORDER BY name ASC").fetchall()
else:
- return [item[0] for item in self.conn.execute("SELECT name FROM \
- struct WHERE parent IN (SELECT id FROM struct WHERE name=?) \
- ORDER BY name ASC", (parent,)).fetchall()]
-
- def get_lowlevel_entries(self, parent=None):
- """Returns list of all lowlevel entries from DATA table."""
+ return self.conn.execute("SELECT DISTINCT name, id FROM struct \
+ WHERE parent=? ORDER BY name ASC", (parent,)).fetchall()
- if not parent:
- return [item[0] for item in self.conn.execute("SELECT DISTINCT \
- value FROM data WHERE field_id=? ORDER BY value ASC", \
- (LOWLEVEL_FIELD,)).fetchall()]
- else:
- print parent
- return [item[0] for item in self.conn.execute("SELECT value FROM \
- data WHERE id IN (SELECT id FROM data WHERE value=?) AND field_id='cn' \
- ORDER BY value ASC", (parent, )).fetchall()]
+ def get_files(self, fields, parent=0):
+ """Returns list of all files from DATA table."""
- def get_entry_by_id(self, _id):
+ 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()]
+ 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
+ return items_dict
+
+ def get_entry(self, _id):
"""Returns full entry by it id."""
- sql_res = self.conn.execute("""SELECT id_field, value FROM data
- WHERE id=?""", (_id,)).fetchall()
- return dict([(self.get_field_name_by_id(field_id), value) for field_id, \
- value in sql_res])
-
- def get_all_entries(self):
- """Returns list of all entries."""
-
- ids = [item[0] for item in self.conn.execute("""SELECT DISTINCT id \
- FROM data""").fetchall()]
- return [self.get_entry_by_id(_id) for _id in ids]
-
-
-
-
-if __name__ == "__main__":
- db = SQLite('/tmp/')
- #db.update_field('test')
- #print db.get_field_id_by_name('fax')
- #print db.get_entry_by_id(1)
- #print db.get_all_entries()
- db.close()
-
-
+ 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
+ return entry_dict
projects / meabook / blobdiff