quizzer: Add methods to upgrade data from v0.1 to v0.2

[quizzer.git] / quizzer / quiz.py

diff --git a/quizzer/quiz.py b/quizzer/quiz.py
index 4a36f86c9856f4995a104d2fe1c9d6a9c1def514..ae1702eed427ff7e46d5738f4d2c903629f433d0 100644 (file)
--- a/quizzer/quiz.py
+++ b/quizzer/quiz.py
@@ -44,8 +44,13 @@ class Quiz (list):
             data = _json.load(f)
         version = data.get('version', None)
         if version != __version__:
-            raise NotImplementedError('upgrade from {} to {}'.format(
-                    version, __version__))
+            try:
+                upgrader = getattr(
+                    self, '_upgrade_from_{}'.format(version.replace('.', '_')))
+            except AttributeError as e:
+                raise NotImplementedError('upgrade from {} to {}'.format(
+                        version, __version__)) from e
+            data = upgrader(data)
         self.copyright = data.get('copyright', None)
         self.introduction = data.get('introduction', None)
         for state in data['questions']:
@@ -86,3 +91,7 @@ class Quiz (list):
             raise KeyError(id)
         raise NotImplementedError(
             'multiple questions with one ID: {}'.format(matches))
+
+    def _upgrade_from_0_1(self, data):
+        data['version'] = __version__
+        return data