ADD file via upload

VPTTA-main/POLYP/config.py

@@ -0,0 +1,26 @@
+import os
+import sys
+import traceback
+
+
+class Logger(object):
+    def __init__(self, filename='default.log', stream=sys.stdout):
+        self.terminal = stream
+        self.filename = filename
+        self.log = open(filename, 'w')
+        self.hook = sys.excepthook
+        sys.excepthook = self.kill
+
+    def write(self, message):
+        self.terminal.write(message)
+        self.terminal.flush()
+        self.log.write(message)
+        self.log.flush()
+
+    def kill(self, ttype, tvalue, ttraceback):
+        for trace in traceback.format_exception(ttype, tvalue, ttraceback):
+            print(trace)
+        os.remove(self.filename)
+
+    def flush(self):
+        pass