Procházet zdrojové kódy

Fix bug when database is not present

theenglishway (time) před 6 roky
rodič
revize
51df24a341
1 změnil soubory, kde provedl 9 přidání a 6 odebrání
  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: