Adding the file backend as a choice in the UI

[doneit] / src / file_backend.py

diff --git a/src/file_backend.py b/src/file_backend.py
index 78360c2..954e38e 100644 (file)
--- a/src/file_backend.py
+++ b/src/file_backend.py
@@ -19,14 +19,23 @@ class FileBackend(object):
                        "items": self._items,
                        "locations": self._locations,
                }
-               pickle.dump(state, self._filename)
+               with open(self._filename, "wb") as pickleFile:
+                       pickle.dump(state, pickleFile)
 
        def load(self):
-               state = pickle.load(self._filename)
+               try:
+                       with open(self._filename, "rb") as pickleFile:
+                               state = pickle.load(pickleFile)
+               except IOError, e:
+                       if e.errno != 2:
+                               raise
                self._projects = state["projects"]
                self._items = state["items"]
                self._locations = state["locations"]
 
+               if len(self._projects) == 0:
+                       self.add_project("Inbox")
+
        def add_project(self, name):
                projId = uuid.uuid4().hex
                projDetails = {