Sort imports: part 1

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

diff --git a/src/nitsi/machine.py b/src/nitsi/machine.py
index 0d790ae0eab5be5c4f5ba059daee6aeab47dca0c..4e60323a159420531cbf1e652a6d34040a3bbacd 100644 (file)
--- a/src/nitsi/machine.py
+++ b/src/nitsi/machine.py
@@ -1,15 +1,12 @@
 #!/usr/bin/python3
 
-import xml.etree.ElementTree as ET
-
-from nitsi.disk import disk
-
-from nitsi.serial_connection import serial_connection
-
-import os
 import libvirt
-
 import logging
+import os
+import xml.etree.ElementTree as ET
+
+from .disk import disk
+from .serial_connection import serial_connection
 
 logger = logging.getLogger("nitsi.machine")
 
@@ -117,9 +114,14 @@ class machine():
 
         #serial_con.close()
 
-    def login(self):
+    def login(self, log_file, log_start_time=None, longest_machine_name=10):
         try:
-            self.serial_con = serial_connection(self.get_serial_device(), username=self.username)
+            self.serial_con = serial_connection(self.get_serial_device(),
+                                username=self.username,
+                                log_file=log_file,
+                                log_start_time=log_start_time,
+                                name=self.name,
+                                longest_machine_name=longest_machine_name)
             self.serial_con.login(self.password)
         except BaseException as e:
             self.log.error("Could not connect to the domain via serial console")
@@ -137,4 +139,4 @@ class machine():
             self.log.error(e)
         finally:
             self.disk.umount("/")
-            self.disk.close()
\ No newline at end of file
+            self.disk.close()