Began USB stick support

2 files changed, 29 insertions, 8 deletions

diff --git a/burntools.py b/burntools.py
index 9acd3bd..1bddbe7 100644
--- a/burntools.py
+++ b/burntools.py
@@ -1,6 +1,7 @@
 # Copyright (c) 2010, Nathaniel Theis
 
-import pexpect, re
+import pexpect, re, fcntl, os, sys
+ 
 
 #Define some handy things (regexes, strings, etc)
 
@@ -34,8 +35,26 @@ fixatedonematcher = re.compile(r"Fixating\stime:\s+(?P<time>\d+\.\d+)s")
                                   
 carriagereturn = chr(13)
 
-
-
+class USBStickWriter():
+    def __init__(self, image, bs=512):
+        self.image = image
+        self.bs = bs
+        
+    def run(self, device):
+        image_file = open(self.image, "r")
+        image_fd = image_file.fileno()
+        
+        fcntl.fcntl(image_fd, fcntl.F_SETFL, os.O_NONBLOCK)
+        #We need to use non-blocking read, so tell Linux that
+        
+        with open(device, "w") as out
+            while True:
+                block = image_file.read(self.bs)
+
+                if block == "": break
+                
+                out.write(block)
+            
 class WodimWrapper():
     '''WodimWrapper is a class to burn CDs with Wodim'''
     def _get_track_data(self):
@@ -140,8 +159,10 @@ if __name__ == "__main__":
     
     def printer(*args):
         for arg in args: print arg
-        
-    ww = WodimWrapper([], printer, printer, printer,
-                      printer, printer, printer, printer,
-                      command_path="./wodim")
-    ww.run()
+
+    f = USBStickWriter("ToasterActivity.py", bs=512)
+    f.run("/dev/mem")
+    #ww = WodimWrapper([], printer, printer, printer,
+    #                  printer, printer, printer, printer,
+    #                 command_path="./wodim")
+    #ww.run()
diff --git a/pexpect.pyo b/pexpect.pyo
index cbcce14..41a9d18 100644
--- a/pexpect.pyo
+++ b/pexpect.pyo