pysize/main.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 sys
  20 import os
  21 import optparse
  22 import locale
  23
  24 if sys.hexversion < 0x02040000:
  25         print >> sys.stderr, 'Your python version is too old (%s)' % \
  26                                                         (sys.version.split()[0])
  27         print >> sys.stderr, 'You need at least Python 2.4'
  28         sys.exit(1)
  29
  30 from pysize.ui.ascii import ui_ascii
  31 from pysize.ui.curses import ui_curses
  32 from pysize.ui.gtk import ui_gtk
  33 from pysize.ui.utils import UINotAvailableException
  34 from pysize.core.sigquit_traceback import install_sigquit_traceback
  35
  36 def _ui_auto(options, args):
  37     """Automatically choose the best available UI."""
  38     for ui_run in ui_gtk.run, ui_curses.run, ui_ascii.run:
  39         try:
  40             ui_run(options, args)
  41             return
  42         except UINotAvailableException:
  43             pass
  44
  45     raise UINotAvailableException
  46
  47 UI = {'ascii': ui_ascii.run, 'curses': ui_curses.run,
  48       'gtk': ui_gtk.run, 'auto': _ui_auto}
  49
  50 def _profile(continuation):
  51     try:
  52         import cProfile
  53         import pstats
  54         print 'Profiling using cProfile'
  55         cProfile.runctx('continuation()', globals(), locals(), 'pysize.prof')
  56         stats = pstats.Stats('pysize.prof')
  57     except ImportError:
  58         import hotshot
  59         import hotshot.stats
  60         prof = hotshot.Profile('pysize.prof')
  61         print 'Profiling using hotshot'
  62         prof.runcall(continuation)
  63         prof.close()
  64         stats = hotshot.stats.load('pysize.prof')
  65     stats.strip_dirs()
  66     stats.sort_stats('time', 'calls')
  67     stats.print_stats(40)
  68
  69 def _try_psyco():
  70     try:
  71         # Try to use psyco if available
  72         import psyco
  73         psyco.full()
  74     except ImportError:
  75         pass
  76
  77 def main():
  78     install_sigquit_traceback()
  79     _try_psyco()
  80     locale.setlocale(locale.LC_ALL, '')
  81     usage = '%s [OPTIONS] [DIRECTORIES...]' % (sys.argv[0])
  82     parser = optparse.OptionParser(usage=usage)
  83     parser.add_option('--ui', type='choice', choices=UI.keys(), default='auto',
  84                       help='choose the ui between: [auto], gtk, curses, ascii')
  85     parser.add_option('--xdev', action='store_false', dest='cross_device',
  86                       default=True,
  87                       help='ignore directories on other filesystems')
  88     parser.add_option('--max-depth', type='int', dest='max_depth', default=5,
  89                       metavar='DEPTH',
  90                       help='maximum depth of the displayed tree [5]')
  91     parser.add_option('--min-size', type='str', dest='min_size', default='auto',
  92                       metavar='SIZE',
  93                       help='minimum size to consider drawing [auto]')
  94     parser.add_option('--fast-drawing', action='store_true', dest='fast',
  95                       default=False, help='faster but simpler drawing with GTK')
  96     parser.add_option('--profile', action='store_true', dest='profile',
  97                       default=False, help=optparse.SUPPRESS_HELP)
  98     options, args = parser.parse_args()
  99
 100     if options.max_depth < 2:
 101         parser.error('maximum depth must be at least 2')
 102
 103     args = map(os.path.realpath, args)
 104     def continuation():
 105         try:
 106             UI[options.ui](options, args)
 107         except UINotAvailableException:
 108             print 'The interface "%s" is not available' % (options.ui)
 109
 110     if options.profile:
 111         _profile(continuation)
 112     else:
 113         continuation()