Changes in tools/mkfat.py [cc1a727:432f68a] in mainline

File:

• tools/mkfat.py (modified) (7 diffs)

tools/mkfat.py

@@ -37 +37 @@
 import xstruct
 import array
-from imgutil import *
+
+exclude_names = set(['.svn', '.bzr'])
+
+def align_up(size, alignment):
+        "Return size aligned up to alignment"
+        
+        if (size % alignment == 0):
+                return size
+        
+        return ((size // alignment) + 1) * alignment
 
 def subtree_size(root, cluster_size, dirent_size):
@@ -45 +54 @@
         files = 2
         
-        for item in listdir_items(root):
-                if item.is_file:
-                        size += align_up(item.size, cluster_size)
+        for name in os.listdir(root):
+                canon = os.path.join(root, name)
+                
+                if (os.path.isfile(canon) and (not name in exclude_names)):
+                        size += align_up(os.path.getsize(canon), cluster_size)
                         files += 1
-                elif item.is_dir:
-                        size += subtree_size(item.path, cluster_size, dirent_size)
+                
+                if (os.path.isdir(canon) and (not name in exclude_names)):
+                        size += subtree_size(canon, cluster_size, dirent_size)
                         files += 1
         
@@ -60 +72 @@
         return len(os.listdir(root))
 
-def write_file(item, outf, cluster_size, data_start, fat, reserved_clusters):
+def write_file(path, outf, cluster_size, data_start, fat, reserved_clusters):
         "Store the contents of a file"
         
+        size = os.path.getsize(path)
         prev = -1
         first = 0
         
-        for data in chunks(item, cluster_size):
+        inf = open(path, "rb")
+        rd = 0;
+        while (rd < size):
                 empty_cluster = fat.index(0)
                 fat[empty_cluster] = 0xffff
@@ -77 +92 @@
                 prev = empty_cluster
                 
+                data = bytes(inf.read(cluster_size));
                 outf.seek(data_start + (empty_cluster - reserved_clusters) * cluster_size)
                 outf.write(data)
-        
-        return first, item.size
+                rd += len(data)
+        inf.close()
+        
+        return first, size
 
 def write_directory(directory, outf, cluster_size, data_start, fat, reserved_clusters, dirent_size, empty_cluster):
@@ -172 +190 @@
         parts = name.split('.')
         
-        if len(parts) > 0:
+        if (len(parts) > 0):
                 fname = parts[0]
         else:
                 fname = ''
-        
-        if len(fname) > 8:
-                sys.stdout.write("mkfat.py: error: Directory entry " + name +
-                    " base name is longer than 8 characters\n")
-                sys.exit(1);
-        
+                
         return (fname + '          ').upper()[0:8]
 
@@ -188 +201 @@
         parts = name.split('.')
         
-        if len(parts) > 1:
+        if (len(parts) > 1):
                 ext = parts[1]
         else:
                 ext = ''
-        
-        if len(parts) > 2:
-                sys.stdout.write("mkfat.py: error: Directory entry " + name +
-                    " has more than one extension\n")
-                sys.exit(1);
-        
-        if len(ext) > 3:
-                sys.stdout.write("mkfat.py: error: Directory entry " + name +
-                    " extension is longer than 3 characters\n")
-                sys.exit(1);
         
         return (ext + '   ').upper()[0:3]
@@ -285 +288 @@
                 empty_cluster = 0
         
-        for item in listdir_items(root):                
-                if item.is_file:
-                        rv = write_file(item, outf, cluster_size, data_start, fat, reserved_clusters)
-                        directory.append(create_dirent(item.name, False, rv[0], rv[1]))
-                elif item.is_dir:
-                        rv = recursion(False, item.path, outf, cluster_size, root_start, data_start, fat, reserved_clusters, dirent_size, empty_cluster)
-                        directory.append(create_dirent(item.name, True, rv[0], rv[1]))
+        for name in os.listdir(root):
+                canon = os.path.join(root, name)
+                
+                if (os.path.isfile(canon) and (not name in exclude_names)):
+                        rv = write_file(canon, outf, cluster_size, data_start, fat, reserved_clusters)
+                        directory.append(create_dirent(name, False, rv[0], rv[1]))
+                
+                if (os.path.isdir(canon) and (not name in exclude_names)):
+                        rv = recursion(False, canon, outf, cluster_size, root_start, data_start, fat, reserved_clusters, dirent_size, empty_cluster)
+                        directory.append(create_dirent(name, True, rv[0], rv[1]))
         
         if (head):