Merge pull request JohnHammond#9 from JohnHammond/fix_issue8

Added check for availability of os.sched_getaffinity.

katana/manager.py

@@ -67,7 +67,7 @@ def __init__(self, monitor: Monitor = None, config_path=None, default_units=True
         # Default values for configuration items
         self["DEFAULT"] = {
             "units": "",
-            "threads": len(os.sched_getaffinity(0)),
+            "threads": 4,
             "outdir": "./results",
             "auto": False,
             "recurse": True,
@@ -85,6 +85,15 @@ def __init__(self, monitor: Monitor = None, config_path=None, default_units=True
         if "manager" not in self:
             self["manager"] = {}
 
+        try:
+            self["DEFAULT"]["threads"] = str(len(os.sched_getaffinity(0)))
+        except AttributeError:
+            # os.sched_getaffinity does not exist. We fallback to
+            # multiprocessing
+            import multiprocessing
+
+            self["DEFAULT"]["threads"] = str(multiprocessing.cpu_count())
+
         # Load a configuration file if specified
         if config_path is not None and len(self.read(config_path)) == 0:
             raise RuntimeError("{0}: configuration file not found")