raise timeout, return child

2 files changed, 2 insertions, 1 deletions

diff --git a/ptlayout.py b/ptlayout.py
index 2d0a3ea..d333c2f 100644
--- a/ptlayout.py
+++ b/ptlayout.py
@@ -131,6 +131,7 @@ class PTLContainer(PTLayoutElem):
 
     def add(self, child):
         self._children.append(child)
+        return child
 
 class PTLHSpace(PTLContainer):
     def __init__(self):
diff --git a/pttarget.py b/pttarget.py
index ff05a1e..24d0e9a 100644
--- a/pttarget.py
+++ b/pttarget.py
@@ -153,7 +153,7 @@ class PTUSB(PTPrinter):
 			wsize = self.if_wr.wMaxPacketSize
 			if wsize > size - offset:
 				wsize = size - offset
-			written = self.if_wr.write(data[offset:offset+wsize])
+			written = self.if_wr.write(data[offset:offset+wsize], timeout = 5000)
 			if written <= 0:
 				raise IOError, 'failed to write'
 			offset += written