cmd-split.py

   1 #!/usr/bin/env python2.5
   2 import sys, time, struct
   3 import hashsplit, git, options, client
   4 from helpers import *
   5 from subprocess import PIPE
   6
   7
   8 optspec = """
   9 bup split [-tcb] [-n name] [--bench] [filenames...]
  10 --
  11 r,remote=  remote repository path
  12 b,blobs    output a series of blob ids
  13 t,tree     output a tree id
  14 c,commit   output a commit id
  15 n,name=    name of backup set to update (if any)
  16 v,verbose  increase log output (can be used more than once)
  17 bench      print benchmark timings to stderr
  18 """
  19 o = options.Options('bup split', optspec)
  20 (opt, flags, extra) = o.parse(sys.argv[1:])
  21
  22 git.check_repo_or_die()
  23 if not (opt.blobs or opt.tree or opt.commit or opt.name):
  24     log("bup split: use one or more of -b, -t, -c, -n\n")
  25     o.usage()
  26
  27 hashsplit.split_verbosely = opt.verbose
  28 if opt.verbose >= 2:
  29     git.verbose = opt.verbose - 1
  30     opt.bench = 1
  31
  32 start_time = time.time()
  33
  34 refname = opt.name and 'refs/heads/%s' % opt.name or None
  35 if opt.remote:
  36     cli = client.Client(opt.remote)
  37     oldref = refname and cli.read_ref(refname) or None
  38     cli.sync_indexes()
  39     w = cli.new_packwriter()
  40 else:
  41     cli = None
  42     oldref = refname and git.read_ref(refname) or None
  43     w = git.PackWriter()
  44
  45 (shalist,tree) = hashsplit.split_to_tree(w, hashsplit.autofiles(extra))
  46
  47 if opt.verbose:
  48     log('\n')
  49 if opt.blobs:
  50     for (mode,name,bin) in shalist:
  51         print bin.encode('hex')
  52 if opt.tree:
  53     print tree.encode('hex')
  54 if opt.commit or opt.name:
  55     msg = 'bup split\n\nGenerated by command:\n%r' % sys.argv
  56     ref = opt.name and ('refs/heads/%s' % opt.name) or None
  57     commit = w.new_commit(oldref, tree, msg)
  58     if opt.commit:
  59         print commit.encode('hex')
  60
  61 w.close()  # must close before we can update the ref
  62
  63 if opt.name:
  64     if cli:
  65         cli.update_ref(refname, commit, oldref)
  66     else:
  67         git.update_ref(refname, commit, oldref)
  68
  69 if cli:
  70     cli.close()
  71
  72 secs = time.time() - start_time
  73 size = hashsplit.total_split
  74 if opt.bench:
  75     log('\nbup: %.2fkbytes in %.2f secs = %.2f kbytes/sec\n'
  76         % (size/1024., secs, size/1024./secs))