From 73923af9bbf03bfb31a6c0817048f1e674b8966b Mon Sep 17 00:00:00 2001
From: Marko Kreen <markokr@gmail.com>
Date: Thu, 3 Dec 2009 16:45:26 +0200
Subject: [PATCH] DBScript: quick exit on MemoryError

---
 python/skytools/scripting.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/skytools/scripting.py b/python/skytools/scripting.py
index 9d7c2b17..c4380d91 100644
--- a/python/skytools/scripting.py
+++ b/python/skytools/scripting.py
@@ -640,6 +640,12 @@ class DBScript(object):
         except UsageError, d:
             self.log.error(str(d))
             sys.exit(1)
+        except MemoryError, d:
+            try: # complex logging may not succeed
+                self.log.exception("Job %s out of memory, exiting" % self.job_name)
+            except MemoryError:
+                self.log.fatal("Out of memory")
+            sys.exit(1)
         except SystemExit, d:
             self.send_stats()
             if prefer_looping:
-- 
2.39.5