07dfb9e66d71e95794fc57cb532b3bcb4276c481
[findit] / src / main.py
1 #!/usr/bin/env python
2 # -*-coding: utf-8 -*-
3 # vim: sw=4 ts=4 expandtab ai
4 # main.py --search files -o outtable -p ". 3"
5
6 import sys
7
8 from config import Config
9
10 __progname__ = 'FindIT'
11 __version__ = '0.2.0'
12
13 #==============================================================================
14
15 class Control(object):
16     def __init__(self):
17         config = Config ###()
18
19         self.abstrac = Abstraction()
20
21         if(len(sys.argv) > 1):
22             Cli_Presentation(self.abstrac)  ###
23         else:
24             Gtk_Presentation(config, self.abstrac)  ###
25
26 #==============================================================================
27
28 class Abstraction(object):
29     def __init__(self):
30         self.progname = __progname__
31         self.version = __version__
32         self.authors = [    'Alex Taker\n   * Email: alteker@gmail.com\n',
33                         'Eugene Gagarin\n   * Email: mosfet07@ya.ru\n',
34                         'Alexandr Popov\n   * Email: popov2al@gmail.com' ]
35         self.comments = 'Tool for find some information on computer.'
36         self.license = \
37 'This program is free software; you can redistribute it and/or\nmodify it \
38 under the terms of the GNU General Public License\nas published by the Free \
39 Software Foundation; either version 3\nof the License, or (at your option) \
40 any later version.'
41
42 #==============================================================================
43
44 class Cli_Presentation(object):
45
46     def __init__(self, abstrac):
47         from optparse import OptionParser
48         import sys
49
50         self.abstrac = abstrac
51
52         parser = OptionParser(version=__progname__ + ' ' + __version__)
53         parser.add_option('--search', '-s', dest='search', type='string')
54         parser.add_option('--output', '-o', dest='output', type='string')
55         parser.add_option('--params', '-p', dest='params', type='string')
56         parser.add_option('--about',   action='callback', callback=self._about)
57         parser.add_option('--license', action='callback', callback=self._license)
58         (options, args) = parser.parse_args()
59
60         config = {}
61         config['search'] = options.search
62         config['outtype'] = options.output
63         config['ignore_dirs'] = ['/dev', '/proc', '/sys', '/mnt']
64         config['start_path'], config['count'] = options.params.split()
65
66         self.show_search_toplevel(config)
67
68     def _about(self, *a):
69         print self.abstrac.comments
70         sys.exit()
71
72     def _license(self, *a):
73         print self.abstrac.license
74         sys.exit()
75
76     def show_search_toplevel(self, config):
77         search_module = __import__(config['search'] + '.search')
78         search_toplevel = search_module.search.Control('cli', config).run()
79
80 #==============================================================================
81
82 class Gtk_Presentation(object):
83     """Main window class."""
84
85     def __init__(self, config, abstrac):
86         import gtk
87
88         self.config = config
89
90         def _create_menu():
91             """Create main menu."""
92             menubar = gtk.MenuBar()
93             fileitem = gtk.MenuItem( '_File' )
94             viewitem = gtk.MenuItem( '_View' )
95             helpitem = gtk.MenuItem( '_Help' )
96             helpitem.connect('activate', about_dialog)
97             menubar.add(fileitem)
98             menubar.add(viewitem)
99             menubar.add(helpitem)
100             return menubar
101
102         def _create_toolbar():
103             """Create toolbar."""
104             toolbar = gtk.Toolbar()
105             filesearch_tbtn = gtk.RadioToolButton(None)
106             debsearch_tbtn = gtk.RadioToolButton(filesearch_tbtn)
107
108             filesearch_tbtn.set_name('files')
109             debsearch_tbtn.set_name('debs')
110
111             filesearch_tbtn.set_label('Files search')
112             debsearch_tbtn.set_label('Debs search')
113
114             filesearch_tbtn.connect('clicked', self.show_search_toplevel, 'files')
115             debsearch_tbtn.connect('clicked', self.show_search_toplevel, 'debs')
116
117             toolbar.insert(filesearch_tbtn, -1)
118             toolbar.insert(debsearch_tbtn, -1)
119
120             search_tbtns = [filesearch_tbtn, debsearch_tbtn]
121
122             # Activate radio tool button
123             for btn in search_tbtns:
124                 if btn.get_name() == self.config['search']:
125                     btn.set_active(True)
126
127             return toolbar
128
129         def about_dialog(widget):
130             """About dialog window."""
131             dialog = gtk.AboutDialog()
132             dialog.set_name(abstrac.progname)
133             dialog.set_version(abstrac.version)
134             dialog.set_authors(abstrac.authors)
135             dialog.set_comments(abstrac.comments)
136             dialog.set_license(abstrac.license)
137             dialog.show_all()
138             dialog.run()
139             dialog.destroy()
140
141         window = gtk.Window()
142         window.set_default_size(560, 400)
143         window.set_geometry_hints(None, 560, 400)
144         window.set_border_width(4)
145         window.set_wmclass('MainWindow', 'FindIT')
146         window.connect('destroy', gtk.main_quit)
147
148         menu = _create_menu()
149         toolbar = _create_toolbar()
150
151         self.vbox = gtk.VBox(False, 4)
152         self.vbox.pack_start(menu, False, False, 0)
153         self.vbox.pack_start(toolbar, False, False, 0)
154         self.show_search_toplevel(None, self.config['search'])
155
156         window.add(self.vbox)
157         window.show_all()
158         gtk.main()
159
160     #=== Search selecting =====================================================
161     def show_search_toplevel(self, btn, searchtype):
162         print 'Entering <' + searchtype + '> search mode...'
163
164         search_module = __import__(searchtype + '.search')
165         search_toplevel = search_module.search.Control('gtk', self.config).run()
166
167         try:
168             self.vbox.remove(self.vbox.get_children()[2])
169         except:
170             pass
171         self.vbox.pack_start(search_toplevel, True, True, 0)
172         search_toplevel.show_all()
173
174 #==============================================================================
175
176 class Hildon_Presentation(object):
177     """Main window class."""
178
179     def __init__(self, config):
180         import gtk
181         import hildon
182
183         self.config = config
184
185 #==============================================================================
186
187 if __name__ == '__main__':
188     Control()