2026-02-16

extensions/rainys_extensions/atomic_data_manager/ops/utils/clean.py

@@ -26,61 +26,131 @@ import bpy
 from ...stats import unused
 
 
-def collections():
+def collections(cached_list=None):
     # removes all unused collections from the project
-    for collection_key in unused.collections_deep():
-        bpy.data.collections.remove(bpy.data.collections[collection_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        collection_keys = cached_list
+    else:
+        collection_keys = unused.collections_deep()
+    
+    for collection_key in collection_keys:
+        if collection_key in bpy.data.collections:
+            bpy.data.collections.remove(bpy.data.collections[collection_key])
 
 
-def images():
+def images(cached_list=None):
     # removes all unused images from the project
-    for image_key in unused.images_deep():
-        bpy.data.images.remove(bpy.data.images[image_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        image_keys = cached_list
+    else:
+        image_keys = unused.images_deep()
+    
+    for image_key in image_keys:
+        if image_key in bpy.data.images:
+            bpy.data.images.remove(bpy.data.images[image_key])
 
 
-def lights():
+def lights(cached_list=None):
     # removes all unused lights from the project
-    for light_key in unused.lights_deep():
-        bpy.data.lights.remove(bpy.data.lights[light_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        light_keys = cached_list
+    else:
+        light_keys = unused.lights_deep()
+    
+    for light_key in light_keys:
+        if light_key in bpy.data.lights:
+            bpy.data.lights.remove(bpy.data.lights[light_key])
 
 
-def materials():
+def materials(cached_list=None):
     # removes all unused materials from the project
-    for light_key in unused.materials_deep():
-        bpy.data.materials.remove(bpy.data.materials[light_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        material_keys = cached_list
+    else:
+        material_keys = unused.materials_deep()
+    
+    for material_key in material_keys:
+        if material_key in bpy.data.materials:
+            bpy.data.materials.remove(bpy.data.materials[material_key])
 
 
-def node_groups():
+def node_groups(cached_list=None):
     # removes all unused node groups from the project
-    for node_group_key in unused.node_groups_deep():
-        bpy.data.node_groups.remove(bpy.data.node_groups[node_group_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        node_group_keys = cached_list
+    else:
+        node_group_keys = unused.node_groups_deep()
+    
+    for node_group_key in node_group_keys:
+        if node_group_key in bpy.data.node_groups:
+            bpy.data.node_groups.remove(bpy.data.node_groups[node_group_key])
 
 
-def particles():
+def particles(cached_list=None):
     # removes all unused particle systems from the project
-    for particle_key in unused.particles_deep():
-        bpy.data.particles.remove(bpy.data.particles[particle_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        particle_keys = cached_list
+    else:
+        particle_keys = unused.particles_deep()
+    
+    for particle_key in particle_keys:
+        if particle_key in bpy.data.particles:
+            bpy.data.particles.remove(bpy.data.particles[particle_key])
 
 
-def textures():
+def textures(cached_list=None):
     # removes all unused textures from the project
-    for texture_key in unused.textures_deep():
-        bpy.data.textures.remove(bpy.data.textures[texture_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        texture_keys = cached_list
+    else:
+        texture_keys = unused.textures_deep()
+    
+    for texture_key in texture_keys:
+        if texture_key in bpy.data.textures:
+            bpy.data.textures.remove(bpy.data.textures[texture_key])
 
 
-def worlds():
+def worlds(cached_list=None):
     # removes all unused worlds from the project
-    for world_key in unused.worlds():
-        bpy.data.worlds.remove(bpy.data.worlds[world_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        world_keys = cached_list
+    else:
+        world_keys = unused.worlds()
+    
+    for world_key in world_keys:
+        if world_key in bpy.data.worlds:
+            bpy.data.worlds.remove(bpy.data.worlds[world_key])
 
 
-def objects():
+def objects(cached_list=None):
     # removes all unused objects from the project
-    for object_key in unused.objects_deep():
-        bpy.data.objects.remove(bpy.data.objects[object_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        object_keys = cached_list
+    else:
+        object_keys = unused.objects_deep()
+    
+    for object_key in object_keys:
+        if object_key in bpy.data.objects:
+            bpy.data.objects.remove(bpy.data.objects[object_key])
 
 
-def armatures():
+def armatures(cached_list=None):
     # removes all unused armatures from the project
-    for armature_key in unused.armatures_deep():
-        bpy.data.armatures.remove(bpy.data.armatures[armature_key])
+    # If cached_list is provided, use it instead of recalculating
+    if cached_list is not None:
+        armature_keys = cached_list
+    else:
+        armature_keys = unused.armatures_deep()
+    
+    for armature_key in armature_keys:
+        if armature_key in bpy.data.armatures:
+            bpy.data.armatures.remove(bpy.data.armatures[armature_key])