]> gitweb.fluxo.info Git - grab-queue.git/commitdiff
Handle MemoryError
authorSilvio Rhatto <rhatto@riseup.net>
Thu, 18 Jul 2019 16:32:11 +0000 (13:32 -0300)
committerSilvio Rhatto <rhatto@riseup.net>
Thu, 18 Jul 2019 16:32:11 +0000 (13:32 -0300)
grab-queue

index dbdc0301ab9b345aacba89ef1598b766352dc147..bfeaf406e5916e167a8ea8799a79a0f7afd1cf03 100755 (executable)
@@ -271,13 +271,13 @@ if __name__ == "__main__":
     parser.set_defaults(progress=True)
     args = parser.parse_args()
 
+    # Initialize our dumper
+    queue = GrabQueue(args)
+
     # Dispatch
     try:
         start_time = time.time()
 
-        # Initialize our dumper
-        queue = GrabQueue(args)
-
         # Record date and invocation
         logs = ''.join(args.dest) + os.sep + 'logs' + os.sep
         queue.download.ensuredir(logs)
@@ -298,6 +298,10 @@ if __name__ == "__main__":
         elapsed = open(logs + 'grab-queue.duration', 'a')
         elapsed.write(str(start_time) + '\t' + str(end_time) + '\t' + duration)
         elapsed.close()
+    except MemoryError as e:
+        print(e)
+        queue.dump_stats()
+        exit(1)
     except (FileNotFoundError, KeyboardInterrupt, subprocess.CalledProcessError) as e:
         print(e)
         exit(1)