file_db: support default data if file not present

2015-01-23 18:59:56 +08:00 · 2015-01-23 18:59:56 +08:00 · f983fdc985
parent 9fb42e4952
commit f983fdc985
1 changed files with 9 additions and 2 deletions
--- a/artiq/protocols/file_db.py
+++ b/artiq/protocols/file_db.py
@ -5,9 +5,16 @@ from artiq.protocols.sync_struct import Notifier


 class FlatFileDB:
-    def __init__(self, filename):
+    def __init__(self, filename, default_data=None):
        self.filename = filename
-        self.data = Notifier(pyon.load_file(self.filename))
+        try:
+            data = pyon.load_file(self.filename)
+        except FileNotFoundError:
+            if default_data is None:
+                raise
+            else:
+                data = default_data
+        self.data = Notifier(data)
        self.hooks = []

    def save(self):