[Bf-blender-cvs] [7ae6959] soc-2014-fluid: Better handling of obj files when writing there

[Roman Pogribnyi]

Commit: 7ae6959df9c38d3a1819748599d96a33909dc1d7
Author: Roman Pogribnyi
Date:   Fri Jun 13 22:07:20 2014 +0200
https://developer.blender.org/rB7ae6959df9c38d3a1819748599d96a33909dc1d7

Better handling of obj files when writing there

===================================================================

M	release/scripts/startup/bl_ui/properties_physics_smoke.py

===================================================================

diff --git a/release/scripts/startup/bl_ui/properties_physics_smoke.py b/release/scripts/startup/bl_ui/properties_physics_smoke.py
index 295fa41..8b88da0 100644
--- a/release/scripts/startup/bl_ui/properties_physics_smoke.py
+++ b/release/scripts/startup/bl_ui/properties_physics_smoke.py
@@ -330,6 +330,10 @@ class OBJECT_OT_MantaButton(bpy.types.Operator):
     bl_label = "Create Python Script and mesh files"
     
     def execute(self, context):
+        def silent_remove(filename):
+            if os.path.exists(filename):
+                os.remove(filename)
+		
         coll_objs = []
         flow_objs = []
         selected_before = []
@@ -344,16 +348,20 @@ class OBJECT_OT_MantaButton(bpy.types.Operator):
                             coll_objs.append(ob)
                         elif modifier.smoke_type == 'FLOW':
                             flow_objs.append(ob)
-        for ob in coll_objs:
-            ob.select = True
-        bpy.ops.export_scene.obj(filepath = "./manta_coll.obj", use_selection = True, use_normals = True, use_materials = False, use_triangles = True, group_by_object = True, use_nurbs=True, check_existing= False)
-        for ob in coll_objs:
-            ob.select = False
-        for ob in flow_objs:
-            ob.select = True
-        bpy.ops.export_scene.obj(filepath = "./manta_flow.obj", use_selection = True, use_normals = True, use_materials = False, use_triangles = True, group_by_object = True, use_nurbs=True, check_existing= False)
-        for ob in flow_objs:
-            ob.select = False
+        silent_remove("./manta_coll.obj")
+        silent_remove("./manta_flow.obj")
+        if coll_objs: 
+            for ob in coll_objs:
+                ob.select = True
+            bpy.ops.export_scene.obj(filepath = "./manta_coll.obj", use_selection = True, use_normals = True, use_materials = False, use_triangles = True, group_by_object = True, use_nurbs=True, check_existing= False)
+            for ob in coll_objs:
+                ob.select = False
+        if flow_objs:
+            for ob in flow_objs:
+                ob.select = True
+            bpy.ops.export_scene.obj(filepath = "./manta_flow.obj", use_selection = True, use_normals = True, use_materials = False, use_triangles = True, group_by_object = True, use_nurbs=True, check_existing= False)
+            for ob in flow_objs:
+                ob.select = False
         for ob in selected_before:
             ob.select = True
         bpy.ops.manta.make_file()