Răsfoiți Sursa

Move WrapLogger to utils file

Hamilton Turner 11 ani în urmă
părinte
comite
6c174d325f
1 a modificat fișierele cu 2 adăugiri și 17 ștergeri
  1. 2 17
      toolset/benchmark/framework_test.py

+ 2 - 17
toolset/benchmark/framework_test.py

@@ -14,6 +14,7 @@ import textwrap
 import logging
 log = logging.getLogger('framework_test')
 
+from utils import WrapLogger
 class FrameworkTest:
   """
   Represents a framework test, including all types (JSON, plaintext, DB, etc)
@@ -957,20 +958,4 @@ def parse_config(config, directory, benchmarker):
   return tests
 ##############################################################
 # End parse_config
-##############################################################
-
-import tempfile
-class WrapLogger():
-  def __init__(self, logger, level):
-    self.logger = logger
-    self.level = level
-    self.file = tempfile.TemporaryFile()
-
-  def write(self, message):
-    self.logger.log(self.level, message)
-
-  def __getattr__(self, name):
-    return getattr(self.file, name)
-
-
-
+##############################################################