| # |
| # BitBake Graphical GTK User Interface |
| # |
| # Copyright (C) 2011-2012 Intel Corporation |
| # |
| # Authored by Joshua Lock <josh@linux.intel.com> |
| # Authored by Dongxiao Xu <dongxiao.xu@intel.com> |
| # Authored by Shane Wang <shane.wang@intel.com> |
| # |
| # 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 glob |
| import gtk |
| import gobject |
| import os |
| import re |
| import shlex |
| import subprocess |
| import tempfile |
| from bb.ui.crumbs.hobwidget import hic, HobButton |
| from bb.ui.crumbs.progressbar import HobProgressBar |
| import bb.ui.crumbs.utils |
| import bb.process |
| from bb.ui.crumbs.hig.crumbsdialog import CrumbsDialog |
| from bb.ui.crumbs.hig.crumbsmessagedialog import CrumbsMessageDialog |
| |
| """ |
| The following are convenience classes for implementing GNOME HIG compliant |
| BitBake GUI's |
| In summary: spacing = 12px, border-width = 6px |
| """ |
| |
| class DeployImageDialog (CrumbsDialog): |
| |
| __dummy_usb__ = "--select a usb drive--" |
| |
| def __init__(self, title, image_path, parent, flags, buttons=None, standalone=False): |
| super(DeployImageDialog, self).__init__(title, parent, flags, buttons) |
| |
| self.image_path = image_path |
| self.standalone = standalone |
| |
| self.create_visual_elements() |
| self.connect("response", self.response_cb) |
| |
| def create_visual_elements(self): |
| self.set_size_request(600, 400) |
| label = gtk.Label() |
| label.set_alignment(0.0, 0.5) |
| markup = "<span font_desc='12'>The image to be written into usb drive:</span>" |
| label.set_markup(markup) |
| self.vbox.pack_start(label, expand=False, fill=False, padding=2) |
| |
| table = gtk.Table(2, 10, False) |
| table.set_col_spacings(5) |
| table.set_row_spacings(5) |
| self.vbox.pack_start(table, expand=True, fill=True) |
| |
| scroll = gtk.ScrolledWindow() |
| scroll.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC) |
| scroll.set_shadow_type(gtk.SHADOW_IN) |
| tv = gtk.TextView() |
| tv.set_editable(False) |
| tv.set_wrap_mode(gtk.WRAP_WORD) |
| tv.set_cursor_visible(False) |
| self.buf = gtk.TextBuffer() |
| self.buf.set_text(self.image_path) |
| tv.set_buffer(self.buf) |
| scroll.add(tv) |
| table.attach(scroll, 0, 10, 0, 1) |
| |
| # There are 2 ways to use DeployImageDialog |
| # One way is that called by HOB when the 'Deploy Image' button is clicked |
| # The other way is that called by a standalone script. |
| # Following block of codes handles the latter way. It adds a 'Select Image' button and |
| # emit a signal when the button is clicked. |
| if self.standalone: |
| gobject.signal_new("select_image_clicked", self, gobject.SIGNAL_RUN_FIRST, |
| gobject.TYPE_NONE, ()) |
| icon = gtk.Image() |
| pix_buffer = gtk.gdk.pixbuf_new_from_file(hic.ICON_IMAGES_DISPLAY_FILE) |
| icon.set_from_pixbuf(pix_buffer) |
| button = gtk.Button("Select Image") |
| button.set_image(icon) |
| #button.set_size_request(140, 50) |
| table.attach(button, 9, 10, 1, 2, gtk.FILL, 0, 0, 0) |
| button.connect("clicked", self.select_image_button_clicked_cb) |
| |
| separator = gtk.HSeparator() |
| self.vbox.pack_start(separator, expand=False, fill=False, padding=10) |
| |
| self.usb_desc = gtk.Label() |
| self.usb_desc.set_alignment(0.0, 0.5) |
| markup = "<span font_desc='12'>You haven't chosen any USB drive.</span>" |
| self.usb_desc.set_markup(markup) |
| |
| self.usb_combo = gtk.combo_box_new_text() |
| self.usb_combo.connect("changed", self.usb_combo_changed_cb) |
| model = self.usb_combo.get_model() |
| model.clear() |
| self.usb_combo.append_text(self.__dummy_usb__) |
| for usb in self.find_all_usb_devices(): |
| self.usb_combo.append_text("/dev/" + usb) |
| self.usb_combo.set_active(0) |
| self.vbox.pack_start(self.usb_combo, expand=False, fill=False) |
| self.vbox.pack_start(self.usb_desc, expand=False, fill=False, padding=2) |
| |
| self.progress_bar = HobProgressBar() |
| self.vbox.pack_start(self.progress_bar, expand=False, fill=False) |
| separator = gtk.HSeparator() |
| self.vbox.pack_start(separator, expand=False, fill=True, padding=10) |
| |
| self.vbox.show_all() |
| self.progress_bar.hide() |
| |
| def set_image_text_buffer(self, image_path): |
| self.buf.set_text(image_path) |
| |
| def set_image_path(self, image_path): |
| self.image_path = image_path |
| |
| def popen_read(self, cmd): |
| tmpout, errors = bb.process.run("%s" % cmd) |
| return tmpout.strip() |
| |
| def find_all_usb_devices(self): |
| usb_devs = [ os.readlink(u) |
| for u in glob.glob('/dev/disk/by-id/usb*') |
| if not re.search(r'part\d+', u) ] |
| return [ '%s' % u[u.rfind('/')+1:] for u in usb_devs ] |
| |
| def get_usb_info(self, dev): |
| return "%s %s" % \ |
| (self.popen_read('cat /sys/class/block/%s/device/vendor' % dev), |
| self.popen_read('cat /sys/class/block/%s/device/model' % dev)) |
| |
| def select_image_button_clicked_cb(self, button): |
| self.emit('select_image_clicked') |
| |
| def usb_combo_changed_cb(self, usb_combo): |
| combo_item = self.usb_combo.get_active_text() |
| if not combo_item or combo_item == self.__dummy_usb__: |
| markup = "<span font_desc='12'>You haven't chosen any USB drive.</span>" |
| self.usb_desc.set_markup(markup) |
| else: |
| markup = "<span font_desc='12'>" + self.get_usb_info(combo_item.lstrip("/dev/")) + "</span>" |
| self.usb_desc.set_markup(markup) |
| |
| def response_cb(self, dialog, response_id): |
| if response_id == gtk.RESPONSE_YES: |
| lbl = '' |
| msg = '' |
| combo_item = self.usb_combo.get_active_text() |
| if combo_item and combo_item != self.__dummy_usb__ and self.image_path: |
| cmdline = bb.ui.crumbs.utils.which_terminal() |
| if cmdline: |
| tmpfile = tempfile.NamedTemporaryFile() |
| cmdline += "\"sudo dd if=" + self.image_path + \ |
| " of=" + combo_item + " && sync; echo $? > " + tmpfile.name + "\"" |
| subprocess.call(shlex.split(cmdline)) |
| |
| if int(tmpfile.readline().strip()) == 0: |
| lbl = "<b>Deploy image successfully.</b>" |
| else: |
| lbl = "<b>Failed to deploy image.</b>" |
| msg = "Please check image <b>%s</b> exists and USB device <b>%s</b> is writable." % (self.image_path, combo_item) |
| tmpfile.close() |
| else: |
| if not self.image_path: |
| lbl = "<b>No selection made.</b>" |
| msg = "You have not selected an image to deploy." |
| else: |
| lbl = "<b>No selection made.</b>" |
| msg = "You have not selected a USB device." |
| if len(lbl): |
| crumbs_dialog = CrumbsMessageDialog(self, lbl, gtk.MESSAGE_INFO, msg) |
| button = crumbs_dialog.add_button("Close", gtk.RESPONSE_OK) |
| HobButton.style_button(button) |
| crumbs_dialog.run() |
| crumbs_dialog.destroy() |
| |
| def update_progress_bar(self, title, fraction, status=None): |
| self.progress_bar.update(fraction) |
| self.progress_bar.set_title(title) |
| self.progress_bar.set_rcstyle(status) |
| |
| def write_file(self, ifile, ofile): |
| self.progress_bar.reset() |
| self.progress_bar.show() |
| |
| f_from = os.open(ifile, os.O_RDONLY) |
| f_to = os.open(ofile, os.O_WRONLY) |
| |
| total_size = os.stat(ifile).st_size |
| written_size = 0 |
| |
| while True: |
| buf = os.read(f_from, 1024*1024) |
| if not buf: |
| break |
| os.write(f_to, buf) |
| written_size += 1024*1024 |
| self.update_progress_bar("Writing to usb:", written_size * 1.0/total_size) |
| |
| self.update_progress_bar("Writing completed:", 1.0) |
| os.close(f_from) |
| os.close(f_to) |
| self.progress_bar.hide() |