Import + Delete Zips works now

Author: justo46

__init__.py

@@ -54,7 +54,8 @@
     BSEQ_OT_batch_sequences,
     BSEQ_PT_batch_sequences_settings,
     BSEQ_OT_meshio_object,
-    BSEQ_OT_import_zip
+    BSEQ_OT_import_zip,
+    BSEQ_OT_delete_zips
 ]
 
 def register():
@@ -76,6 +77,7 @@ def register():
 def unregister():
     for cls in classes:
         bpy.utils.unregister_class(cls)
+    bpy.utils.unregister_class(BSEQ_ImportedZip)
     bpy.types.TEXT_MT_templates.remove(draw_template)
     del bpy.types.Scene.BSEQ
     del bpy.types.Object.BSEQ

bseq/__init__.py

@@ -1,6 +1,6 @@
 from bseq.utils import refresh_obj
-from .operators import BSEQ_OT_load, BSEQ_OT_edit, BSEQ_OT_resetpt, BSEQ_OT_resetmesh, BSEQ_OT_resetins, BSEQ_OT_set_as_split_norm, BSEQ_OT_remove_split_norm, BSEQ_OT_disable_selected, BSEQ_OT_enable_selected, BSEQ_OT_refresh_seq, BSEQ_OT_disable_all, BSEQ_OT_enable_all, BSEQ_OT_refresh_sequences, BSEQ_OT_set_start_end_frames, BSEQ_OT_batch_sequences, BSEQ_PT_batch_sequences_settings, BSEQ_OT_meshio_object, BSEQ_OT_import_zip
-from .properties import BSEQ_scene_property, BSEQ_obj_property, BSEQ_mesh_property
+from .operators import BSEQ_OT_load, BSEQ_OT_edit, BSEQ_OT_resetpt, BSEQ_OT_resetmesh, BSEQ_OT_resetins, BSEQ_OT_set_as_split_norm, BSEQ_OT_remove_split_norm, BSEQ_OT_disable_selected, BSEQ_OT_enable_selected, BSEQ_OT_refresh_seq, BSEQ_OT_disable_all, BSEQ_OT_enable_all, BSEQ_OT_refresh_sequences, BSEQ_OT_set_start_end_frames, BSEQ_OT_batch_sequences, BSEQ_PT_batch_sequences_settings, BSEQ_OT_meshio_object, BSEQ_OT_import_zip, BSEQ_OT_delete_zips
+from .properties import BSEQ_scene_property, BSEQ_obj_property, BSEQ_mesh_property, BSEQ_ImportedZip
 from .panels import BSEQ_UL_Obj_List, BSEQ_List_Panel, BSEQ_Settings, BSEQ_PT_Import, BSEQ_PT_Import_Child1, BSEQ_PT_Import_Child2, BSEQ_Globals_Panel, BSEQ_Advanced_Panel, BSEQ_Templates, BSEQ_UL_Att_List, draw_template
 from .messenger import subscribe_to_selected, unsubscribe_to_selected
 import bpy
@@ -55,5 +55,7 @@ def BSEQ_initialize(scene):
     "BSEQ_OT_batch_sequences",
     "BSEQ_PT_batch_sequences_settings",
     "BSEQ_OT_meshio_object",
-    "BSEQ_OT_import_zip"
+    "BSEQ_OT_import_zip",
+    "BSEQ_OT_delete_zips",
+    "BSEQ_ImportedZip"
 ]

bseq/operators.py

@@ -451,9 +451,34 @@ def execute(self, context):
         import zipfile
         zip_file = zipfile.ZipFile(self.filepath)
         zip_file.extractall(Path(self.filepath).parent)
-            
+        zip_file.close()
+
+        folder = str(Path(self.filepath).parent) + '/' + str(Path(self.filepath).name)
+        folder = folder[:-4]
+
+        seqs = fileseq.findSequencesOnDisk(str(folder))
+        for s in seqs:
+            create_obj(s, folder, transform_matrix=Matrix.Identity(4))
+        
+        created_folder = context.scene.BSEQ.imported_zips.add()
+        created_folder.path = folder
+
         return {'FINISHED'}
 
+class BSEQ_OT_delete_zips(bpy.types.Operator):
+    """Delete a zip file"""
+    bl_idname = "bseq.delete_zips"
+    bl_label = "Delete Zip"
+    bl_options = {'PRESET', 'UNDO'}
+
+    def execute(self, context):
+        folders = context.scene.BSEQ.imported_zips
+        for folder in folders:
+            import shutil
+            shutil.rmtree(folder.path)
+        return {'FINISHED'}
+    
+
 class BSEQ_OT_meshio_object(bpy.types.Operator, ImportHelper):
     """Batch Import Meshio Objects"""
     bl_idname = "wm.meshio_import_batch"

bseq/panels.py

@@ -249,8 +249,13 @@ def draw(self, context):
             box_col3.label(text="Scale:")
             box_col3.prop(importer_prop, "custom_scale", text="")
 
-        layout.operator("bseq.import_zip", text="Import from zip")
-        
+        split = layout.split(factor=0.5)
+        col1 = split.column()
+        col2 = split.column()
+
+        col1.operator("bseq.import_zip", text="Import from zip")
+        col2.operator("bseq.delete_zips", text="Delete created folders")
+
 class BSEQ_PT_Import_Child1(BSEQ_Panel, bpy.types.Panel):
     bl_parent_id = "BSEQ_PT_panel"
     bl_label = "Import from folder"

bseq/properties.py

@@ -2,6 +2,13 @@
 from .callback import *
 from mathutils import Matrix
 
+class BSEQ_ImportedZip(bpy.types.PropertyGroup):
+    path: bpy.props.StringProperty(name="Directory",
+                                    subtype="DIR_PATH",
+                                    )
+
+bpy.utils.register_class(BSEQ_ImportedZip)
+
 class BSEQ_scene_property(bpy.types.PropertyGroup):
     path: bpy.props.StringProperty(name="Directory",
                                    subtype="DIR_PATH",
@@ -112,6 +119,8 @@ class BSEQ_scene_property(bpy.types.PropertyGroup):
                                             description='Filter string for file sequences',
                                             default='',
                                             )
+    
+    imported_zips: bpy.props.CollectionProperty(type=BSEQ_ImportedZip)
 
 class BSEQ_obj_property(bpy.types.PropertyGroup):
     init: bpy.props.BoolProperty(default=False)