pysize.py

   1 #!/usr/bin/env python
   2
   3 import optparse
   4 from ui.ascii import ui_ascii
   5 from ui.curses import ui_curses
   6 from ui.gtk import ui_gtk
   7
   8 def _ui_auto(options, args):
   9     """Automatically choose the best available UI."""
  10     if ui_gtk.is_available():
  11         return ui_gtk.run(options, args)
  12
  13     if ui_curses.is_available():
  14         return ui_curses.run(options, args)
  15
  16     if ui_ascii.is_available():
  17         return ui_ascii.run(options, args)
  18
  19     print 'No UI available'
  20
  21 UI = {'ascii': ui_ascii.run, 'curses': ui_curses.run,
  22       'gtk': ui_gtk.run, 'auto': _ui_auto}
  23
  24 def _main():
  25     parser = optparse.OptionParser(usage='pysize [OPTIONS] DIRECTORY')
  26     parser.add_option('--ui', type='choice', choices=UI.keys(), default='auto')
  27     parser.add_option('--max-depth', type='int', dest='max_depth', default=5)
  28     (options, args) = parser.parse_args()
  29     if len(args) != 1:
  30         print parser.error('A single argument is expected')
  31
  32     UI[options.ui](options, args)
  33
  34 def _prof():
  35     try:
  36         import cProfile, pstats
  37         cProfile.run('_main()', 'pysize.prof')
  38         stats = pstats.Stats('pysize.prof')
  39     except ImportError:
  40         import hotshot, hotshot.stats
  41         prof = hotshot.Profile('pysize.prof')
  42         prof.runcall(_main)
  43         prof.close()
  44         stats = hotshot.stats.load('pysize.prof')
  45     stats.strip_dirs()
  46     stats.sort_stats('time', 'calls')
  47     stats.print_stats(40)
  48
  49 if __name__ == '__main__':
  50     try:
  51         # Try to use psyco if available
  52         import psyco
  53         psyco.full()
  54     except ImportError:
  55         pass
  56     _main()