0ea1a0e6f9f3902af4d66e25d986a2cf97e35c7f
[findit] / src / pkgsearch.py
1 #!/usr/bin/python
2 # -*-coding: utf-8 -*-
3 # vim: sw=4 ts=4 expandtab ai
4
5 import gtk
6 import apt_pkg
7 from heapq import nlargest
8 from misc import size_hum_read
9
10 class Search_Pkg_Control(object):
11
12     def __init__(self, config, show_funct):
13         win_width = config.get('window_width')
14         win_height = config.get('window_height')
15         count = config.get('default_count')
16         self.show_func = show_funct
17         self.abstr = Search_Pkg_Abstraction()
18         self.present = Search_Pkg_Presentation(win_width, win_height, count, self.start_search)
19
20     def start_search(self, count, kill_func):
21         pkglist = []
22         for psize, packg in nlargest(count, self.abstr.pkggetter()):
23             pkglist.append([packg, size_hum_read(psize), psize])
24         kill_func()
25         self.show_func.show(pkglist)
26         self.show_func.run()
27
28     def run(self):
29         self.present.run()
30
31
32 class Search_Pkg_Abstraction(object):
33
34     def __init__(self):
35         apt_pkg.InitConfig()
36         apt_pkg.InitSystem()
37         self.cache = apt_pkg.GetCache()
38
39     def pkggetter(self):
40         for pkg in self.cache.Packages:
41             # pkg is from a list of packages, sorted by name.
42             if pkg.CurrentState == apt_pkg.CurStateInstalled:
43                 yield [version.InstalledSize for version in pkg.VersionList][0], pkg.Name
44
45
46 class Search_Pkg_Presentation(gtk.Window):
47
48     def __init__(self, win_width, win_height, count, st_func):
49         """Create MainWindow."""
50         # Создаем новое окно
51         gtk.Window.__init__(self)
52         self.set_default_size(win_width, win_height)
53         self.set_border_width(4)
54         self.connect('delete_event', gtk.main_quit)
55
56         label1 = gtk.Label('Files quantity')
57         # Окошко ввода количества файлов, мин значение=1 макс=65536 по умолчанию 10
58         adj = gtk.Adjustment(count, 1, 65536, 1, 5, 0)
59         self.file_cnt = gtk.SpinButton(adj, 0, 0)
60
61         hbox = gtk.HBox(False, 0)
62         hbox.pack_start(label1, False, False, 5)
63         hbox.pack_start(self.file_cnt, False, False, 0)
64
65         self.butt_start = gtk.Button('Go >')
66         self.butt_start.connect('released', self.start_srch, st_func)
67         
68         self.butt_prev = gtk.Button('< Previous')
69         #self.butt_prev.connect('released', self.show_prev, prev_window)
70         
71         hbox1 = gtk.HBox(False, 0)
72         hbox1.pack_start(self.butt_prev, False, False, 5)
73         hbox1.pack_start(self.butt_start, False, False, 5)
74
75         vbox = gtk.VBox(False, 0)
76         vbox.pack_start(hbox, False, False, 5)
77         vbox.pack_start(hbox1, False, False, 5)
78
79         self.add(vbox)
80
81     def kill_wind(self):
82         self.destroy()
83         gtk.main_quit()
84
85     def get_data(self):
86         return int( self.file_cnt.get_value() )
87
88     def run(self):
89         self.show_all()
90         gtk.main()
91
92     def start_srch(self, widget, start_func):
93         start_func(self.get_data(), self.kill_wind)