added decorator and getattr

Author: navjotcis

src/class_sqlThread.py

@@ -20,6 +20,17 @@
 # pylint: disable=attribute-defined-outside-init,protected-access
 
 
+def db_upgrade(func):
+    """upgrade the migration"""
+    def inner(*args, **kwargs):
+        reference = args[0]
+        item = '''SELECT value FROM settings WHERE key='version';'''
+        parameters = ''
+        reference.cur.execute(item, parameters)
+        func(reference, int(reference.cur.fetchall()[0][0]))
+    return inner
+
+
 class sqlThread(threading.Thread):
     """A thread for all SQL operations"""
 
@@ -41,6 +52,14 @@ def update_sent(self):
             '''update sent set status='broadcastqueued' where status='broadcastpending'  ''')
         self.conn.commit()
 
+    @db_upgrade
+    def upgrade_migrations(self, currentVersion):
+        versionList = ['versionThree', 'versionTwo', 'versionThree',
+        'versionFour', 'versionFive', 'versionSix', 'versionSeven',
+        'versionEight', 'versionNine', 'versionTen']
+        if currentVersion < 11:
+            getattr(self, versionList[currentVersion - 1])()
+
     def versionTwo(self):
         """
         method for getting rid of the first20bytesofencryptedmessage field in
@@ -393,20 +412,7 @@ def run(self):  # pylint: disable=too-many-locals, too-many-branches, too-many-s
 
         self.update_sent()
 
-        item = '''SELECT value FROM settings WHERE key='version';'''
-        parameters = ''
-        self.cur.execute(item, parameters)
-        currentVersion = int(self.cur.fetchall()[0][0])
-        versionDict = {
-            1: self.versionThree, 2: self.versionTwo,
-            3: self.versionThree, 4: self.versionFour,
-            5: self.versionFive, 6: self.versionSix,
-            7: self.versionSeven, 8: self.versionEight,
-            9: self.versionNine, 10: self.versionTen,
-        }
-
-        if currentVersion in versionDict.keys():
-            versionDict[currentVersion]()
+        self.upgrade_migrations()
 
         # Are you hoping to add a new option to the keys.dat file of existing
         # Bitmessage users or modify the SQLite database? Add it right