4766d990dfed9e9b7ee32b51c3430b726727de29
[portage.git] / bin / xpak-helper.py
1 #!/usr/bin/python
2 # Copyright 2009 Gentoo Foundation
3 # Distributed under the terms of the GNU General Public License v2
4
5 import optparse
6 import sys
7 import portage
8 from portage import os
9
10 def command_recompose(args):
11
12         usage = "usage: recompose <binpkg_path> <metadata_dir>\n"
13
14         if len(args) != 2:
15                 sys.stderr.write(usage)
16                 sys.stderr.write("2 arguments are required, got %s\n" % len(args))
17                 return 1
18
19         binpkg_path, metadata_dir = args
20
21         if not os.path.isfile(binpkg_path):
22                 sys.stderr.write(usage)
23                 sys.stderr.write("Argument 1 is not a regular file: '%s'\n" % \
24                         binpkg_path)
25                 return 1
26
27         if not os.path.isdir(metadata_dir):
28                 sys.stderr.write(usage)
29                 sys.stderr.write("Argument 2 is not a directory: '%s'\n" % \
30                         metadata_dir)
31                 return 1
32
33         t = portage.xpak.tbz2(binpkg_path)
34         t.recompose(metadata_dir)
35         return os.EX_OK
36
37 def main(argv):
38
39         if argv and sys.hexversion < 0x3000000 and not isinstance(argv[0], unicode):
40                 for i, x in enumerate(argv):
41                         argv[i] = portage._unicode_decode(x, errors='strict')
42
43         valid_commands = ('recompose',)
44         description = "Perform metadata operations on a binary package."
45         usage = "usage: %s COMMAND [args]" % \
46                 os.path.basename(argv[0])
47
48         parser = optparse.OptionParser(description=description, usage=usage)
49         options, args = parser.parse_args(argv[1:])
50
51         if not args:
52                 parser.error("missing command argument")
53
54         command = args[0]
55
56         if command not in valid_commands:
57                 parser.error("invalid command: '%s'" % command)
58
59         if command == 'recompose':
60                 rval = command_recompose(args[1:])
61         else:
62                 raise AssertionError("invalid command: '%s'" % command)
63
64         return rval
65
66 if __name__ == "__main__":
67         rval = main(sys.argv[:])
68         sys.exit(rval)