Sort imports: part 1

[nitsi.git] / src / nitsi / serial_connection.py

diff --git a/src/nitsi/serial_connection.py b/src/nitsi/serial_connection.py
index 6328e4f143b2c22353bbc3717d56adb499a0728d..35683161eebcb201e2cc2eb205de0b423fde7eeb 100644 (file)
--- a/src/nitsi/serial_connection.py
+++ b/src/nitsi/serial_connection.py
@@ -1,19 +1,18 @@
 #!/usr/bin/python3
-import serial
 
-import re
+import logging
 import os
-
-from time import sleep
+import re
+import serial
 import sys
-import logging
+import time
 
-from nitsi.logger import TestFormatter
+from .logger import TestFormatter
 
 logger = logging.getLogger("nitsi.serial")
 
 class serial_connection():
-    def __init__(self, device, username=None, log_file=None, name=None):
+    def __init__(self, device, username=None, log_file=None, name=None, log_start_time=None, longest_machine_name=10):
         self.buffer = b""
         self.back_at_prompt_pattern =  None
         self.username = username
@@ -27,7 +26,9 @@ class serial_connection():
         log_file_handler = logging.FileHandler(self.log_file)
         log_file_handler.setLevel(logging.INFO)
         log_file_handler.terminator = ""
-        formatter = TestFormatter(name=self.name, start_time=None)
+        formatter = TestFormatter(name=self.name,
+                                    start_time=log_start_time,
+                                    longest_machine_name=longest_machine_name)
         log_file_handler.setFormatter(formatter)
         self.log_output.addHandler(log_file_handler)
 
@@ -211,4 +212,4 @@ class serial_connection():
         data = data.decode().replace("END: ", "")
         self.log.debug(data)
         self.log.debug(data.strip())
-        return data.strip()
\ No newline at end of file
+        return data.strip()