From 2eed7551b95cebb82515574102952759b8dd81d9 Mon Sep 17 00:00:00 2001 From: James Vasile Date: Fri, 2 Nov 2012 14:38:05 -0400 Subject: make sure storage table exists --- withsqlite.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'withsqlite.py') diff --git a/withsqlite.py b/withsqlite.py index d2d224f..63e8494 100644 --- a/withsqlite.py +++ b/withsqlite.py @@ -117,6 +117,9 @@ True if not os.path.exists(self.fname): self.make_db() self.conn = sqlite3.connect(self.fname) + c = self.conn.cursor() + c.execute('''create table if not exists store (key text unique, val text)''') + self.conn.commit() self.crsr = self.conn.cursor() return self def __exit__(self, type, value, traceback): @@ -125,9 +128,12 @@ True def make_db(self): conn = sqlite3.connect(self.fname) c = conn.cursor() - c.execute('''create table store (key text unique, val text)''') + c.execute('''create table if not exists store (key text unique, val text)''') conn.commit() c.close() + def commit(self): + """This should rarely be necessary.""" + self.conn.commit() def __delitem__(self, key): """del a[k] remove a[k] from a""" self.crsr.execute("delete from store where key=?", [key]) -- cgit v1.2.3