'bup split' takes a list of filenames on the command line.

[bup.git] / cmd-split.py

diff --git a/cmd-split.py b/cmd-split.py
index 04b6d7875a2be614999a2e969c758930f1c329a4..e755efec45338d73b7f88fdcbe9321ab1f694a30 100755 (executable)
--- a/cmd-split.py
+++ b/cmd-split.py
@@ -52,7 +52,7 @@ def hashsplit_iter(f):
     lv = 0
     while blob or not eof:
         if not eof and (buf.used() < BLOB_LWM or not blob):
-            bnew = sys.stdin.read(BLOB_HWM)
+            bnew = f.read(BLOB_HWM)
             if not len(bnew): eof = 1
             #log('got %d, total %d\n' % (len(bnew), buf.used()))
             buf.put(bnew)
@@ -74,25 +74,53 @@ def hashsplit_iter(f):
             log('%d\t' % nv)
             lv = nv
 
+
+def autofiles(filenames):
+    if not filenames:
+        yield sys.stdin
+    else:
+        for n in filenames:
+            yield open(n)
+
+
 optspec = """
-bup split [-t] <filename
+bup split [-t] [filenames...]
 --
-t,tree     output a tree instead of a series of blobs
+b,blobs    output a series of blob ids
+t,tree     output a tree id
+c,commit   output a commit id
+n,name=    name of backup set to update (if any)
+bench      print benchmark timings to stderr
 """
-(opt, flags, extra) = options.Options('bup split', optspec).parse(sys.argv[1:])
+o = options.Options('bup split', optspec)
+(opt, flags, extra) = o.parse(sys.argv[1:])
+
+if not (opt.blobs or opt.tree or opt.commit or opt.name):
+    log("bup split: use one or more of -b, -t, -c, -n\n")
+    o.usage()
 
 start_time = time.time()
 shalist = []
 
 ofs = 0
-for (ofs, size, sha) in hashsplit_iter(sys.stdin):
-    #log('SPLIT @ %-8d size=%-8d\n' % (ofs, size))
-    if not opt.tree:
-        print sha
-    shalist.append(('100644', '%016x.bupchunk' % ofs, sha))
+
+for f in autofiles(extra):
+    for (ofs, size, sha) in hashsplit_iter(f):
+        #log('SPLIT @ %-8d size=%-8d\n' % (ofs, size))
+        if opt.blobs:
+            print sha
+        shalist.append(('100644', '%016x.bupchunk' % ofs, sha))
+tree = git.gen_tree(shalist)
 if opt.tree:
-    print git.gen_tree(shalist)
+    print tree
+if opt.commit or opt.name:
+    msg = 'Generated by command:\n%r' % sys.argv
+    ref = opt.name and ('refs/heads/%s' % opt.name) or None
+    commit = git.gen_commit_easy(ref, tree, msg)
+    if opt.commit:
+        print commit
 
 secs = time.time() - start_time
-log('\n%.2fkbytes in %.2f secs = %.2f kbytes/sec\n'
-    % (ofs/1024., secs, ofs/1024./secs))
+if opt.bench:
+    log('\nbup: %.2fkbytes in %.2f secs = %.2f kbytes/sec\n'
+        % (ofs/1024., secs, ofs/1024./secs))