xref: /freebsd/cddl/contrib/opensolaris/cmd/pyzfs/pyzfs.py (revision 3c4ba5f55438f7afd4f4b0b56f88f2bb505fd6a6)
1#! /usr/bin/python2.4 -S
2#
3# CDDL HEADER START
4#
5# The contents of this file are subject to the terms of the
6# Common Development and Distribution License (the "License").
7# You may not use this file except in compliance with the License.
8#
9# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10# or http://www.opensolaris.org/os/licensing.
11# See the License for the specific language governing permissions
12# and limitations under the License.
13#
14# When distributing Covered Code, include this CDDL HEADER in each
15# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16# If applicable, add the following below this CDDL HEADER, with the
17# fields enclosed by brackets "[]" replaced with your own identifying
18# information: Portions Copyright [yyyy] [name of copyright owner]
19#
20# CDDL HEADER END
21#
22# Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23# Use is subject to license terms.
24#
25
26# Note, we want SIGINT (control-c) to exit the process quietly, to mimic
27# the standard behavior of C programs.  The best we can do with pure
28# Python is to run with -S (to disable "import site"), and start our
29# program with a "try" statement.  Hopefully nobody hits ^C before our
30# try statement is executed.
31
32try:
33	import site
34	import gettext
35	import zfs.util
36	import zfs.ioctl
37	import sys
38	import errno
39
40	"""This is the main script for doing zfs subcommands.  It doesn't know
41	what subcommands there are, it just looks for a module zfs.<subcommand>
42	that implements that subcommand."""
43
44	_ = gettext.translation("SUNW_OST_OSCMD", "/usr/lib/locale",
45	    fallback=True).gettext
46
47	if len(sys.argv) < 2:
48		sys.exit(_("missing subcommand argument"))
49
50	zfs.ioctl.set_cmdstr(" ".join(["zfs"] + sys.argv[1:]))
51
52	try:
53		# import zfs.<subcommand>
54		# subfunc =  zfs.<subcommand>.do_<subcommand>
55
56		subcmd = sys.argv[1]
57		__import__("zfs." + subcmd)
58		submod = getattr(zfs, subcmd)
59		subfunc = getattr(submod, "do_" + subcmd)
60	except (ImportError, AttributeError):
61		sys.exit(_("invalid subcommand"))
62
63	try:
64		subfunc()
65	except zfs.util.ZFSError, e:
66		print(e)
67		sys.exit(1)
68
69except IOError, e:
70	import errno
71	import sys
72
73	if e.errno == errno.EPIPE:
74		sys.exit(1)
75	raise
76except KeyboardInterrupt:
77	import sys
78
79	sys.exit(1)
80