Update upload_to_cuckoo.py

scorpionantimalware · Jul 25, 2024 · c7dbb1a · c7dbb1a
1 parent 4075098
commit c7dbb1a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/resources/scripts/upload_to_cuckoo.py b/resources/scripts/upload_to_cuckoo.py
@@ -96,6 +96,13 @@
 
         while True:
             try:
+                temp = driver.find_element(By.XPATH, '//*[@id="info"]/span[5]/strong').text
+
+                if temp == 'Something went wrong!':
+                    print('Something went wrong! Refreshing the page...\n')
+                    refresh = True
+                    break
+
                 b1 = None
                 b1 = driver.find_element(By.XPATH, '//*[@id="timeout"]/ul/li[1]')
 
@@ -110,6 +117,9 @@
                 time.sleep(1)
                 continue
 
+        if refresh:
+            continue
+
         time.sleep(0.5)
 
         while True: