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 max-pack-size=  maximum bytes in a single pack
  19 max-pack-objects=  maximum number of objects in a single pack
  20 fanout=  maximum number of blobs in a single tree
  21 """
  22 o = options.Options('bup split', optspec)
  23 (opt, flags, extra) = o.parse(sys.argv[1:])
  24
  25 git.check_repo_or_die()
  26 if not (opt.blobs or opt.tree or opt.commit or opt.name):
  27     log("bup split: use one or more of -b, -t, -c, -n\n")
  28     o.usage()
  29
  30 hashsplit.split_verbosely = opt.verbose
  31 if opt.verbose >= 2:
  32     git.verbose = opt.verbose - 1
  33     opt.bench = 1
  34 if opt.max_pack_size:
  35     hashsplit.max_pack_size = int(opt.max_pack_size)
  36 if opt.max_pack_objects:
  37     hashsplit.max_pack_objects = int(opt.max_pack_objects)
  38 if opt.fanout:
  39     hashsplit.fanout = int(opt.fanout)
  40 if opt.blobs:
  41     hashsplit.fanout = 0
  42
  43 start_time = time.time()
  44
  45 refname = opt.name and 'refs/heads/%s' % opt.name or None
  46 if opt.remote:
  47     cli = client.Client(opt.remote)
  48     oldref = refname and cli.read_ref(refname) or None
  49     w = cli.new_packwriter()
  50 else:
  51     cli = None
  52     oldref = refname and git.read_ref(refname) or None
  53     w = git.PackWriter()
  54
  55 (shalist,tree) = hashsplit.split_to_tree(w, hashsplit.autofiles(extra))
  56
  57 if opt.verbose:
  58     log('\n')
  59 if opt.blobs:
  60     for (mode,name,bin) in shalist:
  61         print bin.encode('hex')
  62 if opt.tree:
  63     print tree.encode('hex')
  64 if opt.commit or opt.name:
  65     msg = 'bup split\n\nGenerated by command:\n%r' % sys.argv
  66     ref = opt.name and ('refs/heads/%s' % opt.name) or None
  67     commit = w.new_commit(oldref, tree, msg)
  68     if opt.commit:
  69         print commit.encode('hex')
  70
  71 w.close()  # must close before we can update the ref
  72
  73 if opt.name:
  74     if cli:
  75         cli.update_ref(refname, commit, oldref)
  76     else:
  77         git.update_ref(refname, commit, oldref)
  78
  79 if cli:
  80     cli.close()
  81
  82 secs = time.time() - start_time
  83 size = hashsplit.total_split
  84 if opt.bench:
  85     log('\nbup: %.2fkbytes in %.2f secs = %.2f kbytes/sec\n'
  86         % (size/1024., secs, size/1024./secs))