lib/bup/cmd/join.py

   1
   2 from __future__ import absolute_import
   3
   4 import sys
   5
   6 from bup import git, options
   7 from bup.compat import argv_bytes
   8 from bup.helpers import linereader, log
   9 from bup.io import byte_stream
  10 from bup.repo import LocalRepo, RemoteRepo
  11
  12
  13 optspec = """
  14 bup join [-r host:path] [refs or hashes...]
  15 --
  16 r,remote=  remote repository path
  17 o=         output filename
  18 """
  19
  20 def main(argv):
  21     o = options.Options(optspec)
  22     opt, flags, extra = o.parse_bytes(argv[1:])
  23     if opt.remote:
  24         opt.remote = argv_bytes(opt.remote)
  25
  26     git.check_repo_or_die()
  27
  28     stdin = byte_stream(sys.stdin)
  29
  30     if not extra:
  31         extra = linereader(stdin)
  32
  33     ret = 0
  34     repo = RemoteRepo(opt.remote) if opt.remote else LocalRepo()
  35
  36     if opt.o:
  37         outfile = open(opt.o, 'wb')
  38     else:
  39         sys.stdout.flush()
  40         outfile = byte_stream(sys.stdout)
  41
  42     for ref in [argv_bytes(x) for x in extra]:
  43         try:
  44             for blob in repo.join(ref):
  45                 outfile.write(blob)
  46         except KeyError as e:
  47             outfile.flush()
  48             log('error: %s\n' % e)
  49             ret = 1
  50
  51     sys.exit(ret)