updates to process and pipe import code...need a way to turn

billiard on/off

Author: FredworkLemmas

memory_profiler.py

@@ -21,16 +21,34 @@
 
 
 # TODO: provide alternative when multiprocessing is not available
-try:
-    if "MEMPROF_PREFERS_BILLIARD" in os.environ and \
-            os.environ["MEMPROF_PREFERS_BILLIARD"] in ["True", "true", "TRUE", "1"]:
-        from billiard import Process, Pipe
-except ImportError:
-    try:
-        from multiprocessing import Process, Pipe
-    except ImportError:
-        from multiprocessing.dummy import Process, Pipe
+import billiard, multiprocessing
+MULTILIBS = (
+    (billiard, lambda: True if "MEMPROF_PREFERS_BILLIARD" in os.environ and \
+            os.environ["MEMPROF_PREFERS_BILLIARD"] in ["True", "true", "TRUE", "1"]),
+    (multiprocessing,),
+    (multiprocessing.dummy,)
+)
+for module_tuple in MULTILIBS:
+    multi_module, use_if_test = module_tuple
+    if use_if_test is None or use_if_test():
+        try:
+            from multi_module import Process, Pipe
+            break
+        except ImportError:
+            continue
+
+
 
+# try:
+#     if "MEMPROF_PREFERS_BILLIARD" in os.environ and \
+#             os.environ["MEMPROF_PREFERS_BILLIARD"] in ["True", "true", "TRUE", "1"]:
+#         from billiard import Process, Pipe
+# except ImportError:
+#     try:
+#         from multiprocessing import Process, Pipe
+#     except ImportError:
+#         from multiprocessing.dummy import Process, Pipe
+#
 
 try:
     from IPython.core.magic import Magics, line_cell_magic, magics_class