1#!@PYTHON@ 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 (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved. 23# 24 25import zfs.util 26 27class Table: 28 __slots__ = "fields", "rjustfields", "maxfieldlen", "lines" 29 __repr__ = zfs.util.default_repr 30 31 def __init__(self, fields, rjustfields=()): 32 # XXX maybe have a defaults, too? 33 self.fields = fields 34 self.rjustfields = rjustfields 35 self.maxfieldlen = dict.fromkeys(fields, 0) 36 self.lines = list() 37 38 def __updatemax(self, k, v): 39 self.maxfieldlen[k] = max(self.maxfieldlen.get(k, None), v) 40 41 def addline(self, sortkey, values): 42 """values is a dict from field name to value""" 43 44 va = list() 45 for f in self.fields: 46 v = str(values[f]) 47 va.append(v) 48 self.__updatemax(f, len(v)) 49 self.lines.append((sortkey, va)) 50 51 def printme(self, headers=True): 52 if headers: 53 d = dict([(f, f.upper()) for f in self.fields]) 54 self.addline(None, d) 55 56 self.lines.sort() 57 for (k, va) in self.lines: 58 line = str() 59 for i in range(len(self.fields)): 60 if not headers: 61 line += va[i] 62 line += "\t" 63 else: 64 if self.fields[i] in self.rjustfields: 65 fmt = "%*s " 66 else: 67 fmt = "%-*s " 68 mfl = self.maxfieldlen[self.fields[i]] 69 line += fmt % (mfl, va[i]) 70 print(line) 71