Bläddra i källkod

Fix bug when database is not present

theenglishway (time) 6 år sedan
förälder
incheckning
51df24a341
1 ändrade filer med 9 tillägg och 6 borttagningar
  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: