completed storage magic

3 files changed, 65 insertions, 3 deletions

diff --git a/storage/factory.py b/storage/factory.py
index c1fa049..024c045 100644
--- a/storage/factory.py
+++ b/storage/factory.py
@@ -1,3 +1,5 @@
+# -*- coding: utf-8 -*-
+
 from sqlite import *
 
 def getStorage():
diff --git a/storage/sqlite.py b/storage/sqlite.py
index 8a24858..be3db7f 100644
--- a/storage/sqlite.py
+++ b/storage/sqlite.py
@@ -1,17 +1,75 @@
+# -*- coding: utf-8 -*-
+
 from storageBase import storageBase
 from itemList import itemList
+from item import item
 from pysqlite2 import dbapi2 as sqliteBackend
 
 class sqlite(storageBase):
+    dbVersion = '0.1'
+
     def __init__(self):
         self.con = sqliteBackend.connect(self.getConfigDir() + '/data.sqlite')
+        self.cur = self.con.cursor()
+
+        # wenn todo tabelle noch nicht exsistiert, dann sollten wir sie anlegen
+        cur = self.cur.execute("select name from sqlite_master where type='table';")
+        tables = cur.fetchall()
+        if ('todo',) not in tables or ('control',) not in tables:
+            self.init_db()
+
+        # überprüfung ob die db-version identisch ist
+        version = self.cur.execute("select value from control where setting='db-version'").fetchone()
+        if (sqlite.dbVersion,) != version:
+            # TODO: update not init
+            self.init_db()
+            
+
+
+    def init_db(self):
+        print '(re)create todo table...'
+        self.cur.execute('drop table if exists todo')
+        self.cur.execute('''create table todo (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                title TEXT(255),
+                createdAt INTEGER,
+                priority INTEGER,
+                desc BLOB
+        )''')
+
+        self.cur.execute('drop table if exists control')
+        self.cur.execute('''create table control (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                setting TEXT,
+                value TEXT
+        )''')
+
+        self.cur.execute("insert into control (setting, value) VALUES ('db-version', ?)", (sqlite.dbVersion,))
+        self.con.commit()
 
     def __del__(self):
         self.con.close()
         
     def load(self):
-        return itemList(self)
+        items = itemList(self)
+        todos = self.cur.execute('select * from todo').fetchall()
+        for todo in todos:
+            items += item(row=todo)
+            
+        return items
 
     def notifyChange(self, sender):
-        print '%s %s' % ('Save changes:', sender.getCreatedAt())
+        if sender.getId() >= 0:
+            print sender.getId()
+            self.cur.execute('update todo set title=?, createdAt=?, priority=? where id=?',
+                             (sender.getTitle(), sender.getCreatedAt(), sender.getPriority(), sender.getId()))
+            self.con.commit()
+        else:      
+            self.cur.execute('insert into todo (title, createdAt, priority) VALUES (?, ?, ?)',
+                             (sender.getTitle(), sender.getCreatedAt(), sender.getPriority()))
+            self.con.commit()
+            sender.setId(self.cur.execute('select last_insert_rowid()').fetchone()[0])
+
+            
+        print '%s %s' % ('Save changes:', sender.getId())
         return
diff --git a/storage/storageBase.py b/storage/storageBase.py
index 24abd2c..064c8a3 100644
--- a/storage/storageBase.py
+++ b/storage/storageBase.py
@@ -1,10 +1,12 @@
+# -*- coding: utf-8 -*-
+
 import os
 
 class storageBase:
     def notifyChange(self, sender): abstract
 
     def getConfigDir(self):
-        dir = os.path.expanduser('~/.todolist/blub')
+        dir = os.path.expanduser('~/.todolist/')
         if not os.path.exists(dir):
             os.makedirs(dir)