Merge branch '8784-dir-listings'
[arvados.git] / sdk / python / bin / arv-normalize
1 #!/usr/bin/env python
2 # Copyright (C) The Arvados Authors. All rights reserved.
3 #
4 # SPDX-License-Identifier: Apache-2.0
5
6 import argparse
7 import hashlib
8 import os
9 import re
10 import string
11 import sys
12
13 import arvados
14 from arvados._version import __version__
15
16 parser = argparse.ArgumentParser(
17     description='Read manifest on standard input and put normalized manifest on standard output.')
18
19 parser.add_argument('--extract', type=str,
20                     help="The file to extract from the input manifest")
21 parser.add_argument('--strip', action='store_true',
22                     help="Strip authorization tokens")
23 parser.add_argument('--version', action='version',
24                     version="%s %s" % (sys.argv[0], __version__),
25                     help='Print version and exit.')
26
27 args = parser.parse_args()
28
29 r = sys.stdin.read()
30
31 cr = arvados.CollectionReader(r)
32
33 if args.extract:
34     i = args.extract.rfind('/')
35     if i == -1:
36         stream = '.'
37         fn = args.extract
38     else:
39         stream = args.extract[:i]
40         fn = args.extract[(i+1):]
41     for s in cr.all_streams():
42         if s.name() == stream:
43             if fn in s.files():
44                 sys.stdout.write(s.files()[fn].as_manifest())
45 else:
46     sys.stdout.write(cr.manifest_text(strip=args.strip, normalize=True))