From 93049f3908eb1d83d5c6ca3f3faebe146bdcb27e Mon Sep 17 00:00:00 2001
From: kramo <93832451+kra-mo@users.noreply.github.com>
Date: Wed, 20 Dec 2023 09:40:10 +0100
Subject: [PATCH] (Potentially) fix Flatpak source

---
 cartridges/importer/flatpak_source.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/cartridges/importer/flatpak_source.py b/cartridges/importer/flatpak_source.py
index a12bd5b..6848bdf 100644
--- a/cartridges/importer/flatpak_source.py
+++ b/cartridges/importer/flatpak_source.py
@@ -54,10 +54,11 @@ class FlatpakSourceIterable(SourceIterable):
             }
         )
 
-        for entry in chain(
-            (self.source.locations.system_data["applications"]).iterdir(),
-            (self.source.locations.user_data["applications"]).iterdir(),
-        ):
+        if source.locations
+
+        generators = set(location.iterdir() for location in (self.source.locations.user_data["applications"], self.source.locations.system_data["applications"]) if location)
+
+        for entry in chain(*generators):
             if entry.suffix != ".desktop":
                 continue