cat-file-cmd: accommodate python 3

[bup.git] / cmd / cat-file-cmd.py

diff --git a/cmd/cat-file-cmd.py b/cmd/cat-file-cmd.py
index 8948ff2bea844e85a8749a4714c2272e933ba87b..3f776a284d46f1ca83f8f31793c74f4d30b38ec0 100755 (executable)
--- a/cmd/cat-file-cmd.py
+++ b/cmd/cat-file-cmd.py
@@ -1,7 +1,18 @@
-#!/usr/bin/env python
-import sys, stat
+#!/bin/sh
+"""": # -*-python-*-
+bup_python="$(dirname "$0")/bup-python" || exit $?
+exec "$bup_python" "$0" ${1+"$@"}
+"""
+# end of bup preamble
+
+from __future__ import absolute_import
+import re, stat, sys
+
 from bup import options, git, vfs
-from bup.helpers import *
+from bup.compat import argv_bytes
+from bup.helpers import chunkyreader, handle_ctrl_c, log, saved_errors
+from bup.io import byte_stream
+from bup.repo import LocalRepo
 
 optspec = """
 bup cat-file [--meta|--bupm] /branch/revision/[path]
@@ -16,7 +27,6 @@ o = options.Options(optspec)
 (opt, flags, extra) = o.parse(sys.argv[1:])
 
 git.check_repo_or_die()
-top = vfs.RefList(None)
 
 if not extra:
     o.fatal('must specify a target')
@@ -25,35 +35,39 @@ if len(extra) > 1:
 if opt.bupm and opt.meta:
     o.fatal('--meta and --bupm are incompatible')
     
-target = extra[0]
+target = argv_bytes(extra[0])
 
-if not re.match(r'/*[^/]+/[^/]+', target):
+if not re.match(br'/*[^/]+/[^/]+', target):
     o.fatal("path %r doesn't include a branch and revision" % target)
 
-try:
-    n = top.lresolve(target)
-except vfs.NodeError, e:
-    o.fatal(e)
+repo = LocalRepo()
+resolved = vfs.resolve(repo, target, follow=False)
+leaf_name, leaf_item = resolved[-1]
+if not leaf_item:
+    log('error: cannot access %r in %r\n'
+        % ('/'.join(name for name, item in resolved), path))
+    sys.exit(1)
+
+mode = vfs.item_mode(leaf_item)
 
-if isinstance(n, vfs.FakeSymlink):
-    # Source is actually /foo/what, i.e. a top-level commit
-    # like /foo/latest, which is a symlink to ../.commit/SHA.
-    # So dereference it.
-    target = n.dereference()
+sys.stdout.flush()
+out = byte_stream(sys.stdout)
 
 if opt.bupm:
-    if not stat.S_ISDIR(n.mode):
+    if not stat.S_ISDIR(mode):
         o.fatal('%r is not a directory' % target)
-    mfile = n.metadata_file() # VFS file -- cannot close().
-    if mfile:
-        meta_stream = mfile.open()
-        sys.stdout.write(meta_stream.read())
+    _, bupm_oid = vfs.tree_data_and_bupm(repo, leaf_item.oid)
+    if bupm_oid:
+        with vfs.tree_data_reader(repo, bupm_oid) as meta_stream:
+            out.write(meta_stream.read())
 elif opt.meta:
-    sys.stdout.write(n.metadata().encode())
+    augmented = vfs.augment_item_meta(repo, leaf_item, include_size=True)
+    out.write(augmented.meta.encode())
 else:
-    if stat.S_ISREG(n.mode):
-        for b in chunkyreader(n.open()):
-            sys.stdout.write(b)
+    if stat.S_ISREG(mode):
+        with vfs.fopen(repo, leaf_item) as f:
+            for b in chunkyreader(f):
+                out.write(b)
     else:
         o.fatal('%r is not a plain file' % target)