From 905e9f1ed7667f9b23fe1980ecf6d5d23fd08200 Mon Sep 17 00:00:00 2001 From: Max Usachev Date: Tue, 8 Jun 2010 15:43:00 +0300 Subject: [PATCH] added update_filed function --- database/SQLite.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/database/SQLite.py b/database/SQLite.py index 6d67aa4..656a331 100644 --- a/database/SQLite.py +++ b/database/SQLite.py @@ -19,7 +19,7 @@ class SQLite: self.conn = sqlite3.connect(self._path) self.conn.execute("""CREATE TABLE data (user_id int, field_id int, \ value str)""") - self.conn.execute("""CREATE TABLE field (field_id int, name str)""") + self.conn.execute("""CREATE TABLE field (id int primary key, name str)""") self.conn.execute("""CREATE TABLE relation (data_id int, \ struct_id int)""") self.conn.execute("""CREATE TABLE struct (id int, name str, \ @@ -32,9 +32,18 @@ class SQLite: self.conn.commit() self.conn.close() + def update_field(self, fname): + """Adds new field to database.""" + + fields = self.conn.execute("""SELECT name from field""").fetchall() + if not fname in fields: + self.conn.execute("""INSERT INTO field values((SELECT max(id) + FROM field)+1, ?)""", (fname,)) + if __name__ == "__main__": db = SQLite('/tmp/') + db.update_field('test') db.close() -- 1.7.9.5