Fix not saving session after adb connection error

alexal1 · Aug 22, 2020 · 36c759e · 36c759e
1 parent 6f8815a
commit 36c759e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/utils.py b/src/utils.py
@@ -78,8 +78,11 @@ def close_instagram(device_id):
 def take_screenshot(device):
     os.makedirs("screenshots/", exist_ok=True)
     filename = "Crash-" + datetime.now().strftime("%Y-%m-%d-%H-%M-%S") + ".png"
-    device.screenshot("screenshots/" + filename)
-    print(COLOR_OKGREEN + "Screenshot taken and saved as " + filename + COLOR_ENDC)
+    try:
+        device.screenshot("screenshots/" + filename)
+        print(COLOR_OKGREEN + "Screenshot taken and saved as " + filename + COLOR_ENDC)
+    except RuntimeError:
+        print(COLOR_FAIL + "Cannot save screenshot." + COLOR_ENDC)
 
 
 def print_copyright(username):