cddl/contrib/opensolaris/cmd/pyzfs/pyzfs.py

   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
  32 try:
  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
  69 except IOError, e:
  70         import errno
  71         import sys
  72
  73         if e.errno == errno.EPIPE:
  74                 sys.exit(1)
  75         raise
  76 except KeyboardInterrupt:
  77         import sys
  78
  79         sys.exit(1)