git.maemo.org Git - meabook/blob - database/SQLite.py

   1 import os
   2 import sqlite3
   3
   4 DATABASE_NAME = 'contacts.db'
   5
   6 TOPLEVEL_FIELD = 'o'
   7 MIDDLELEVEL_FIELD = 'ou'
   8 LOWLEVEL_FIELD = 'cn'
   9
  10 SCHEMA = """
  11     begin;
  12
  13     create table data(
  14         id integer,
  15         field_id text,
  16         value text
  17     );
  18
  19     create table fields(
  20         id integer primary key,
  21         name text
  22     );
  23
  24     create table relation(
  25         data_id integer,
  26         struct_id integer
  27     );
  28
  29     create table struct(
  30         id integer primary key,
  31         name text,
  32         parent integer
  33     );
  34
  35     commit;
  36
  37     create index i_fields on fields (id);
  38     create index i_data on data (id);
  39     create index i_struct on struct(parent);
  40     create index i_relation on relation(data_id);
  41
  42     commit;
  43
  44 """
  45
  46
  47 class SQLite:
  48     def __init__(self, basedir):
  49         self._path = os.path.join(basedir, DATABASE_NAME)
  50         self.conn = None
  51         if not os.path.exists(self._path):
  52             self.new()
  53         else:
  54             self.conn = sqlite3.connect(self._path)
  55
  56     def new(self):
  57         """Creates new databse."""
  58
  59         self.conn = sqlite3.connect(self._path)
  60         self.conn.executescript(SCHEMA)
  61         self.conn.commit()
  62
  63     def close(self):
  64         """Closes connection with database."""
  65
  66         self.conn.commit()
  67         self.conn.close()
  68
  69     def save(self):
  70         """Save all changes."""
  71
  72         self.conn.commit()
  73
  74     def clear(self):
  75         """Clear all database tables."""
  76
  77         execute = self.conn.execute
  78         execute("DELETE from data")
  79         execute("DELETE from fields")
  80         execute("DELETE from struct")
  81         execute("DELETE from relation")
  82         self.conn.commit()
  83
  84     def get_fields(self):
  85         """Returns all fields from FIELDS table."""
  86
  87         return [item[0] for item in self.conn.execute( \
  88             "SELECT name FROM fields").fetchall()]
  89
  90     # operations with DATA table
  91     def add_entry(self, entry):
  92         """Adds new entry to database."""
  93
  94         execute = self.conn.execute
  95
  96         if not 'uid' in entry or not 'o' in entry:
  97             return
  98
  99         try:
 100             _id = execute("SELECT MAX(id) FROM data").fetchone()[0] \
 101                 + 1
 102         except TypeError:
 103             _id = 1
 104
 105         for field, values in entry.items():
 106             # update FIELDS table
 107             field_id = execute("SELECT id, name FROM fields WHERE name=?", \
 108                 (field,)).fetchone()
 109             if field_id is None:
 110                 execute("INSERT INTO fields values(NULL, ?)", (field,))
 111                 field_id = execute("SELECT last_insert_rowid()").fetchone()[0]
 112             else:
 113                 field_id = field_id[0]
 114
 115             # update DATA table
 116             for value in values:
 117                 execute("INSERT INTO data values(?,?,?)", (_id, field_id, value))
 118
 119         # update STRUCT table
 120         name = entry[TOPLEVEL_FIELD][0]
 121         parent_id = execute("SELECT id FROM struct WHERE name=? ", \
 122             (name,)).fetchone()
 123         if parent_id is None: # update STRUCT table (TOPLEVEL_FIELD)
 124             execute("INSERT INTO struct values(NULL,?,0)", (name,))
 125             parent_id = execute("SELECT last_insert_rowid()").fetchone()[0]
 126         else:
 127             parent_id = parent_id[0]
 128
 129         name = entry[MIDDLELEVEL_FIELD][0]
 130         child_id = execute("SELECT id FROM struct WHERE name=? AND parent=?", \
 131             (name, parent_id)).fetchone()
 132         if child_id is None: # update STRUCT table (MIDDLELEVEL_FIELD)
 133             execute("INSERT INTO struct values(NULL,?,?)", (name, parent_id))
 134             child_id = execute("SELECT last_insert_rowid()").fetchone()[0]
 135         else:
 136             child_id = child_id[0]
 137
 138         # update RELATION table
 139         execute("INSERT INTO relation values(?,?)", (_id, child_id))
 140
 141     def get_entry(self, value):
 142         """Gets full entry imformation from DATA table."""
 143
 144         _id = self.conn.execute("SELECT id FROM data WHERE value=?", \
 145             (value,)).fetchone()[0]
 146         return dict([(key, value) for key, value in self.conn.execute("\
 147             SELECT field_id, value FROM data WHERE id=?", (_id,)).fetchall()])
 148
 149     def get_folders(self, parent=None):
 150         """
 151         Returns list of all folders (items with folders and files)
 152         from STRUCT table.
 153         """
 154
 155         if parent is None:  # return all folders on level2
 156             return self.conn.execute("SELECT DISTINCT name, id FROM struct \
 157                 WHERE parent!=0 ORDER BY name ASC").fetchall()
 158         else:
 159             return self.conn.execute("SELECT DISTINCT name, id FROM struct \
 160                 WHERE parent=? ORDER BY name ASC", (parent,)).fetchall()
 161
 162     def get_files(self, fields, parent=0):
 163         """Returns list of all files from DATA table."""
 164
 165         items_dict = {}
 166         fields_string = ','.join('%r' % (field,) for field in fields)
 167         if parent == 0: # get all files
 168             sql_res = self.conn.execute("select data.id as id, fields.name, \
 169             data.value as name from data left join fields on data.field_id=\
 170             fields.id left join relation on data.id=relation.data_id where \
 171             fields.name in (%s)" % fields_string)
 172         else: # get files for selected parent
 173             sql_res = self.conn.execute("select data.id as id, fields.name, \
 174             data.value as name from data left join fields on data.field_id=\
 175             fields.id left join relation on data.id=relation.data_id where \
 176             fields.name in (%s) and relation.struct_id=?" % fields_string, \
 177             (parent,))
 178         if sql_res is None:
 179             return {}
 180         for _id, fname, fvalue in sql_res.fetchall():
 181             if not items_dict.has_key(_id):
 182                 items_dict[_id] = {}
 183             items_dict[_id][fname] = fvalue
 184         return items_dict
 185
 186     def get_entry_by_id(self, _id):
 187         """Returns full entry by it id."""
 188
 189         sql_res = self.conn.execute("SELECT fields.name, data.value AS name \
 190         FROM data LEFT JOIN fields ON data.field_id=fields.id LEFT JOIN \
 191         relation ON data.id=relation.data_id WHERE data.id=?", (_id,))
 192         if sql_res is None:
 193             return {}
 194         else:
 195             return dict(sql_res.fetchall())
 196
 197     def get_all_entries(self):
 198         """Returns list of all entries."""
 199
 200         ids = [item[0] for item in self.conn.execute("""SELECT DISTINCT id \
 201             FROM data""").fetchall()]
 202         return [self.get_entry_by_id(_id) for _id in ids]
 203
 204
 205
 206
 207 if __name__ == "__main__":
 208     db = SQLite('/tmp/')
 209     #db.update_field('test')
 210     #print db.get_field_id_by_name('fax')
 211     #print db.get_entry_by_id(1)
 212     #print db.get_all_entries()
 213     db.close()
 214
 215