Changes in tools/mktmpfs.py [78eb2dc:432f68a] in mainline

File:

• tools/mktmpfs.py (modified) (2 diffs)

tools/mktmpfs.py

@@ -35 +35 @@
 import os
 import xstruct
-from imgutil import listdir_items, chunks
+
+exclude_names = set(['.svn', '.bzr'])
 
 HEADER = """little:
@@ -70 +71 @@
         "Recursive directory walk"
         
-        for item in listdir_items(root):                
-                if item.is_file:                        
-                        dentry = xstruct.create(DENTRY_FILE % len(item.name))
+        for name in os.listdir(root):
+                canon = os.path.join(root, name)
+                
+                if (os.path.isfile(canon) and (not name in exclude_names)):
+                        size = os.path.getsize(canon)
+                        
+                        dentry = xstruct.create(DENTRY_FILE % len(name))
                         dentry.kind = TMPFS_FILE
-                        dentry.fname_len = len(item.name)
-                        dentry.fname = item.name.encode('ascii')
-                        dentry.flen = item.size
+                        dentry.fname_len = len(name)
+                        dentry.fname = name.encode('ascii')
+                        dentry.flen = size
                         
                         outf.write(dentry.pack())
                         
-                        for data in chunks(item, 4096):
+                        inf = open(canon, "rb")
+                        rd = 0;
+                        while (rd < size):
+                                data = inf.read(4096);
                                 outf.write(data)
+                                rd += len(data)
+                        inf.close()
                 
-                elif item.is_dir:
-                        dentry = xstruct.create(DENTRY_DIRECTORY % len(item.name))
+                if (os.path.isdir(canon) and (not name in exclude_names)):
+                        dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
                         dentry.kind = TMPFS_DIRECTORY
-                        dentry.fname_len = len(item.name)
-                        dentry.fname = item.name.encode('ascii')
+                        dentry.fname_len = len(name)
+                        dentry.fname = name.encode('ascii')
                         
                         outf.write(dentry.pack())
                         
-                        recursion(item.path, outf)
+                        recursion(canon, outf)
                         
                         dentry = xstruct.create(DENTRY_NONE)