Changeset 13205 for main/waeup.uniben/trunk/src/waeup/uniben/scripts.py

Timestamp:

21 Aug 2015, 13:44:03 (9 years ago)

Author:

uli

Message:

Add get_mem_info().

We return memory infos if psutil is locally available.

File:

• main/waeup.uniben/trunk/src/waeup/uniben/scripts.py (modified) (2 diffs)

main/waeup.uniben/trunk/src/waeup/uniben/scripts.py

@@ -127 +127 @@
 
 
+def get_mem_info():
+    """Get current memory info.
+
+    This works only, if `psutil` is installed locally (in virtualenv).
+    Otherwise we return `None`.
+    """
+    try:
+        # late import. We do not want to make it a waeup.uniben dependency.
+        import psutil
+    except ImportError:
+        return
+    proc = psutil.Process(os.getpid())
+    return proc.get_memory_info().rss
+
+
 def export_part(container, part_num, start, end, path):
-    import psutil  # late import. Don't want to make it waeup.uniben req.
-    PROC = psutil.Process(os.getpid())
     gc.collect()
-    mem1 = PROC.get_memory_info().rss
+    mem1 = get_mem_info()
     print("  Export %s-%s to %s (mem: %s)" % (start, end, path, mem1))
     tp1 = time.time()
@@ -137 +150 @@
     exporter.export(get_all_students(container, start, end), path)
     tp2 = time.time()
-    mem2 = PROC.get_memory_info().rss
+    mem2 = get_mem_info()
     print("  Done (%s secs, mem: %s)" % (tp2 - tp1, mem2))