add JSON and PARQUET

Author: kolibril13

csv_importer/addon.py

@@ -1,7 +1,7 @@
 import bpy
 from . import ops, props, ui
 from .props import CSVImporterObjectProperties, CSVExporterSceneProperties
-from bpy.props import PointerProperty
+from bpy.props import PointerProperty, EnumProperty
 from .ops import ImportCsvPolarsOperator
 from .utils import add_current_module_to_path
 
@@ -20,6 +20,18 @@ def register():
     bpy.types.TOPBAR_MT_file_import.append(menu_func_import)
     bpy.types.Object.csv = PointerProperty(type=CSVImporterObjectProperties)  # type: ignore
     bpy.types.Scene.csv_export = PointerProperty(type=CSVExporterSceneProperties)  # type: ignore
+    
+    # Add export type property to window manager for UI
+    bpy.types.WindowManager.csv_export_type = EnumProperty(  # type: ignore
+        name="Export Type",
+        description="Type of file to export",
+        items=[
+            ('CSV', "CSV", "Export as CSV"),
+            ('JSON', "JSON", "Export as JSON"), 
+            ('PARQUET', "Parquet", "Export as Parquet")
+        ],
+        default='CSV'
+    )
 
 
 def unregister():
@@ -28,3 +40,4 @@ def unregister():
         bpy.utils.unregister_class(cls)
     del bpy.types.Object.csv  # type: ignore
     del bpy.types.Scene.csv_export  # type: ignore
+    del bpy.types.WindowManager.csv_export_type  # type: ignore

csv_importer/exporters.py

@@ -71,4 +71,7 @@ def from_polars_df_to_csv(df: pl.DataFrame, export_path: str) -> None:
     expanded_df = expanded_df.select(column_order)
 
     # Write processed DataFrame to CSV file
-    expanded_df.write_csv(export_path) 
+    expanded_df.write_csv(export_path) 
+
+
+

csv_importer/ops.py

@@ -134,12 +134,22 @@ def execute(self, context):
             self.report({"INFO"}, "Hot reload started")
         return {"FINISHED"}
 
-
 class CSV_OT_ExportData(bpy.types.Operator):
     bl_idname = "csv.export_data"
-    bl_label = "Export CSV"
+    bl_label = "Export Data"
     bl_options = {"REGISTER", "UNDO"}
-    bl_description = "Export mesh attribute data to CSV file"
+    bl_description = "Export mesh attribute data to file"
+
+    export_type: bpy.props.EnumProperty(  # type: ignore
+        name="Export Type",
+        description="Type of file to export",
+        items=[
+            ('CSV', "CSV", "Export as CSV"),
+            ('JSON', "JSON", "Export as JSON"), 
+            ('PARQUET', "Parquet", "Export as Parquet")
+        ],
+        default='CSV'
+    )
 
     def execute(self, context):
         scene = context.scene
@@ -167,14 +177,19 @@ def execute(self, context):
             # Convert Blender object to Polars DataFrame
             df = from_blender_to_polars_df(export_object)
 
-            # Export DataFrame to CSV
-            from_polars_df_to_csv(df, export_path)
+            # Export DataFrame based on selected type
+            if self.export_type == 'CSV':
+                from_polars_df_to_csv(df, export_path)
+            elif self.export_type == 'JSON':
+                df.write_json(export_path)
+            else:  # PARQUET
+                df.write_parquet(export_path)
 
-            self.report({"INFO"}, f"CSV file exported to: {export_path} for object: {export_object.name} ({len(df)} rows, {len(df.columns)} columns)")
+            self.report({"INFO"}, f"{self.export_type} file exported to: {export_path} for object: {export_object.name} ({len(df)} rows, {len(df.columns)} columns)")
             return {"FINISHED"}
 
         except Exception as e:
-            self.report({"ERROR"}, f"Failed to export CSV file: {e}")
+            self.report({"ERROR"}, f"Failed to export {self.export_type} file: {e}")
             return {"CANCELLED"}
 
 

csv_importer/ui.py

@@ -40,7 +40,6 @@ def draw(self, context):
             "csv.toggle_hot_reload", text=message, icon=icon, depress=obj.csv.hot_reload
         )
 
-
 class CSV_PT_ExportPanel(bpy.types.Panel):
     bl_label = "Export"
     bl_idname = "CSV_PT_ExportPanel"
@@ -62,9 +61,14 @@ def draw(self, context):
         row = layout.row()
         row.prop(scene.csv_export, "export_path", text="Export Path")
 
-        # Export button
-        row = layout.row()
-        row.operator("csv.export_data", text="Export CSV", icon="EXPORT")
+        # Export format buttons
+        row = layout.row(align=True)
+        op = row.operator("csv.export_data", text="CSV", icon="FILE")
+        op.export_type = 'CSV'
+        op = row.operator("csv.export_data", text="JSON", icon="FILE")
+        op.export_type = 'JSON'
+        op = row.operator("csv.export_data", text="Parquet", icon="FILE") 
+        op.export_type = 'PARQUET'
 
 
 CLASSES = (CSV_PT_ObjectPanel, CSV_PT_ExportPanel)