#!/usr/bin/env python # # BitBake Graphical GTK User Interface # # Copyright (C) 2012 Intel Corporation # # Authored by Dongxiao Xu # Authored by Shane Wang # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License version 2 as # published by the Free Software Foundation. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License along # with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. import gtk import glib from bb.ui.crumbs.hobcolor import HobColors from bb.ui.crumbs.hobwidget import HobViewTable, HobNotebook, HobAltButton, HobButton from bb.ui.crumbs.hoblistmodel import PackageListModel from bb.ui.crumbs.hobpages import HobPage # # PackageSelectionPage # class PackageSelectionPage (HobPage): pages = [ { 'name' : 'Included packages', 'tooltip' : 'The packages currently included for your image', 'filter' : { PackageListModel.COL_INC : [True] }, 'search' : 'Search packages by name', 'searchtip' : 'Enter a package name to find it', 'columns' : [{ 'col_name' : 'Package name', 'col_id' : PackageListModel.COL_NAME, 'col_style': 'text', 'col_min' : 100, 'col_max' : 300, 'expand' : 'True' }, { 'col_name' : 'Size', 'col_id' : PackageListModel.COL_SIZE, 'col_style': 'text', 'col_min' : 100, 'col_max' : 300, 'expand' : 'True' }, { 'col_name' : 'Recipe', 'col_id' : PackageListModel.COL_RCP, 'col_style': 'text', 'col_min' : 100, 'col_max' : 250, 'expand' : 'True' }, { 'col_name' : 'Brought in by (+others)', 'col_id' : PackageListModel.COL_BINB, 'col_style': 'binb', 'col_min' : 100, 'col_max' : 350, 'expand' : 'True' }, { 'col_name' : 'Included', 'col_id' : PackageListModel.COL_INC, 'col_style': 'check toggle', 'col_min' : 100, 'col_max' : 100 }] }, { 'name' : 'All packages', 'tooltip' : 'All packages that have been built', 'filter' : {}, 'search' : 'Search packages by name', 'searchtip' : 'Enter a package name to find it', 'columns' : [{ 'col_name' : 'Package name', 'col_id' : PackageListModel.COL_NAME, 'col_style': 'text', 'col_min' : 100, 'col_max' : 400, 'expand' : 'True' }, { 'col_name' : 'Size', 'col_id' : PackageListModel.COL_SIZE, 'col_style': 'text', 'col_min' : 100, 'col_max' : 500, 'expand' : 'True' }, { 'col_name' : 'Recipe', 'col_id' : PackageListModel.COL_RCP, 'col_style': 'text', 'col_min' : 100, 'col_max' : 250, 'expand' : 'True' }, { 'col_name' : 'Included', 'col_id' : PackageListModel.COL_INC, 'col_style': 'check toggle', 'col_min' : 100, 'col_max' : 100 }] } ] (INCLUDED, ALL) = range(2) def __init__(self, builder): super(PackageSelectionPage, self).__init__(builder, "Edit packages") # set invisible members self.recipe_model = self.builder.recipe_model self.package_model = self.builder.package_model # create visual elements self.create_visual_elements() def included_clicked_cb(self, button): self.ins.set_current_page(self.INCLUDED) def create_visual_elements(self): self.label = gtk.Label("Packages included: 0\nSelected packages size: 0 MB") self.eventbox = self.add_onto_top_bar(self.label, 73) self.pack_start(self.eventbox, expand=False, fill=False) self.pack_start(self.group_align, expand=True, fill=True) # set visible members self.ins = HobNotebook() self.tables = [] # we need to modify table when the dialog is shown search_names = [] search_tips = [] # append the tab for page in self.pages: columns = page['columns'] name = page['name'] tab = HobViewTable(columns, name) search_names.append(page['search']) search_tips.append(page['searchtip']) filter = page['filter'] sort_model = self.package_model.tree_model(filter, initial=True) tab.set_model(sort_model) tab.connect("toggled", self.table_toggled_cb, name) tab.connect("button-release-event", self.button_click_cb) tab.connect("cell-fadeinout-stopped", self.after_fadeout_checkin_include, filter) self.ins.append_page(tab, page['name'], page['tooltip']) self.tables.append(tab) self.ins.set_entry(search_names, search_tips) self.ins.search.connect("changed", self.search_entry_changed) # add all into the dialog self.box_group_area.pack_start(self.ins, expand=True, fill=True) self.button_box = gtk.HBox(False, 6) self.box_group_area.pack_start(self.button_box, expand=False, fill=False) self.build_image_button = HobButton('Build image') #self.build_image_button.set_size_request(205, 49) self.build_image_button.set_tooltip_text("Build target image") self.build_image_button.set_flags(gtk.CAN_DEFAULT) self.build_image_button.grab_default() self.build_image_button.connect("clicked", self.build_image_clicked_cb) self.button_box.pack_end(self.build_image_button, expand=False, fill=False) self.back_button = HobAltButton('Cancel') self.back_button.connect("clicked", self.back_button_clicked_cb) self.button_box.pack_end(self.back_button, expand=False, fill=False) def search_entry_changed(self, entry): text = entry.get_text() if self.ins.search_focus: self.ins.search_focus = False elif self.ins.page_changed: self.ins.page_change = False self.filter_search(entry) elif text not in self.ins.search_names: self.filter_search(entry) def filter_search(self, entry): text = entry.get_text() current_tab = self.ins.get_current_page() filter = self.pages[current_tab]['filter'] filter[PackageListModel.COL_NAME] = text self.tables[current_tab].set_model(self.package_model.tree_model(filter, search_data=text)) if self.package_model.filtered_nb == 0: if not self.ins.get_nth_page(current_tab).top_bar: self.ins.get_nth_page(current_tab).add_no_result_bar(entry) self.ins.get_nth_page(current_tab).top_bar.set_no_show_all(True) self.ins.get_nth_page(current_tab).top_bar.show() self.ins.get_nth_page(current_tab).scroll.hide() else: if self.ins.get_nth_page(current_tab).top_bar: self.ins.get_nth_page(current_tab).top_bar.hide() self.ins.get_nth_page(current_tab).scroll.show() if entry.get_text() == '': entry.set_icon_sensitive(gtk.ENTRY_ICON_SECONDARY, False) else: entry.set_icon_sensitive(gtk.ENTRY_ICON_SECONDARY, True) def button_click_cb(self, widget, event): path, col = widget.table_tree.get_cursor() tree_model = widget.table_tree.get_model() if path and col.get_title() != 'Included': # else activation is likely a removal properties = {'binb': '' , 'name': '', 'size':'', 'recipe':'', 'files_list':''} properties['binb'] = tree_model.get_value(tree_model.get_iter(path), PackageListModel.COL_BINB) properties['name'] = tree_model.get_value(tree_model.get_iter(path), PackageListModel.COL_NAME) properties['size'] = tree_model.get_value(tree_model.get_iter(path), PackageListModel.COL_SIZE) properties['recipe'] = tree_model.get_value(tree_model.get_iter(path), PackageListModel.COL_RCP) properties['files_list'] = tree_model.get_value(tree_model.get_iter(path), PackageListModel.COL_FLIST) self.builder.show_recipe_property_dialog(properties) def open_log_clicked_cb(self, button, log_file): if log_file: log_file = "file:///" + log_file gtk.show_uri(screen=button.get_screen(), uri=log_file, timestamp=0) def show_page(self, log_file): children = self.button_box.get_children() or [] for child in children: self.button_box.remove(child) # re-packed the buttons as request, add the 'open log' button if build success self.button_box.pack_end(self.build_image_button, expand=False, fill=False) if log_file: open_log_button = HobAltButton("Open log") open_log_button.connect("clicked", self.open_log_clicked_cb, log_file) open_log_button.set_tooltip_text("Open the build's log file") self.button_box.pack_end(open_log_button, expand=False, fill=False) self.button_box.pack_end(self.back_button, expand=False, fill=False) self.show_all() def build_image_clicked_cb(self, button): self.builder.parsing_warnings = [] self.builder.build_image() def refresh_tables(self): self.ins.reset_entry(self.ins.search, 0) for tab in self.tables: index = self.tables.index(tab) filter = self.pages[index]['filter'] tab.set_model(self.package_model.tree_model(filter, initial=True)) def back_button_clicked_cb(self, button): if self.builder.previous_step == self.builder.IMAGE_GENERATED: self.builder.restore_initial_selected_packages() self.refresh_selection() self.builder.show_image_details() else: self.builder.show_configuration() self.refresh_tables() def refresh_selection(self): self.builder.configuration.selected_packages = self.package_model.get_selected_packages() self.builder.configuration.user_selected_packages = self.package_model.get_user_selected_packages() selected_packages_num = len(self.builder.configuration.selected_packages) selected_packages_size = self.package_model.get_packages_size() selected_packages_size_str = HobPage._size_to_string(selected_packages_size) if self.builder.configuration.image_packages == self.builder.configuration.selected_packages: image_total_size_str = self.builder.configuration.image_size else: image_overhead_factor = self.builder.configuration.image_overhead_factor image_rootfs_size = self.builder.configuration.image_rootfs_size / 1024 # image_rootfs_size is KB image_extra_size = self.builder.configuration.image_extra_size / 1024 # image_extra_size is KB base_size = image_overhead_factor * selected_packages_size image_total_size = max(base_size, image_rootfs_size) + image_extra_size if "zypper" in self.builder.configuration.selected_packages: image_total_size += (51200 * 1024) image_total_size_str = HobPage._size_to_string(image_total_size) self.label.set_label("Packages included: %s\nSelected packages size: %s\nEstimated image size: %s" % (selected_packages_num, selected_packages_size_str, image_total_size_str)) self.ins.show_indicator_icon("Included packages", selected_packages_num) def toggle_item_idle_cb(self, path, view_tree, cell, pagename): if not self.package_model.path_included(path): self.package_model.include_item(item_path=path, binb="User Selected") else: self.pre_fadeout_checkout_include(view_tree) self.package_model.exclude_item(item_path=path) self.render_fadeout(view_tree, cell) self.refresh_selection() if not self.builder.customized: self.builder.customized = True self.builder.set_base_image() self.builder.configuration.selected_image = self.recipe_model.__custom_image__ self.builder.rcppkglist_populated() self.builder.window_sensitive(True) view_model = view_tree.get_model() vpath = self.package_model.convert_path_to_vpath(view_model, path) view_tree.set_cursor(vpath) def table_toggled_cb(self, table, cell, view_path, toggled_columnid, view_tree, pagename): # Click to include a package self.builder.window_sensitive(False) view_model = view_tree.get_model() path = self.package_model.convert_vpath_to_path(view_model, view_path) glib.idle_add(self.toggle_item_idle_cb, path, view_tree, cell, pagename) def pre_fadeout_checkout_include(self, tree): #after the fadeout the table will be sorted as before self.sort_column_id = self.package_model.sort_column_id self.sort_order = self.package_model.sort_order self.package_model.resync_fadeout_column(self.package_model.get_iter_first()) # Check out a model which base on the column COL_FADE_INC, # it's save the prev state of column COL_INC before do exclude_item filter = { PackageListModel.COL_FADE_INC : [True]} new_model = self.package_model.tree_model(filter, excluded_items_ahead=True) tree.set_model(new_model) tree.expand_all() def get_excluded_rows(self, to_render_cells, model, it): while it: path = model.get_path(it) prev_cell_is_active = model.get_value(it, PackageListModel.COL_FADE_INC) curr_cell_is_active = model.get_value(it, PackageListModel.COL_INC) if (prev_cell_is_active == True) and (curr_cell_is_active == False): to_render_cells.append(path) if model.iter_has_child(it): self.get_excluded_rows(to_render_cells, model, model.iter_children(it)) it = model.iter_next(it) return to_render_cells def render_fadeout(self, tree, cell): if (not cell) or (not tree): return to_render_cells = [] view_model = tree.get_model() self.get_excluded_rows(to_render_cells, view_model, view_model.get_iter_first()) cell.fadeout(tree, 1000, to_render_cells) def after_fadeout_checkin_include(self, table, ctrl, cell, tree, filter): self.package_model.sort_column_id = self.sort_column_id self.package_model.sort_order = self.sort_order tree.set_model(self.package_model.tree_model(filter)) tree.expand_all() def set_packages_curr_tab(self, curr_page): self.ins.set_current_page(curr_page)