Add. command timeout

kazenniy · Feb 24, 2016 · 61c7711 · 61c7711
1 parent 1c3114a
commit 61c7711
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/atapt/atapt.py b/atapt/atapt.py
@@ -156,7 +156,8 @@ class atapt:
     def __init__(self, dev):
         self.smart = {}
         self.ssd = 0
-        self.duration = 0 
+        self.duration = 0
+        self.timeout = 1000
         self.readCommand = ATA_READ_SECTORS
         self.verifyCommand = ATA_READ_VERIFY_SECTORS
         self.writeCommand = ATA_WRITE_SECTORS
@@ -239,7 +240,7 @@ def prepareSgio(self, cmd, feature, count, lba, buf):
                        dxfer_len=buf_len,
                        dxferp=buf_p,
                        cmdp=ctypes.addressof(ata_cmd),
-                       sbp=ctypes.cast(self.sense, ctypes.c_void_p), timeout=1000,
+                       sbp=ctypes.cast(self.sense, ctypes.c_void_p), timeout=self.timeout,
                        flags=0, pack_id=0, usr_ptr=None, status=0, masked_status=0,
                        msg_status=0, sb_len_wr=0, host_status=0, driver_status=0,
                        resid=0, duration=0, info=0)

diff --git a/example/example.py b/example/example.py
@@ -32,7 +32,7 @@
 print("log. sector size:   %d bytes" % disk.logicalSectorSize)
 print("phys. sector size:  %d bytes" % disk.physicalSectorSize)
 
-
+disk.timeout = 1000
 # Read SMART
 print()
 print("Read SMART")