Esempio n. 1
0
    def run(self):
        #
        #    Main text
        #
        text = 'Thank you for reporting your bugs, it helps us improve our'
        text += ' scanning engine. If you want to get involved with the project'
        text += ' please send an email to our <a href="mailto:%s">mailing list</a>.'
        text = text % ('*****@*****.**')
        # All these lines are here to add a label instead of the easy "set_markup"
        # in order to avoid a bug where the label text appears selected
        msg_area = self.get_message_area()
        [msg_area.remove(c) for c in msg_area.get_children()]
        label = gtk.Label()
        label.set_markup( text )
        label.set_line_wrap(True)
        label.select_region(0, 0)
        msg_area.pack_start(label)
        

        self.worker = bug_report_worker(self.bug_report_function, self.bugs_to_report)
        self.worker.start()        
        gobject.timeout_add(200, self.add_result_from_worker)
        
        self.status_hbox = gtk.HBox()
        
        #
        #    Empty markup for the ticket ids
        #
        self.link_label = gtk.Label( '' )
        self.link_label.set_line_wrap(True)
        self.link_label.set_use_markup( True )
        self.status_hbox.pack_end(self.link_label)
        
        #
        #    Throbber, only show while still running.
        #
        
        self.throbber = Throbber()
        self.throbber.running(True)
        self.status_hbox.pack_end(self.throbber)
        
        #
        #    Check, hidden at the beginning 
        #    http://www.pygtk.org/docs/pygtk/gtk-stock-items.html
        #
        self.done_icon = gtk.Image()
        self.done_icon.set_from_stock(gtk.STOCK_YES, gtk.ICON_SIZE_BUTTON)
        self.status_hbox.pack_end(self.done_icon)
        
        self.vbox.pack_start(self.status_hbox, True, True)

        self.add(self.vbox)
        self.show_all()
        self.done_icon.hide()
        
        super(report_bug_show_result, self).run()
        self.destroy()
        
        return self.reported_ids
Esempio n. 2
0
class report_bug_show_result(gtk.MessageDialog):
    '''
    A class that shows the result of one or more bug reports to the user. The
    window shows a "Thanks" message and links to the bugs that were generated.
    
    Unlike previous versions, this window actually sends the bugs to the network
    since we want to show the ticket IDs "in real time" to the user instead of
    reporting them all and then showing a long list of items.
    '''
    
    def __init__(self, bug_report_function, bugs_to_report):
        '''
        @param bug_report_function: The function that's used to report bugs.
                                    apply(bug_report_function, bug_to_report)
                                    
        @param bugs_to_report: An iterable with the bugs to report. These are
                               going to be the parameters for the bug_report_function.
        '''
        gtk.MessageDialog.__init__(self,
                                   None,
                                   gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
                                   gtk.MESSAGE_INFO,
                                   gtk.BUTTONS_OK,
                                   None)
        
        self.bug_report_function = bug_report_function
        self.bugs_to_report = bugs_to_report
        self.ticket_ids_in_markup = 0
        self.reported_ids = []
        
        self.set_title('Bug report results')
        self.set_icon_from_file(W3AF_ICON)

        # Disable OK button until the worker finishes the bug reporting process
        ok_button = self.get_widget_for_response(gtk.RESPONSE_OK)
        ok_button.set_sensitive(False)
        
        
    def run(self):
        #
        #    Main text
        #
        text = 'Thank you for reporting your bugs, it helps us improve our'
        text += ' scanning engine. If you want to get involved with the project'
        text += ' please send an email to our <a href="mailto:%s">mailing list</a>.'
        text = text % ('*****@*****.**')
        # All these lines are here to add a label instead of the easy "set_markup"
        # in order to avoid a bug where the label text appears selected
        msg_area = self.get_message_area()
        [msg_area.remove(c) for c in msg_area.get_children()]
        label = gtk.Label()
        label.set_markup( text )
        label.set_line_wrap(True)
        label.select_region(0, 0)
        msg_area.pack_start(label)
        

        self.worker = bug_report_worker(self.bug_report_function, self.bugs_to_report)
        self.worker.start()        
        gobject.timeout_add(200, self.add_result_from_worker)
        
        self.status_hbox = gtk.HBox()
        
        #
        #    Empty markup for the ticket ids
        #
        self.link_label = gtk.Label( '' )
        self.link_label.set_line_wrap(True)
        self.link_label.set_use_markup( True )
        self.status_hbox.pack_end(self.link_label)
        
        #
        #    Throbber, only show while still running.
        #
        
        self.throbber = Throbber()
        self.throbber.running(True)
        self.status_hbox.pack_end(self.throbber)
        
        #
        #    Check, hidden at the beginning 
        #    http://www.pygtk.org/docs/pygtk/gtk-stock-items.html
        #
        self.done_icon = gtk.Image()
        self.done_icon.set_from_stock(gtk.STOCK_YES, gtk.ICON_SIZE_BUTTON)
        self.status_hbox.pack_end(self.done_icon)
        
        self.vbox.pack_start(self.status_hbox, True, True)

        self.add(self.vbox)
        self.show_all()
        self.done_icon.hide()
        
        super(report_bug_show_result, self).run()
        self.destroy()
        
        return self.reported_ids
        
      
    def add_result_from_worker(self):
        '''
        Adds the results from the worker to the text that's shown in the window
        '''
        # The links to the reported tickets
        try:
            bug_report_result = self.worker.output.get(block=False)
        except Queue.Empty:
            # The worker is reporting stuff to the network and doesn't
            # have any results at this moment. Call me in some seconds.
            return True
        
        if bug_report_result == self.worker.FINISHED:
            self.throbber.running(False)
            self.throbber.hide()
            self.done_icon.show()
            ok_button = self.get_widget_for_response(gtk.RESPONSE_OK)
            ok_button.set_sensitive(True)
            # don't call me anymore !
            return False
        else:
            # Add the data to the label and ask to be called again
            ticket_id, ticket_url = bug_report_result
            self.add_link(ticket_id, ticket_url)
            return True

    def add_link(self, ticket_id, ticket_url):
        self.reported_ids.append(ticket_id)
        
        new_link = '<a href="%s">%s</a>'
        new_link = new_link % (ticket_url, ticket_id)
        
        current_markup = self.link_label.get_label()

        needs_new_line = False
        if self.ticket_ids_in_markup == 4:
            needs_new_line = True
            self.ticket_ids_in_markup = 0
        else:
            self.ticket_ids_in_markup += 1

        needs_delim = True
        if len(current_markup) == 0 or needs_new_line:
            needs_delim = False
                    
        current_markup += ('\n' if needs_new_line else '') + (', ' if needs_delim else '')
        current_markup += new_link
        
        self.link_label.set_markup(current_markup)