fix: include ExclusionModel in profile export/import (#2471)

Fix #2414 — ExclusionModel data (custom and preset exclusion rules) was missing from profile export/import.

Changes:
- Add ExclusionModel serialization in `ProfileExport.from_db()`
- Add ExclusionModel restoration in `ProfileExport.to_db()` (with backward compat guard for old exports)
- Add test coverage for exclusion round-trip in export/import tests

Co-authored-by: Denay2468 <denayaung13@gamil.com>

Author: Denay2468

src/vorta/profile_export.py

@@ -8,6 +8,7 @@
 from vorta.store.connection import DB, SCHEMA_VERSION, init_db
 from vorta.store.models import (
     BackupProfileModel,
+    ExclusionModel,
     RepoModel,
     SchemaVersion,
     SettingsModel,
@@ -64,6 +65,11 @@ def from_db(cls, profile, store_password=True, include_settings=True):
             model_to_dict(source, recurse=False, exclude=[SourceFileModel.id])
             for source in SourceFileModel.select().where(SourceFileModel.profile == profile)
         ]
+        # Add ExclusionModel
+        profile_dict['ExclusionModel'] = [
+            model_to_dict(exclusion, recurse=False, exclude=[ExclusionModel.id])
+            for exclusion in ExclusionModel.select().where(ExclusionModel.profile == profile)
+        ]
         # Add SchemaVersion
         profile_dict['SchemaVersion'] = model_to_dict(SchemaVersion.get(id=1))
 
@@ -127,18 +133,27 @@ def to_db(self, overwrite_profile=False, overwrite_settings=True):
             SettingsModel.insert_many(self._profile_dict['SettingsModel']).execute()
             WifiSettingModel.insert_many(self._profile_dict['WifiSettingModel']).execute()
 
-        # Set the profile ids to be match new profile
+        # Set the profile ids to match new profile
         for source in self._profile_dict['SourceFileModel']:
             source['profile'] = self.id
         # Delete existing Sources to avoid duplicates
         SourceFileModel.delete().where(SourceFileModel.profile == self.id).execute()
         SourceFileModel.insert_many(self._profile_dict['SourceFileModel']).execute()
 
+        # Restore ExclusionModel entries
+        if 'ExclusionModel' in self._profile_dict:
+            for exclusion in self._profile_dict['ExclusionModel']:
+                exclusion['profile'] = self.id
+            ExclusionModel.delete().where(ExclusionModel.profile == self.id).execute()
+            ExclusionModel.insert_many(self._profile_dict['ExclusionModel']).execute()
+
         # Delete added dictionaries to make it match BackupProfileModel
         del self._profile_dict['SettingsModel']
         del self._profile_dict['SourceFileModel']
         del self._profile_dict['WifiSettingModel']
         del self._profile_dict['SchemaVersion']
+        if 'ExclusionModel' in self._profile_dict:
+            del self._profile_dict['ExclusionModel']
 
         # dict to profile
         new_profile = dict_to_model(BackupProfileModel, self._profile_dict)

tests/unit/profile_exports/valid.json

@@ -52,6 +52,20 @@
             "added_at": "2020-07-03 04:37:05.106150"
         }
     ],
+    "ExclusionModel": [
+        {
+            "name": "*.pyc",
+            "enabled": true,
+            "source": "custom",
+            "profile": 1
+        },
+        {
+            "name": "__pycache__",
+            "enabled": true,
+            "source": "preset",
+            "profile": 1
+        }
+    ],
     "WifiSettingModel": [],
     "SchemaVersion": {
         "id": 1,

tests/unit/test_import_export.py

@@ -6,7 +6,7 @@
 from PyQt6.QtWidgets import QDialogButtonBox, QFileDialog, QMessageBox
 
 from vorta.profile_export import VersionException
-from vorta.store.models import BackupProfileModel, SourceFileModel
+from vorta.store.models import BackupProfileModel, ExclusionModel, SourceFileModel
 from vorta.views.dialogs.profile.import_window import ImportWindow
 
 VALID_IMPORT_FILE = Path(__file__).parent / 'profile_exports' / 'valid.json'
@@ -32,6 +32,7 @@ def test_import_success(qapp, qtbot, rootdir, monkeypatch):
     restored_repo = restored_profile.repo
     assert restored_repo is not None
     assert len(SourceFileModel.select().where(SourceFileModel.profile == restored_profile)) == 3
+    assert len(ExclusionModel.select().where(ExclusionModel.profile == restored_profile)) == 2
 
 
 @pytest.mark.parametrize(
@@ -128,6 +129,12 @@ def getSaveFileName(*args, **kwargs):
 
     assert os.path.isfile(FILE_PATH)
 
+    import json
+
+    with open(FILE_PATH) as f:
+        exported = json.load(f)
+    assert 'ExclusionModel' in exported
+
 
 def test_export_fail_unwritable(qapp, qtbot, monkeypatch):
     FILE_PATH = os.path.join(os.path.abspath(os.sep), "testresult.vortabackup")