pysize/ui/ascii/ui_ascii.py

   1 # This program is free software; you can redistribute it and/or modify
   2 # it under the terms of the GNU General Public License as published by
   3 # the Free Software Foundation; either version 2 of the License, or
   4 # (at your option) any later version.
   5 #
   6 # This program is distributed in the hope that it will be useful,
   7 # but WITHOUT ANY WARRANTY; without even the implied warranty of
   8 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9 # GNU Library General Public License for more details.
  10 #
  11 # You should have received a copy of the GNU General Public License
  12 # along with this program; if not, write to the Free Software
  13 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  14 #
  15 # See the COPYING file for license information.
  16 #
  17 # Copyright (c) 2006 Guillaume Chazarain <guichaz@yahoo.fr>
  18
  19 import math
  20 import sys
  21
  22 from pysize.core.pysize_fs_tree import pysize_tree
  23 from pysize.core.compute_size import size_observable
  24 from pysize.ui.utils import human_unit, min_size_to_consider, update_progress
  25 from pysize.ui.utils import sanitize_string
  26 from pysize.ui.char_matrix import CharMatrix, HLINE_START, HLINE, HLINE_END
  27 from pysize.ui.char_matrix import VLINE_START, VLINE, VLINE_END
  28 from pysize.ui.ascii.terminal_size import terminal_size
  29
  30 MATRIX_TO_ASCII = {
  31     0: ' ',
  32     HLINE_START: '-', HLINE: '-', HLINE_END: '-',
  33     VLINE_START: '|', VLINE: '|', VLINE_END: '|'
  34 }
  35
  36 def _transform(char):
  37     if isinstance(char, int):
  38         return MATRIX_TO_ASCII.get(char, '+')
  39     return char
  40
  41 def _draw_progress():
  42     progress = update_progress()
  43     if progress:
  44         print '', progress, '\r',
  45         sys.stdout.flush()
  46
  47 def run(options, args):
  48     columns, lines = terminal_size()
  49     lines -= 3 # Summary + Prompt + Last line
  50     # An entry needs 2 lines
  51     min_size = min_size_to_consider(options.min_size, lines / 2)
  52     size_observable.add_observer(_draw_progress)
  53     args = args or ['.']
  54     tree = pysize_tree(args, options.max_depth, min_size, options)
  55     if not tree.root:
  56         sys.exit(1)
  57     # The last entry needs an additional terminating line
  58     total_lines = int(math.ceil(2.0 * tree.root.size /
  59                                 tree.root.minimum_node_size())) + 1
  60     matrix = CharMatrix(columns, total_lines, tree)
  61     print '\n'.join([''.join(map(_transform, line)) for line in matrix.matrix])
  62     print sanitize_string(tree.root.get_name()), human_unit(tree.root.size)