Error handling

sowwic · Jul 28, 2020 · 507a78d · 507a78d
1 parent b040b77
commit 507a78d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/dsPlayblast/clientFn.py b/dsPlayblast/clientFn.py
@@ -26,6 +26,8 @@ def connect(self, port: int = -1):
         return True
 
     def disconnect(self):
+        if not self.maya_socket:
+            return False
         try:
             self.maya_socket.close()
         except BaseException:

diff --git a/dsPlayblast/playblastUI.py b/dsPlayblast/playblastUI.py
@@ -59,6 +59,10 @@ def always_on_top(self, state):
     def closeEvent(self, event):
         super().closeEvent(event)
         self.save_settings()
+        try:
+            self.maya_client.disconnect()
+        except AttributeError:
+            pass
 
     # Settings
     def save_settings(self):
@@ -324,7 +328,6 @@ def playblast(self) -> None:
         avi_result_path = self.output_path.get_path().replace(".mp4", ".avi")
         self.update_progress_bar()
         self.maya_client.send(self.playblast_command())
-        self.maya_client.disconnect()
         self.update_progress_bar()
 
         # Conversion