diff options
| -rw-r--r-- | freebase/fcl/cmdutil.py | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/freebase/fcl/cmdutil.py b/freebase/fcl/cmdutil.py new file mode 100644 index 0000000..4b6f0dc --- /dev/null +++ b/freebase/fcl/cmdutil.py @@ -0,0 +1,157 @@ +# ================================================================== +# Copyright (c) 2007,2008,2009 Metaweb Technologies, Inc. +# All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions +# are met: +# * Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above +# copyright notice, this list of conditions and the following +# disclaimer in the documentation and/or other materials provided +# with the distribution. +# +# THIS SOFTWARE IS PROVIDED BY METAWEB TECHNOLOGIES AND CONTRIBUTORS +# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS +# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL METAWEB +# TECHNOLOGIES OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, +# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, +# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER +# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN +# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE +# POSSIBILITY OF SUCH DAMAGE. +# ==================================================================== + + +import sys +import optparse + +from math import ceil + +import logging +log = logging.getLogger() + + +class CmdException(Exception): + pass + + +# decorator for adding subcommand option parsers +def option(argname, *args, **kws): + kws['dest'] = 'long' + def decorate(f): + oparser = getattr(f, 'oparser', None) + if oparser is None: + # set usage= later? + f.oparser = oparser = optparse.OptionParser() + oparser.disable_interspersed_args() + + oparser.add_option(*args, **kws) + return f + return decorate + + + +class TableOut(object): + def __init__(self): + self.outf = sys.stdout + self.rows = [] + self.cols = 1 + + # maximum number of lines to buffer + self.bufmax = 200 + + self.termwidth = 80 + self.minspacing = 4 + + # should be false if using in a pipeline, + # true for human user (like ls) + if hasattr(self.outf, "isatty") and self.outf.isatty(): + self.humane = True + else: + self.humane = False + + def __call__(self, *items): + self.rows.append(items) + if self.cols < len(items): + self.cols = len(items) + if len(self.rows) > self.bufmax: + self.flush() + + def flush(self): + if self.humane and self.cols == 1: + self.pack_list() + else: + self.tty_table() + + self.rows = [] + self.cols = 1 + + def pack_list(self): + l = [row[0] for row in self.rows] + maxlength = 0 + for item in l: + if maxlength < len(item): + maxlength = len(item) + + # round maxlength up to tab width + colwidth = int(ceil(maxlength + self.minspacing - 1)) + + if colwidth > 80: + colwidth = 80 + + ncols = self.termwidth // colwidth + + colwidths = [colwidth] * ncols + all = l + [None]*ncols + + row_major = False + if row_major: + items = [all[i:i+ncols] for i in range(0, len(l), ncols)] + else: + # standard unix ls order + nrows = (len(l)+ncols-1) // ncols + items = [all[i:i+nrows*ncols:nrows] for i in range(0, nrows)] + + self.showtty(colwidths, items) + + + def tty_table(self): + l = self.rows + + colwidths = [] + for row in l: + if len(colwidths) < len(row): + colwidths = colwidths + [0] * (len(row) - len(colwidths)) + for i, item in enumerate(row): + if colwidths[i] < len(item) + self.minspacing: + colwidths[i] = len(item) + self.minspacing + + self.showtty(colwidths, l) + + def showtty(self, colwidths, rows): + """Simple unix style text table output""" + + for row in rows: + #print '%s %r' % (type(row), row) + for i, item in enumerate(row): + if item is None: continue + self.outf.write(item) + # fill out column, except for last cell + if i < len(row) - 1: + padding = colwidths[i] - len(item) + if padding < 1: + padding = 1 + self.outf.write(' ' * padding) + self.outf.write('\n') + + +# "out" is a function for generating a "row" of output +# the row can be one or more strings +# +out = TableOut() + |