4 from ui
.ascii
import ui_ascii
5 from ui
.curses
import ui_curses
6 from ui
.gtk
import ui_gtk
8 def _ui_auto(options
, args
):
9 if ui_gtk
.is_available():
10 return ui_gtk
.run(options
, args
)
12 if ui_curses
.is_available():
13 return ui_curses
.run(options
, args
)
15 if ui_ascii
.is_available():
16 return ui_ascii
.run(options
, args
)
18 print 'No UI available'
20 UI
= {'ascii': ui_ascii
.run
, 'curses': ui_curses
.run
,
21 'gtk': ui_gtk
.run
, 'auto': _ui_auto
}
24 parser
= optparse
.OptionParser(usage
='pysize [OPTIONS] DIRECTORY')
25 parser
.add_option('--ui', type='choice', choices
=UI
.keys(), default
='auto')
26 parser
.add_option('--max-depth', type='int', dest
='max_depth', default
=5)
27 (options
, args
) = parser
.parse_args()
29 print parser
.error('A single argument is expected')
31 UI
[options
.ui
](options
, args
)
34 import hotshot
, hotshot
.stats
35 prof
= hotshot
.Profile('pysize.prof')
38 stats
= hotshot
.stats
.load('pysize.prof')
40 stats
.sort_stats('time', 'calls')
43 if __name__
== '__main__':
45 # Try to use psyco if available