Forráskód Böngészése

Fix bug when database is not present

theenglishway (time) 6 éve
szülő
commit
51df24a341
1 módosított fájl, 9 hozzáadás és 6 törlés
  1. 9 6
      planner/database.py

+ 9 - 6
planner/database.py

@@ -17,12 +17,15 @@ class YAMLStorage(Storage):
         self.filename = filename
 
     def read(self):
-        with open(self.filename) as handle:
-            try:
-                data = yaml.safe_load(handle.read())
-                return data
-            except yaml.YAMLError:
-                return None
+        try:
+            with open(self.filename) as handle:
+                try:
+                    data = yaml.safe_load(handle.read())
+                    return data
+                except yaml.YAMLError:
+                    return None
+        except FileNotFoundError:
+            return None
 
     def write(self, data):
         with open(self.filename, 'w') as handle: