add database.get_version()

Author: mikez

tests/test_things.py

@@ -98,6 +98,11 @@ def test_areas(self):
         test_areas = things.areas(**FILEPATH)
         self.assertEqual(1, len(test_areas))
 
+    def test_database_version(self):
+        """Test database version."""
+        version = things.Database(**FILEPATH).get_version()
+        self.assertEqual(18, version)
+
 
 if __name__ == "__main__":
     unittest.main()

things/database.py

@@ -64,6 +64,7 @@ class Database:
     TABLE_TASKTAG = "TMTaskTag"
     TABLE_AREATAG = "TMAreaTag"
     TABLE_CHECKLIST_ITEM = "TMChecklistItem"
+    TABLE_META = "Meta"
     DATE_CREATE = "creationDate"
     DATE_MOD = "userModificationDate"
     DATE_DUE = "dueDate"
@@ -378,10 +379,20 @@ def get_tags_of_area(self, area_uuid):
                     AREA_TAG.areas = ?
                 ORDER BY AREA."index"
                 """
+
         return self.execute_query(
             query, parameters=(area_uuid,), row_factory=list_factory
         )
 
+    def get_version(self):
+        """Get Things Database version."""
+        import plistlib
+
+        query = f"SELECT value FROM {self.TABLE_META} WHERE key = 'databaseVersion'"
+        result = self.execute_query(query, row_factory=list_factory)
+        plist_bytes = result[0].encode()
+        return plistlib.loads(plist_bytes)
+
     def get_count(self, sql):
         sql = f"""SELECT COUNT(uuid) FROM ({sql})"""
         return self.execute_query(sql, row_factory=list_factory)[0]