eric6/Snapshot/SnapshotWaylandGrabber.py

changeset 8205
4a0f1f896341
parent 8143
2c730d5fd177
child 8218
7c09585bd960
--- a/eric6/Snapshot/SnapshotWaylandGrabber.py	Thu Apr 08 17:27:12 2021 +0200
+++ b/eric6/Snapshot/SnapshotWaylandGrabber.py	Thu Apr 08 18:27:47 2021 +0200
@@ -105,16 +105,20 @@
         
         @exception RuntimeError raised to indicate an unsupported grab mode
         """
+        if self.__mode not in (
+            SnapshotModes.Fullscreen, SnapshotModes.SelectedScreen,
+            SnapshotModes.SelectedWindow, SnapshotModes.Rectangle,
+        ):
+            raise RuntimeError("unsupported grab mode given")
+        
         if self.__mode == SnapshotModes.Fullscreen:
             self.__grabFullscreen()
         elif self.__mode == SnapshotModes.SelectedScreen:
             self.__grabSelectedScreen()
         elif self.__mode == SnapshotModes.SelectedWindow:
             self.__grabSelectedWindow()
-        elif self.__mode == SnapshotModes.Rectangle:
+        else:
             self.__grabRectangle()
-        else:
-            raise RuntimeError("unsupported grab mode given")
     
     def __grabFullscreen(self):
         """

eric ide

mercurial