Adds error exceptions

sosoyan · Jul 1, 2019 · 22fe98d · 22fe98d
1 parent dd2e9a0
commit 22fe98d
Showing 1 changed file with 29 additions and 22 deletions.
diff --git a/scripts/aton_houdini.py b/scripts/aton_houdini.py
@@ -2081,41 +2081,48 @@ def __export_ass(self):
         """
         for output in self.__output_list_box.selected_items():
 
-            output.set_status("Exporting ASS...")
-
             if output.rop is not None:
                 session_id = int(time.time())
                 ass_path = self.export_ass_path(output.rop_path, session_id)
                 ass_name = self.export_ass_name(output.rop_path, session_id)
 
                 if ass_path and ass_name:
-                    rop_ass_enable_param = output.rop.parm("ar_ass_export_enable")
-                    rop_ass_file_parm = output.rop.parm("ar_ass_file")
-                    rop_picture_param = output.rop.parm("ar_picture")
 
-                    if rop_ass_file_parm is not None:
+                    if os.path.isdir(ass_path):
+
+                        output.set_status("Exporting ASS...")
+
+                        rop_ass_enable_param = output.rop.parm("ar_ass_export_enable")
+                        rop_ass_file_parm = output.rop.parm("ar_ass_file")
+                        rop_picture_param = output.rop.parm("ar_picture")
 
-                        default_state = rop_ass_enable_param.eval()
-                        default_path = rop_ass_file_parm.rawValue()
-                        default_picture = rop_picture_param.eval()
+                        if rop_ass_file_parm is not None:
 
-                        rop_picture_param.set("")
-                        rop_ass_enable_param.set(1)
-                        ass_file_path = os.path.join(ass_path, ass_name)
-                        rop_ass_file_parm.set(ass_file_path)
-                        ass_file_path = rop_ass_file_parm.eval()
+                            default_state = rop_ass_enable_param.eval()
+                            default_path = rop_ass_file_parm.rawValue()
+                            default_picture = rop_picture_param.eval()
 
-                        output.rop.parm("execute").pressButton()
+                            rop_picture_param.set("")
+                            rop_ass_enable_param.set(1)
+                            ass_file_path = os.path.join(ass_path, ass_name)
+                            rop_ass_file_parm.set(ass_file_path)
+                            ass_file_path = rop_ass_file_parm.eval()
 
-                        rop_ass_enable_param.set(default_state)
-                        rop_ass_file_parm.set(default_path)
-                        rop_picture_param.set(default_picture)
+                            output.rop.parm("execute").pressButton()
 
-                        # Exported
-                        output.set_status()
+                            rop_ass_enable_param.set(default_state)
+                            rop_ass_file_parm.set(default_path)
+                            rop_picture_param.set(default_picture)
 
-                        if self.__add_ass_overrides(output, ass_file_path, session_id):
-                            self.__init_farm_job(output, ass_file_path, session_id)
+                            # Exported
+                            output.set_status()
+
+                            if self.__add_ass_overrides(output, ass_file_path, session_id):
+                                self.__init_farm_job(output, ass_file_path, session_id)
+                    else:
+                        output.set_status("Error: Invalid ASS path!")
+                else:
+                    output.set_status("Error: ASS path or ASS name is None!")
 
     def __init_farm_job(self, output, ass_file_path, session_id):
         """ Initialises farm job requirements