summaryrefslogtreecommitdiff
path: root/geneticone/gui
diff options
context:
space:
mode:
Diffstat (limited to 'geneticone/gui')
-rw-r--r--geneticone/gui/__init__.py3
-rw-r--r--geneticone/gui/main.py558
2 files changed, 561 insertions, 0 deletions
diff --git a/geneticone/gui/__init__.py b/geneticone/gui/__init__.py
new file mode 100644
index 0000000..1efeeaf
--- /dev/null
+++ b/geneticone/gui/__init__.py
@@ -0,0 +1,3 @@
+#!/usr/bin/python
+
+from main import *
diff --git a/geneticone/gui/main.py b/geneticone/gui/main.py
new file mode 100644
index 0000000..9398b91
--- /dev/null
+++ b/geneticone/gui/main.py
@@ -0,0 +1,558 @@
+#!/usr/bin/python
+
+import geneticone
+
+import pygtk
+pygtk.require("2.0")
+import gtk
+
+# for doing emerge
+from subprocess import *
+
+# for the terminal
+import pty
+import vte
+
+from portage_util import unique_array
+
+class EmergeQueue:
+ """This class manages the emerge queue."""
+
+ def __init__ (self, tree = None, console = None):
+ """"tree" is a gtk.TreeStore to show the queue in; "console" is a vte.Terminal to print the output to."""
+ self.mergequeue = {}
+ self.unmergequeue = []
+ self.tree = tree
+ self.console = console
+
+ if self.tree:
+ self.emergeIt = self.tree.append(None, ["Emerge"])
+ self.unmergeIt = self.tree.append(None, ["Unmerge"])
+ else:
+ self.emergeIt = self.unmergeIt = None
+
+ def append (self, sth, unmerge = False):
+ """Appends a cpv either to the merge queue or to the unmerge-queue.
+ Also update the tree-view."""
+ if not unmerge:
+ # insert dependencies
+ pkg = geneticone.find_packages("="+sth)[0]
+ try:
+ self.mergequeue.update({sth : pkg.get_dep_packages()})
+ except geneticone.BlockedException, e :
+ blocks = e[0]
+ blocked_dialog(sth, blocks)
+ return
+ else:
+ # update tree
+ if self.emergeIt:
+ pkgIt = self.tree.append(self.emergeIt, [sth])
+ for p in self.mergequeue[sth]:
+ self.tree.append(pkgIt, [p])
+ else:
+ self.unmergequeue.append(sth)
+ if self.unmergeIt: # update tree
+ self.tree.append(self.unmergeIt, [sth])
+
+ def _emerge (self, options, it):
+ """Calls emerge and updates the terminal."""
+ # open pty
+ (master, slave) = pty.openpty()
+ self.console.set_pty(master)
+ self.process = Popen("emerge "+options, stdout = slave, stderr = STDOUT, shell = True)
+ self.remove_all(it)
+
+ def emerge (self, force = False):
+ """Emerges everything in the merge-queue. If force is 'False' (default) only 'emerge -pv' is called."""
+ if len(self.mergequeue) == 0: return
+
+ list = ""
+ for k in self.mergequeue.keys():
+ list += " '="+k+"'"
+
+ s = ""
+ print list
+ if not force: s = "-pv "
+ self._emerge(s+list, self.emergeIt)
+
+ def unmerge (self, force = False):
+ """Unmerges everything in the umerge-queue. If force is 'False' (default) only "emerge -pv -C" is called."""
+ if len(self.unmergequeue) == 0: return
+
+ list = " ".join(self.unmergequeue)
+ s = ""
+ if not force: s = "-pv "
+ self._emerge("-C "+s+list, self.unmergeIt)
+
+ def remove_all (self, parentIt):
+ """Removes all children of a given parent TreeIter."""
+ childIt = self.tree.iter_children(parentIt)
+
+ while childIt:
+ temp = childIt
+ childIt = self.tree.iter_next(childIt)
+ self.remove(temp)
+
+ def remove (self, it):
+ if self.tree.iter_parent(it): # NEVER remove our top stuff
+ if self.tree.get_string_from_iter(it).split(":")[0] == self.tree.get_string_from_iter(self.emergeIt):
+ del self.mergequeue[self.tree.get_value(it,0)]
+ else:
+ self.unmergequeue.remove(self.tree.get_value(it,0))
+
+ self.tree.remove(it)
+
+class PackageWindow:
+ """A window with data about a specfic package."""
+
+ def cb_changed (self, combo, data = None):
+ """Callback for the changed ComboBox.
+ It then rebuilds the useList and the checkboxes."""
+ # remove old useList
+ self.useListScroll.remove(self.useList)
+
+ # build new
+ self.useList = self.build_use_list()
+ self.useListScroll.add(self.useList)
+ self.update_checkboxes()
+
+ self.useListScroll.set_policy(gtk.POLICY_NEVER, gtk.POLICY_NEVER)
+
+ # set emerge-button-label
+ if not self.actual_package().is_installed():
+ self.emergeBtn.set_label("_Emerge")
+ else:
+ self.emergeBtn.set_label("_Unmerge")
+
+ # refresh - make window as small as possible
+ self.table.show_all()
+ self.window.resize(1,1)
+ return True
+
+ def build_vers_combo (self):
+ """Creates the combo box with the different versions."""
+ combo = gtk.combo_box_new_text()
+
+ # append versions
+ for s in [x.get_version() for x in self.packages]:
+ combo.append_text(s)
+
+ # activate the first one
+ try:
+ best_version = ""
+ if self.version:
+ best_version = self.version
+ else:
+ best_version = geneticone.find_best_match(self.packages[0].get_cp(), (self.instPackages != [])).get_version()
+ for i in range(len(self.packages)):
+ if self.packages[i].get_version() == best_version:
+ combo.set_active(i)
+ break
+ except AttributeError: # no package found
+ combo.set_active(0)
+
+ combo.connect("changed", self.cb_changed)
+
+ return combo
+
+ def actual_package (self):
+ """Returns the actual package (a geneticone.Package-object)."""
+ return self.packages[self.vCombo.get_active()]
+
+ def cb_button_pressed (self, b, event, data = None):
+ """Callback for pressed checkboxes. Just quits the event-loop - no redrawing."""
+ b.emit_stop_by_name("button-press-event")
+ return True
+
+ def cb_emerge_clicked (self, button, data = None):
+ """Adds the package to the EmergeQueue."""
+ if not geneticone.am_i_root():
+ errorMB = gtk.MessageDialog(self.window, gtk.DIALOG_MODAL, gtk.MESSAGE_ERROR, gtk.BUTTONS_OK, "You cannot (un)merge without being root.")
+ errorMB.run()
+ errorMB.destroy()
+ else:
+ unmerge = self.actual_package().is_installed()
+ self.queue.append(self.actual_package().get_cpv(), unmerge)
+ self.window.destroy()
+ return True
+
+ def update_checkboxes (self):
+ """Updates the checkboxes."""
+ self.installedCheck.set_active(self.actual_package().is_installed())
+ self.maskedCheck.set_active(self.actual_package().is_masked())
+ self.testingCheck.set_active((self.actual_package().get_mask_status() % 3) == 1)
+
+ def build_use_list (self):
+ """Builds the useList."""
+ store = gtk.ListStore(bool, str, str)
+
+ pkg = self.actual_package()
+ for use in pkg.get_all_useflags():
+ if pkg.is_installed() and use in pkg.get_set_useflags(): # flags set during install
+ enabled = True
+ elif (not pkg.is_installed()) and use in pkg.get_settings("USE").split(): # flags that would be set
+ enabled = True
+ else:
+ enabled = False
+ store.append([enabled, use, geneticone.get_use_desc(use, self.cp)])
+
+ # build view
+ view = gtk.TreeView(store)
+ cell = gtk.CellRendererText()
+ tCell = gtk.CellRendererToggle()
+ view.append_column(gtk.TreeViewColumn("Enabled", tCell, active = 0))
+ view.append_column(gtk.TreeViewColumn("Flags", cell, text = 1))
+ view.append_column(gtk.TreeViewColumn("Description", cell, text = 2))
+
+ if store.iter_n_children(None) == 0:
+ view.set_child_visible(False)
+ else:
+ view.set_child_visible(True)
+ return view
+
+ def cb_size_check (self, event, data = None):
+ if self.useListScroll:
+ width, height = self.window.get_size()
+ if height > gtk.gdk.screen_height():
+ self.window.set_default_size(width, gtk.gdk.screen_height())
+ self.window.resize(width, gtk.gdk.screen_height())
+ self.useListScroll.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
+
+ return False
+
+ def __init__ (self, parent, cp, queue = None, version = None):
+ """Build up window contents."""
+ self.parent = parent # parent window
+ self.cp = cp # category/package
+ self.version = version # version - if not None this is used
+ self.queue = queue
+
+ # window
+ self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
+ self.window.set_title(cp)
+ self.window.set_modal(True)
+ self.window.set_transient_for(parent)
+ self.window.set_destroy_with_parent(True)
+ self.window.set_resizable(False)
+ self.window.set_default_size(1,1) # as small as possible
+ self.window.connect("delete-event", lambda a,b: False)
+ #self.window.connect("configure-event", self.cbSizeCheck)
+
+ # packages and installed packages
+ self.packages = geneticone.sort_package_list(geneticone.find_packages(cp, masked=True))
+ self.instPackages = geneticone.sort_package_list(geneticone.find_installed_packages(cp, masked=True))
+
+ # main structure - the table
+ self.table = gtk.Table(rows=4,columns=2)
+ self.window.add(self.table)
+
+ # version-combo-box
+ self.vCombo = self.build_vers_combo()
+ self.table.attach(self.vCombo, 0, 1, 1, 2, yoptions = gtk.FILL)
+
+ # the label (must be here, because it depends on the combo box)
+ desc = self.actual_package().get_env_var("DESCRIPTION")
+ use_markup = True
+ if not desc:
+ desc = "<no description>"
+ use_markup = False
+ else:
+ desc = "<b>"+desc+"</b>"
+ self.descLabel = gtk.Label(desc)
+ self.descLabel.set_line_wrap(True)
+ self.descLabel.set_justify(gtk.JUSTIFY_CENTER)
+ self.descLabel.set_use_markup(use_markup)
+ self.table.attach(self.descLabel, 0, 2, 0, 1, xoptions = gtk.FILL, ypadding = 10)
+
+ # the check boxes
+ checkHB = gtk.HBox (True, 1)
+ self.table.attach(checkHB, 1, 2, 1, 2, yoptions = gtk.FILL)
+
+ self.installedCheck = gtk.CheckButton()
+ self.installedCheck.connect("button-press-event", self.cb_button_pressed)
+ self.installedCheck.set_label("Installed")
+ checkHB.pack_start(self.installedCheck, True, False)
+
+ self.maskedCheck = gtk.CheckButton()
+ self.maskedCheck.connect("button-press-event", self.cb_button_pressed)
+ self.maskedCheck.set_label("Masked")
+ checkHB.pack_start(self.maskedCheck, True, False)
+
+ self.testingCheck = gtk.CheckButton()
+ self.testingCheck.connect("button-press-event", self.cb_button_pressed)
+ self.testingCheck.set_label("Testing")
+ checkHB.pack_start(self.testingCheck, True, False)
+
+ # use list
+ self.useList = self.build_use_list()
+ self.useListScroll = gtk.ScrolledWindow()
+ self.useListScroll.add(self.useList)
+ self.useListScroll.set_policy(gtk.POLICY_NEVER, gtk.POLICY_NEVER) # XXX: make this work correctly
+ self.table.attach(self.useListScroll, 0, 2, 2, 3, ypadding = 10)
+
+ # buttons
+ buttonHB = gtk.HButtonBox()
+ buttonHB.set_layout(gtk.BUTTONBOX_SPREAD)
+ self.table.attach(buttonHB, 0, 2, 3, 4)
+
+ self.emergeBtn = gtk.Button("_Emerge")
+ if not self.queue: self.emergeBtn.set_sensitive(False)
+ self.cancelBtn = gtk.Button("_Cancel")
+ self.cancelBtn.connect("clicked", lambda x: self.window.destroy())
+ self.emergeBtn.connect("clicked", self.cb_emerge_clicked)
+ buttonHB.pack_start(self.emergeBtn)
+ buttonHB.pack_start(self.cancelBtn)
+
+ # current status
+ self.cb_changed(self.vCombo)
+
+ # show
+ self.window.show_all()
+
+class MainWindow:
+ """Application main window."""
+
+ def cb_delete (self, widget, data = None):
+ """Returns false -> window is deleted."""
+ return False
+
+ def cb_destroy (self, widget, data = None):
+ """Calls main_quit()."""
+ gtk.main_quit()
+
+ def create_main_menu (self):
+ """Creates the main menu."""
+ # the menu-list
+ mainMenuDesc = [
+ ( "/_File", None, None, 0, "<Branch>"),
+ ( "/File/_Close", None, self.cb_destroy, 0, "")
+ ]
+ self.itemFactory = gtk.ItemFactory(gtk.MenuBar, "<main>", None)
+ self.itemFactory.create_items(mainMenuDesc)
+ return self.itemFactory.get_widget("<main>")
+
+ def cb_cat_list_selection (self, view, data = None):
+ """Callback for a category-list selection. Updates the package list with these packages in the category."""
+ if view == self.catList: # be sure it is the catList
+ # get the selected category
+ sel = view.get_selection()
+ store, it = sel.get_selected()
+
+ if it:
+ # remove old one
+ self.scroll_2.remove(self.pkgList)
+
+ # create new package list
+ self.pkgList = self.create_pkg_list(store.get_value(it,0))
+ self.scroll_2.add(self.pkgList)
+ self.scroll_2.show_all()
+ return False
+
+ def cb_row_activated (self, view, path, col, store = None):
+ """Callback for an activated row in the pkgList. Opens a package window."""
+ if view == self.pkgList:
+ package = store.get_value(store.get_iter(path), 0)
+ if package[-1] == '*': package = package[:-1]
+ PackageWindow(self.window, self.selCatName+"/"+package, self.queue)
+ elif view == self.emergeView:
+ if len(path) > 1:
+ package = store.get_value(store.get_iter(path), 0)
+ cat, name, vers, rev = geneticone.split_package_name(package)
+ PackageWindow(self.window, cat+"/"+name, queue = None, version = vers+"-"+rev)
+ return True
+
+ def create_cat_list (self):
+ """Creates the category list."""
+ store = gtk.ListStore(str)
+
+ # build categories
+ for p in geneticone.list_categories():
+ store.append([p])
+ # sort them alphabetically
+ store.set_sort_column_id(0, gtk.SORT_ASCENDING)
+
+ view = gtk.TreeView(store)
+ cell = gtk.CellRendererText()
+ col = gtk.TreeViewColumn("Categories", cell, text = 0)
+ view.append_column(col)
+ view.connect("cursor-changed", self.cb_cat_list_selection)
+ view.connect("row-activated", lambda v,p,c : self.cb_cat_list_selection(v))
+ view.set_search_column(0)
+
+ return view
+
+ packages = {} # directory category -> [packages]
+ def create_pkg_list (self, name = None):
+ """Creates the package list. Gets the name of the category."""
+ self.selCatName = name # actual category
+ store = gtk.ListStore(str)
+
+ # calculate packages
+ if name:
+ if name not in self.packages: # only calc packages if not already done
+ self.packages[name] = []
+ for p in unique_array([x.get_name() for x in geneticone.find_all_packages(name+"/")]):
+ if geneticone.find_installed_packages(name+"/"+p, masked=True) != []:
+ p += "*" # append a '*' if the package is installed
+ self.packages[name].append(p)
+
+ for p in self.packages[name]:
+ store.append([p])
+
+ # sort alphabetically
+ store.set_sort_column_id(0, gtk.SORT_ASCENDING)
+
+ # build view
+ pkgList = gtk.TreeView(store)
+ cell = gtk.CellRendererText()
+ col = gtk.TreeViewColumn("Packages", cell, text = 0)
+ pkgList.append_column(col)
+ pkgList.connect("row-activated", self.cb_row_activated, store)
+
+ return pkgList
+
+ def cb_remove_clicked (self, button, data = None):
+ selected = self.emergeView.get_selection()
+
+ if selected:
+ model, iter = selected.get_selected()
+
+ if not model.iter_parent(iter): # top-level
+ if model.iter_n_children(iter) > 0: # and has children which can be removed :)
+ askMB = gtk.MessageDialog(self.window, gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, "Do you really want to clear the whole queue?")
+ if askMB.run() == gtk.RESPONSE_YES :
+ self.queue.remove_all(iter)
+ askMB.destroy()
+ elif model.iter_parent(model.iter_parent(iter)): # this is in the 3rd level => dependency
+ infoMB = gtk.MessageDialog(self.window, gtk.DIALOG_MODAL, gtk.MESSAGE_INFO, gtk.BUTTONS_OK, "You cannot remove dependencies. :)")
+ infoMB.run()
+ infoMB.destroy()
+ else:
+ self.queue.remove(iter)
+
+ return True
+
+ def cb_emerge_clicked (self, button, data = None):
+ if button == self.emergeBtn:
+ self.queue.emerge(force=True)
+ elif button == self.unmergeBtn:
+ self.queue.unmerge(force=True)
+
+ return True
+
+ def cb_search_clicked (self, button, data = None):
+ if self.searchEntry.get_text() != "":
+ packages = geneticone.find_all_packages(self.searchEntry.get_text())
+ packages = unique_array([p.get_cp() for p in packages])
+ dialog = gtk.MessageDialog(None, gtk.DIALOG_MODAL, gtk.MESSAGE_INFO, gtk.BUTTONS_OK, str(packages))
+ dialog.run()
+ dialog.destroy()
+
+ def __init__ (self):
+ """Build up window"""
+
+ # window
+ self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
+ self.window.set_title("Genetic/One")
+ self.window.connect("delete_event", self.cb_delete)
+ self.window.connect("destroy", self.cb_destroy)
+ self.window.set_border_width(2)
+ self.window.set_geometry_hints (self.window, min_width = 600, min_height = 800, max_height = gtk.gdk.screen_height(), max_width = gtk.gdk.screen_width())
+ self.window.set_resizable(True)
+
+ # main vb
+ vb = gtk.VBox(False, 1)
+ self.window.add(vb)
+
+ # menubar
+ menubar = self.create_main_menu()
+ vb.pack_start(menubar, False)
+
+ # search
+ self.searchEntry = gtk.Entry()
+ self.searchBtn = gtk.Button("_Search")
+ self.searchBtn.connect("clicked", self.cb_search_clicked)
+ hbSearch = gtk.HBox(False, 5)
+ hbSearch.pack_start(self.searchEntry, True, True)
+ hbSearch.pack_start(self.searchBtn, False, False)
+ vb.pack_start(hbSearch, False, False, 5)
+
+ # VPaned holding the lists and the Terminal
+ vpaned = gtk.VPaned()
+ vpaned.set_position(400)
+ vb.pack_start(vpaned, True, True)
+
+ # a HB holding the lists
+ hb = gtk.HBox(True, 5)
+ hbFrame = gtk.Frame()
+ hbFrame.add(hb)
+ hbFrame.set_shadow_type(gtk.SHADOW_IN)
+ vpaned.pack1(hbFrame, shrink = True, resize = True)
+
+ self.scroll_1 = gtk.ScrolledWindow()
+ self.scroll_2 = gtk.ScrolledWindow()
+ self.scroll_1.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
+ self.scroll_2.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
+ hb.pack_start(self.scroll_1, True, True)
+ hb.pack_start(self.scroll_2, True, True)
+
+ # create cat List
+ self.catList = self.create_cat_list()
+ self.scroll_1.add(self.catList)
+
+ # create pkg list
+ self.pkgList = self.create_pkg_list()
+ self.scroll_2.add(self.pkgList)
+ # queue list
+ queueVB = gtk.VBox(False, 0)
+ hb.pack_start(queueVB, True, True)
+
+ emergeStore = gtk.TreeStore(str)
+ self.emergeView = gtk.TreeView(emergeStore)
+ cell = gtk.CellRendererText()
+ col = gtk.TreeViewColumn("Queue", cell, text = 0)
+ self.emergeView.append_column(col)
+ self.emergeView.connect("row-activated", self.cb_row_activated, emergeStore)
+ queueVB.pack_start(self.emergeView, True, True)
+
+ # buttons right unter the queue list
+ buttonBox = gtk.HButtonBox()
+ queueVB.pack_start(buttonBox, False)
+ self.emergeBtn = gtk.Button("_Emerge")
+ self.emergeBtn.connect("clicked", self.cb_emerge_clicked)
+ self.unmergeBtn = gtk.Button("_Unmerge")
+ self.unmergeBtn.connect("clicked", self.cb_emerge_clicked)
+ self.removeBtn = gtk.Button("_Remove")
+ self.removeBtn.connect("clicked", self.cb_remove_clicked)
+ buttonBox.pack_start(self.emergeBtn)
+ buttonBox.pack_start(self.removeBtn)
+ buttonBox.pack_start(self.unmergeBtn)
+
+ # the terminal
+ term = vte.Terminal()
+ term.set_scrollback_lines(1024)
+ term.set_scroll_on_output(True)
+ termBox = gtk.HBox(False, 0)
+ termScroll = gtk.VScrollbar(term.get_adjustment())
+ termBox.pack_start(term, True, True)
+ termBox.pack_start(termScroll, False)
+
+ termFrame = gtk.Frame("Console")
+ termFrame.set_shadow_type(gtk.SHADOW_IN)
+ termFrame.add(termBox)
+ vpaned.pack2(termFrame, shrink = True, resize = True)
+
+ # show
+ self.window.show_all()
+
+ # set emerge queue
+ self.queue = EmergeQueue(console=term, tree = emergeStore)
+
+ def main (self):
+ """Main."""
+ gtk.main()
+
+def blocked_dialog (blocked, blocks):
+ dialog = gtk.MessageDialog(None, gtk.DIALOG_MODAL, gtk.MESSAGE_ERROR, gtk.BUTTONS_OK, blocked+" is blocked by "+blocks+".\nPlease unmerge the blocking package.")
+ dialog.run()
+ dialog.destroy()