adds button to show removable disks

vanilla_installer/core/disks.py

@@ -103,6 +103,16 @@ def pretty_size(self):
         else:
             return f"{size} B"
 
+    @property
+    def is_removable(self):
+        if os.path.isfile("/sys/block/" + self.__disk + "/removable"):
+            with open("/sys/block/" + self.__disk + "/removable") as f:
+                removable = int(f.readlines()[0].strip())
+                if removable == 1:
+                    return True
+
+        return False
+
 
 class Partition:
     def __init__(self, disk: str, partition: str):
@@ -217,23 +227,24 @@ def __get_disks(self):
             if disk.startswith(("loop", "ram", "sr", "zram", "dm-")):
                 continue
 
-            if os.path.isfile("/sys/block/" + disk + "/removable"):
-                with open("/sys/block/" + disk + "/removable") as f:
-                    removable = int(f.readlines()[0].strip())
-
-                if removable == 1:
-                    continue
 
             disks.append(Disk(disk))
 
         return disks
 
-    @property
-    def all_disks(self):
-        return self.__disks
+    def all_disks(self, include_removable: bool = True):
+        selected_disks = []
+        if include_removable:
+            selected_disks = self.__disks
+        else:
+            for disk in self.__disks:
+                if not disk.is_removable:
+                    selected_disks.append(disk)
+
+        return selected_disks
 
     def get_disk(self, disk: str):
-        for disk in self.all_disks:
+        for disk in self.all_disks():
             if disk.disk == disk:
                 return disk
 

vanilla_installer/defaults/disk.py

@@ -688,20 +688,33 @@ def __init__(self, window, distro_info, key, step, **kwargs):
         )
 
         # append the disks widgets
-        for disk in self.__disks.all_disks:
+        for disk in self.__disks.all_disks(include_removable=False):
             entry = VanillaDefaultDiskEntry(self, disk)
             self.group_disks.add(entry)
 
             self.__registry_disks.append(entry)
 
+        all_disks_button = Adw.ButtonRow()
+        all_disks_button.set_title(_("Show removable disks"))
+        self.group_disks.add(all_disks_button)
+
         # signals
+        all_disks_button.connect("activated", self.__on_btn_all_disks)
         self.btn_next.connect("clicked", self.__on_btn_next_clicked)
         self.btn_auto.connect("clicked", self.__on_auto_clicked)
         self.btn_manual.connect("clicked", self.__on_manual_clicked)
 
     def get_finals(self):
         return {"disk": self.__partition_recipe}
 
+    def __on_btn_all_disks(self, widget):
+        widget.set_visible(False)
+        for disk in self.__disks.all_disks(include_removable=True):
+            if disk.is_removable:
+                entry = VanillaDefaultDiskEntry(self, disk)
+                self.group_disks.add(entry)
+                self.__registry_disks.append(entry)
+
     def __on_close_default_disk_part_modal(self, *args):
         self.btn_next.set_visible(self.__partition_recipe is not None)
         self.btn_next.set_sensitive(self.__partition_recipe is not None)