Move python files to root folder; remove built docs

miplearn/log.py

@@ -0,0 +1,46 @@
+#  MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
+#  Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
+#  Released under the modified BSD license. See COPYING.md for more details.
+
+from datetime import timedelta
+import logging
+import time
+import sys
+
+if sys.stdout.isatty():
+    log_colors = {
+        "green":  '\033[92m',
+        "yellow": '\033[93m',
+        "red":    '\033[91m',
+        "reset":  '\033[0m',
+    }
+else:
+    log_colors = {
+        "green": "",
+        "yellow": "",
+        "red": "",
+        "reset": "",
+    }
+
+class TimeFormatter():
+    def __init__(self, start_time):
+        self.start_time = start_time
+
+    def format(self, record):
+        if record.levelno >= logging.ERROR:
+            color = log_colors["red"]
+        elif record.levelno >= logging.WARNING:
+            color = log_colors["yellow"]
+        else:
+            color = log_colors["green"]
+        return "%s[%12.3f]%s %s" % (color,
+                                    record.created - self.start_time,
+                                    log_colors["reset"],
+                                    record.getMessage())
+
+def setup_logger(start_time):
+    handler = logging.StreamHandler()
+    handler.setFormatter(TimeFormatter(start_time))
+    logging.getLogger().addHandler(handler)
+    logging.getLogger("miplearn").setLevel(logging.INFO)
+    lg = logging.getLogger("miplearn")