Various bug fixes and tweaks found through 0, 1, and 2

[gc-dialer] / src / file_backend.py

diff --git a/src/file_backend.py b/src/file_backend.py
index ddd89b3..b373561 100644 (file)
--- a/src/file_backend.py
+++ b/src/file_backend.py
@@ -48,7 +48,12 @@ class CsvAddressBook(object):
 
        @classmethod
        def read_csv(cls, csvPath):
-               csvReader = iter(csv.reader(open(csvPath, "rU")))
+               try:
+                       csvReader = iter(csv.reader(open(csvPath, "rU")))
+               except IOError, e:
+                       if e.errno != 2:
+                               raise
+                       return
 
                header = csvReader.next()
                nameColumn, phoneColumns = cls._guess_columns(header)
@@ -129,13 +134,17 @@ class FilesystemAddressBookFactory(object):
                """
                for root, dirs, filenames in os.walk(self.__path):
                        for filename in filenames:
-                               name, ext = filename.rsplit(".", 1)
+                               try:
+                                       name, ext = filename.rsplit(".", 1)
+                               except ValueError:
+                                       continue
+
                                if ext in self.FILETYPE_SUPPORT:
                                        yield self, os.path.join(root, filename), name
 
        def open_addressbook(self, bookId):
                name, ext = bookId.rsplit(".", 1)
-               assert ext in self.FILETYPE_SUPPORT
+               assert ext in self.FILETYPE_SUPPORT, "Unsupported file extension %s" % ext
                return self.FILETYPE_SUPPORT[ext](bookId)
 
        @staticmethod