use print() - import print_function - should be fine for both py2 and py3

[nepi.git] / src / nepi / __init__.py

diff --git a/src/nepi/__init__.py b/src/nepi/__init__.py
index e69de29..18764d0 100644 (file)
--- a/src/nepi/__init__.py
+++ b/src/nepi/__init__.py
@@ -0,0 +1,57 @@
+#
+#    NEPI, a framework to manage network experiments
+#    Copyright (C) 2013 INRIA
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License version 2 as
+#    published by the Free Software Foundation;
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# Author: Alina Quereilhac <alina.quereilhac@inria.fr>
+
+from __future__ import print_function
+
+import logging
+import os
+import traceback
+
+LOGLEVEL = os.environ.get("NEPI_LOGLEVEL", "INFO").upper()
+LOGLEVEL = getattr(logging, LOGLEVEL)
+FORMAT = "%(asctime)s %(name)s %(levelname)-4s %(message)s"
+
+# NEPI_LOG variable contains space separated components 
+# on which logging should be enabled
+LOG = os.environ.get("NEPI_LOG", "ALL").upper()
+
+if LOG != 'ALL':
+    # Set by default loglevel to error
+    logging.basicConfig(format = FORMAT, level = logging.ERROR)
+
+    # Set logging level to that defined by the user
+    # only for the enabled components
+    for component in LOG.split(" "):
+        try:
+           log = logging.getLogger(component)
+           log.setLevel(LOGLEVEL)
+        except:
+            err = traceback.format_exc()
+            print("ERROR ", err)
+else:
+    # Set the logging level defined by the user for all
+    # components
+    logging.basicConfig(format = FORMAT, level = LOGLEVEL)
+
+
+# Add RMs to ResourceFactory. Use NEPI_SEARCH_PATH to 
+# override the default path to search for RMs
+from nepi.execution.resource import populate_factory
+populate_factory()
+
+