def __init__(self, daddy=None):
        HIGMainWindow.__init__(self)

        self.daddy = daddy

        self.tip_timer = -1
        self.main_accel_group = gtk.AccelGroup()
        self.add_accel_group(self.main_accel_group)

        self.viewernb = HIGNotebook()

        self.invsearch = SearchDB(umitdb)
        self.invdb = ConnectInventoryDB(umitdb)
        self.invtree = InventoryTree(self)
        self.invnb = HIGNotebook()
        self.invnb_close_btn = startup_options()["tabs_close_btn"]
        self.invnbpages_titles = []
        self.invnbpages_objects = []

        # timeline tab
        self.timeline = TLHolder()

        # statusbars
        self.std_statusbar = gtk.Statusbar()
        self.std_sb = self.std_statusbar.get_context_id("stdbar")
        self._write_statusbar("                              ")

        self.tip_statusbar = gtk.Statusbar()
        self.tip_statusbar.set_has_resize_grip(False)
        self.tip_sb = self.tip_statusbar.get_context_id("tipbar")
        self.write_tips = startup_options()["tips"]

        # timer for checking updates
        self.db_stat = os.stat(umitdb).st_mtime
        self.schema_stat = os.stat(Path.sched_schemas).st_mtime
        self.timer_updater = gobject.timeout_add(
            4200,  # nice number
            self.__check_for_updates)

        # gui scheduler controller
        self.schedctrl = SchedControl(self)

        self.inventories = self.invtree.fill_tree()
        self._create_ui_manager()
        self.__set_props()
        self.__do_layout()

        self.connect('inventory-activated', self._inventory_activated)
        self.connect('delete-event', self._exit_ni)
        self.connect('realize', self.on_realize)
예제 #2
0
    def __init__(self, daddy=None):
        HIGMainWindow.__init__(self)

        self.daddy = daddy

        self.tip_timer = -1
        self.main_accel_group = gtk.AccelGroup()
        self.add_accel_group(self.main_accel_group)

        self.viewernb = HIGNotebook()

        self.invsearch = SearchDB(umitdb)
        self.invdb = ConnectInventoryDB(umitdb)
        self.invtree = InventoryTree(self)
        self.invnb = HIGNotebook()
        self.invnb_close_btn = startup_options()["tabs_close_btn"]
        self.invnbpages_titles = [ ]
        self.invnbpages_objects = [ ]

        # timeline tab
        self.timeline = TLHolder()

        # statusbars
        self.std_statusbar = gtk.Statusbar()
        self.std_sb = self.std_statusbar.get_context_id("stdbar")
        self._write_statusbar("                              ")

        self.tip_statusbar = gtk.Statusbar()
        self.tip_statusbar.set_has_resize_grip(False)
        self.tip_sb = self.tip_statusbar.get_context_id("tipbar")
        self.write_tips = startup_options()["tips"]

        # timer for checking updates
        self.db_stat = os.stat(umitdb).st_mtime
        self.schema_stat = os.stat(Path.sched_schemas).st_mtime
        self.timer_updater = gobject.timeout_add(4200, # nice number
            self.__check_for_updates)

        # gui scheduler controller
        self.schedctrl = SchedControl(self)


        self.inventories = self.invtree.fill_tree()
        self._create_ui_manager()
        self.__set_props()
        self.__do_layout()

        self.connect('inventory-activated', self._inventory_activated)
        self.connect('delete-event', self._exit_ni)
        self.connect('realize', self.on_realize)
class InventoryViewer(HIGMainWindow):
    """
    UMIT Network Inventory, Main Window.
    """
    def __init__(self, daddy=None):
        HIGMainWindow.__init__(self)

        self.daddy = daddy

        self.tip_timer = -1
        self.main_accel_group = gtk.AccelGroup()
        self.add_accel_group(self.main_accel_group)

        self.viewernb = HIGNotebook()

        self.invsearch = SearchDB(umitdb)
        self.invdb = ConnectInventoryDB(umitdb)
        self.invtree = InventoryTree(self)
        self.invnb = HIGNotebook()
        self.invnb_close_btn = startup_options()["tabs_close_btn"]
        self.invnbpages_titles = []
        self.invnbpages_objects = []

        # timeline tab
        self.timeline = TLHolder()

        # statusbars
        self.std_statusbar = gtk.Statusbar()
        self.std_sb = self.std_statusbar.get_context_id("stdbar")
        self._write_statusbar("                              ")

        self.tip_statusbar = gtk.Statusbar()
        self.tip_statusbar.set_has_resize_grip(False)
        self.tip_sb = self.tip_statusbar.get_context_id("tipbar")
        self.write_tips = startup_options()["tips"]

        # timer for checking updates
        self.db_stat = os.stat(umitdb).st_mtime
        self.schema_stat = os.stat(Path.sched_schemas).st_mtime
        self.timer_updater = gobject.timeout_add(
            4200,  # nice number
            self.__check_for_updates)

        # gui scheduler controller
        self.schedctrl = SchedControl(self)

        self.inventories = self.invtree.fill_tree()
        self._create_ui_manager()
        self.__set_props()
        self.__do_layout()

        self.connect('inventory-activated', self._inventory_activated)
        self.connect('delete-event', self._exit_ni)
        self.connect('realize', self.on_realize)

    def on_realize(self, event):
        """
        After realizing, change to first page in main notebook (Historic page).
        """
        self.viewernb.set_current_page(0)

    def _close_current_hosttab(self, event):
        """
        Closes current active Host tab.
        """
        holder = self.invnb.get_nth_page(self.invnb.get_current_page())
        if holder:
            title = self.invnb.get_tab_label_text(holder)
            p = self.invnbpages_objects[self.invnbpages_titles.index(title)]

            if len(p.pages):
                tab_label = p.pages[p.get_current_page()]
                content = p.get_nth_page(p.get_current_page())
                p.remove_page(None, content, tab_label)

    def _close_current_invtab(self, event):
        """
        Closes current active inventory notebook.
        """
        holder = self.invnb.get_nth_page(self.invnb.get_current_page())
        if holder:
            label = self.invnb.get_tab_label_text(holder)
            del_index = self.invnbpages_titles.index(label)

            page = self.invnbpages_objects[del_index]
            for p in xrange(page.get_n_pages() - 1):
                tab_label = page.pages[p]
                content = page.get_nth_page(p)
                page.remove_page(None, content, tab_label)
            self._delete_inventory_page()

    def _delete_inventory_page(self):
        """
        Removed last page from an Inventory notebook, now remove the Inventory
        notebook.
        """
        cur_page = self.invnb.get_current_page()
        page = self.invnb.get_nth_page(cur_page)
        page_label = self.invnb.get_tab_label_text(page)
        del_index = self.invnbpages_titles.index(page_label)
        del self.invnbpages_titles[del_index]
        del self.invnbpages_objects[del_index]

        self.invnb.remove_page(cur_page)

    def _inventory_activated(self, event, data):
        """
        Activated some item on Inventory tree.
        """
        if not data['host_addr']:  # empty inventory was selected
            # clicked on Inventory title, dont do nothing
            return

        title = data["root"]  # inventory name

        # check if activated inventory isnt open already
        if not title in self.invnbpages_titles:
            # create new notebook to hold this Inventory
            newinvnb = InventoryNB(self)
            self.invnb.append_page(newinvnb, gtk.Label(title))

            self.invnbpages_titles.append(title)
            self.invnbpages_objects.append(newinvnb)

            # check if a host was activated
            if data["host_addr"][0] != inventory_info:
                box = self._load_inventory_host_data(title,
                                                     data["host_addr"][1])
                newinvnb.append_host(data["host_addr"][0], box)
            else:
                # inventory info
                box = self._load_inventory_data(title)
                if box:
                    newinvnb.append_inv(box=box)

        else:
            # get Inventory notebook
            p = self.invnbpages_objects[self.invnbpages_titles.index(title)]

            # check if a host was activated
            if data["host_addr"][0] != inventory_info and \
                not p.page_exists(data["host_addr"]):
                box = self._load_inventory_host_data(title,
                                                     data["host_addr"][1])
                p.append_host(data["host_addr"][0], box)

            # if it is not a host, it was inventory info
            elif data["host_addr"][0] == inventory_info:
                box = self._load_inventory_data(title)
                if box:
                    p.append_inv(box=box)

    def _load_inventory_data(self, inventory_name):
        """
        Load data for activated Inventory.
        """
        fk_inventory = self.invdb.get_inventory_id_for_name(inventory_name)
        scans = self.invdb.get_scans_id_for_inventory(fk_inventory)
        last_scan_id = scans[len(scans) - 1][0]
        hosts = self.invtree.invdata[inventory_name]

        box = gtk.VBox()

        # scan info
        hb = gtk.HBox()
        hb.pack_start(gtk.Label(_("Scan count:") + (" %d" % len(scans))),
                      False, False, 0)
        box.pack_start(hb, False, False, 0)

        hb = gtk.HBox()
        last_scan_date = self.invdb.get_finish_timestamp_for_scan_from_db(
            last_scan_id)
        hb.pack_start(
            gtk.Label(_("Last scan date:") + (" %s" % last_scan_date)), False,
            False, 0)
        box.pack_start(hb, False, False, 0)

        details = self.invdb.get_scan_details_for_scan_from_db(scans[0][0])
        detc = [
            _("Scan args"),
            _("XML output version"),
            _("Verbose"),
            _("Debugging"),
            _("Scanner name"),
            _("Scanner version")
        ]
        for indx, item in enumerate(details):
            hb = gtk.HBox()
            hb.pack_start(gtk.Label("%s: %s" % (detc[indx], item)), False,
                          False, 0)
            box.pack_start(hb, False, False, 0)

        # "separator"
        box.pack_start(gtk.HSeparator(), False, False, 6)

        # hosts info
        hb = gtk.HBox()
        hb.pack_start(
            gtk.Label(_("Hosts: ") + ', '.join([host[0] for host in hosts])),
            False, False, 0)
        box.pack_start(hb, False, False, 0)

        hb = gtk.HBox()
        hb.pack_start(gtk.Label(_("Host count:") + (" %s" % len(hosts))),
                      False, False, 0)
        box.pack_start(hb, False, False, 0)

        sw = gtk.ScrolledWindow()
        sw.add_with_viewport(box)
        sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        sw.show_all()

        return sw

    def _load_inventory_host_data(self, inventory_name, host_addr):
        """
        Load data for a host in an inventory.
        """
        changesd = ChangesDiff(self.invdb)
        changesl = ChangesList(self,
                               self.invdb,
                               changesd,
                               data={-1: (inventory_name, host_addr)})

        paned = gtk.VPaned()
        paned.add1(changesl)
        paned.add2(changesd)

        return paned

    def _write_statusbar(self, msg):
        """
        Write some message at standard statusbar.
        """
        self.std_statusbar.pop(self.std_sb)
        self.std_statusbar.push(self.std_sb, msg)

    def _clear_statusbar(self):
        """
        Clear standard statusbar.
        """
        self.std_statusbar.pop(self.std_sb)

    def _write_statusbar_tip(self, tipmsg, timeout=True):
        """
        Write a tip at statusbar.
        """
        if not self.write_tips:
            # tips disabled
            return

        if not self.window:  # widget being destroyed
            gobject.source_remove(self.tip_timer)
            return

        self.tip_statusbar.pop(self.tip_sb)
        self.tip_statusbar.push(self.tip_sb, tipmsg)

        if self.tip_timer != -1:
            gobject.source_remove(self.tip_timer)

        if timeout:
            # remove tip after 15 seconds
            self.tip_timer = gobject.timeout_add(15000,
                                                 self._clear_tip_statusbar)

    def _clear_tip_statusbar(self):
        """
        Clear statusbar.
        """
        self.tip_statusbar.pop(self.tip_sb)
        gobject.source_remove(self.tip_timer)
        self.tip_timer = -1

        return False

    def _update_viewer(self):
        """
        Call this when needed to update Network Inventory Viewer.
        """
        inventories = self.invtree.fill_tree()

        # update Edit Inventories menu
        new_inventories = set(inventories).difference(set(self.inventories))
        self.inventories = inventories

        for inv in new_inventories:
            self.main_action_group.add_actions([("_" + inv, None, inv, None,
                                                 inv, self._edit_inventory)])

            self.ui_manager.add_ui(self.ui_manager.new_merge_id(),
                                   "/menubar/Edit/Inventory", inv, "_" + inv,
                                   "menuitem", False)

    def _build_results_view(self, query, returned, buildnow=False):
        """
        Build a TreeView with returned results from search or perform search
        and then Build TreeView.
        """
        def close_page(widget, data):
            """
            Close search page.
            """
            page_num = self.viewernb.page_num(data)
            self.viewernb.remove_page(page_num)

        def row_activated(tv, path, tvcolumn):
            """
            Activated some row in results treeview.
            """
            # create page at "Inventory"
            model = tv.get_model()
            tv_iter = model.get_iter(path)
            host_addr = model.get_value(tv_iter, 1)

            data = {}
            root = model[path[0]][0]
            data["root"] = root
            data["host_addr"] = (host_addr, host_addr)  # new format

            self.emit('inventory-activated', data)

            # write tip to statusbar
            self._write_statusbar_tip(_("A page was open at Historic tab!"))

        def buildtv(*args):
            """
            Build treeview and append results.
            """
            lsmodel = gtk.ListStore(str, str, str, str)
            tview = gtk.TreeView(lsmodel)
            columns = 4
            columns_text = (_("Inventory"), _("Host"),
                            _("First Matched Category"), _("Entry Date"))
            tview.columns = [None] * columns

            for n in range(columns):
                tview.columns[n] = gtk.TreeViewColumn(columns_text[n])
                tview.columns[n].cell = gtk.CellRendererText()
                tview.columns[n].pack_start(tview.columns[n].cell, True)
                tview.columns[n].set_attributes(tview.columns[n].cell, text=n)

                tview.append_column(tview.columns[n])

            tview.connect('row-activated', row_activated)

            # layout
            matches = len(args[1])
            word = append_s(_("result"), matches)
            upper_title = gtk.Label()
            upper_title.set_markup(("<b>%d " % matches) + word + _(" found.") +
                                   "</b>")
            hutbox = HIGHBox()
            hutbox._pack_noexpand_nofill(upper_title)
            sw = HIGScrolledWindow()
            sw.add(tview)
            vbox = gtk.VBox()
            vbox.pack_start(hutbox, False, False, 3)
            vbox.pack_start(sw, True, True, 0)

            vbox.show_all()

            # append results to treeview
            results = args[1]

            for res, date in results.items():
                invname = self.invdb.get_inventory_name_for_id(res[0])
                lsmodel.append((invname, res[1], res[2], str(date)))

            title = _("Search: %s") % args[0]
            tab_label = HIGAnimatedTabLabel(title)
            tab_label.connect("close-clicked", close_page, vbox)
            self.viewernb.append_page(vbox, tab_label)

        if buildnow:
            # uhm.. nice, we already have the results.
            buildtv(query, returned)

        else:
            # no results yet, will get them now!
            host_matches = {}

            for key, values in returned.items():
                host_id = key
                inv_id = values[0]
                host_addr = values[1]
                date = values[2]
                res = self.invsearch.search(host_id, query)
                if res:  # res = category returned
                    host_matches[(inv_id, host_addr, res)] = date

            # build treeview now
            buildtv(query, host_matches)

    def _create_ui_manager(self):
        """
        Set up user interface.
        """
        self.ui_manager = gtk.UIManager()
        self.main_action_group = gtk.ActionGroup("MainActionGroup")

        about_icon = None
        try:
            about_icon = gtk.STOCK_ABOUT
        except:
            pass

        main_actions = [ \
            ("File", None, _("_File"), None),
            ("Edit", None, _("_Edit"), None),
            ("Scheduler", None, _("_Scheduler"), None),
            ("Help", None, _("_Help"), None),

            ("ToolbarOpt", None, _("Tool_bar"), None),
            ("InvTabs", None, _("Devices _tab"), None),
            ("Inventory", None, _("_Inventories"), None),

            # File
            ("New",
             gtk.STOCK_NEW, _("_New Inventory"),
             None, _("Create new Inventory"),
             self._create_new),

            ("CloseHostTab",
             None, _("Close Device tab"),
             "<Control>W", _("Close current host tab"),
             self._close_current_hosttab),

            ("CloseInvTab",
             None, _("Close Inventory tab"),
             "<Shift><Control>W", _("Close current Inventory tab"),
             self._close_current_invtab),

            ("Quit",
             gtk.STOCK_QUIT, _("_Quit"),
             None, _("Close Network Inventory"),
             self._exit_ni),

            # Edit
            ('Control Data Removal',
             None, _("Data Removal"),
             None, _("Data Removal"),
             self._edit_data_removal),

            ('Startup',
             None, _("Startup settings"),
             None, _("Startup settings"),
             self._edit_startup),

            # Scheduler
            # Scheduler log
            ('Sched Log',
             None, _("Scheduler log"),
             None, _("View Scheduler log"),
             self._open_sched_log),

            # Scheduler status
            ('Sched Control',
             self.schedctrl.stock_icon, self.schedctrl.status_text,
             None, self.schedctrl.status_text,
             self.schedctrl._scheduler_control),

            # Help
            ('Show Help',
             gtk.STOCK_HELP, _('_Help'),
             None, _('Shows the application help'),
             self._show_help),

            ('Report a bug',
             gtk.STOCK_DIALOG_INFO, _('_Report a bug'),
             '<Control>b', _("Report a bug"),
             self._show_bug_report),

            ('About',
             about_icon, _("_About"),
             '<Control>a', _("About UMIT Network Inventory"),
             self._show_about)

        ]

        toggle_actions = [
            # Edit
            ("ShowTips", None, _("Show ti_ps"), None,
             _("Enable disable tip showing"),
             lambda c: self.set_write_tips(not self.write_tips),
             self.write_tips),

            # Edit/Notebooks
            ("TabCloseBtn", None, _("Place close button"), None,
             _("Place a close button upon tab creation"),
             lambda c: self.set_invnb_cbtn(not self.invnb_close_btn),
             self.invnb_close_btn)
        ]

        default_ui = """
            <menubar>
                <menu action='File'>
                    <menuitem action='New' />
                    <separator />
                    <menuitem action='CloseHostTab' />
                    <menuitem action='CloseInvTab' />
                    <menuitem action='Quit' />
                </menu>
                <menu action='Edit'>
                    <menu action='InvTabs'>
                         <menuitem action='TabCloseBtn' />
                    </menu>
                    <menu action='Inventory'>
                    </menu>
                    <menuitem action='ShowTips' />
                    <menuitem action='Control Data Removal' />
                    <menuitem action='Startup' />
                </menu>
                <menu action='Scheduler'>
                    <menuitem action='Sched Log' />
                    <menuitem action='Sched Control' />
                </menu>
                <menu action='Help'>
                    <menuitem action='Show Help' />
                    <menuitem action='Report a bug' />
                    <menuitem action='About' />
                </menu>
            </menubar>
            <toolbar>
                <toolitem action='Sched Control' />
                <separator />
            </toolbar>
            """

        self.main_action_group.add_actions(main_actions)
        self.main_action_group.add_toggle_actions(toggle_actions)

        for action in self.main_action_group.list_actions():
            action.set_accel_group(self.main_accel_group)
            action.connect_accelerator()

        self.ui_manager.insert_action_group(self.main_action_group, 0)
        self.ui_manager.add_ui_from_string(default_ui)

        self.schedctrl.ui_action = self.main_action_group.get_action(
            'Sched Control')

        # add Inventories to edit menu
        for inv in self.inventories:
            self.main_action_group.add_actions([("_" + inv, None, inv, None,
                                                 inv, self._edit_inventory)])

            self.ui_manager.add_ui(self.ui_manager.new_merge_id(),
                                   "/menubar/Edit/Inventory", inv, "_" + inv,
                                   "menuitem", False)

    def _edit_data_removal(self, event):
        """
        Open window for editing data removal for Inventories.
        """
        w = ConfigureDataRemoval()
        w.show_all()

    def _edit_inventory(self, event):
        """
        Open inventory for editing.
        """
        try:
            w = NewInventory(event.get_name()[1:], edit_mode=True)
        except NoInventory:
            return
        w.show_all()

    def _create_new(self, event):
        """
        Open dialog for creating New Inventory.
        """
        w = NewInventory()
        w.show_all()

    def _open_sched_log(self, event):
        """
        Open scheduler log.
        """
        winlog = SchedLog()
        winlog.show_all()

    def _edit_startup(self, event):
        """
        Open window for editing startup settings.
        """
        settwin = NISettings()
        settwin.show_all()

    def _show_help(self, event):
        """
        Open help manual.
        """
        show_help(self, "index.html")

    def _show_bug_report(self, event):
        """
        Open bug report window.
        """
        bug = BugReport()
        bug.show_all()

    def _show_about(self, event):
        """
        Open about window.
        """
        awin = About()
        awin.show_all()

    def _exit_ni(self, *args):
        """
        Do necessary cleanup before destroying window.
        """
        # remove tip timer if still running
        if self.tip_timer != -1:
            gobject.source_remove(self.tip_timer)
            self._clear_tip_statusbar()

        # stop updater timer
        gobject.source_remove(self.timer_updater)

        self.hide()

        # remove old data
        if get_decays()[1]:
            win = RemoveOldData()
            win.show_all()
            win.connect('destroy', self.__leave_ni)
        else:
            self.__leave_ni()

    def __leave_ni(self, *args):
        """
        If you are here, Network Inventory has been cleanedup succesfully.
        """
        if self.daddy:
            self.daddy.running_ni = False
            self.destroy()
        else:
            gtk.main_quit()

    def __check_for_updates(self):
        """
        Check for some possible visual update needed to be done.
        """
        do_update = False

        # check for database changes
        prev_stat = self.db_stat
        try:
            possibly_new = os.stat(umitdb).st_mtime
        except OSError:
            return True

        if prev_stat != possibly_new:
            do_update = True
            self.db_stat = possibly_new

        # check for changes in schemas
        prev_state = self.schema_stat
        try:
            possibly_new = os.stat(Path.sched_schemas).st_mtime
        except OSError:
            return True

        if prev_state != possibly_new:
            do_update = True
            self.schema_stat = possibly_new

        if do_update:
            self._update_viewer()

        return True

    def __set_props(self):
        """
        Set window properties.
        """
        self.set_title(_("UMIT Network Inventory"))
        self.invnb.set_scrollable(True)
        self.viewernb.set_scrollable(True)
        # a size for testing
        self.set_position(gtk.WIN_POS_CENTER)
        width, height = gtk.gdk.get_default_root_window().get_size()
        self.set_default_size((width * 3) / 4, (height * 3) / 4)

    def __do_layout(self):
        """
        Layout window widgets.
        """
        main_vbox = gtk.VBox()
        main_hpaned = gtk.HPaned()
        nb_tl_hpaned = gtk.HPaned()

        menubar = self.ui_manager.get_widget('/menubar')
        main_vbox.pack_start(menubar, False, False, 0)

        schedbar = self.ui_manager.get_widget('/toolbar')
        schedbar.set_show_arrow(False)
        schedbar.set_style(gtk.TOOLBAR_ICONS)
        toolbar = SearchBar(self, self.invdb)

        toolbar_box = gtk.HBox()
        toolbar_box.pack_start(schedbar, False, False, 0)
        toolbar_box.pack_start(toolbar, True, True, 0)
        main_vbox.pack_start(toolbar_box, False, False, 0)

        left_pane_box = gtk.VBox()
        left_pane_box.pack_start(self.invtree, True, True, 0)

        main_hpaned.add1(left_pane_box)
        main_hpaned.add2(nb_tl_hpaned)

        # inventories notebook
        nb_tl_hpaned.pack1(self.invnb, True, False)

        self.viewernb.append_page(main_hpaned, gtk.Label(_(" Historic ")))
        self.viewernb.append_page(self.timeline, gtk.Label(_(" Timeline ")))

        self.timeline.show_all()

        main_vbox.pack_start(self.viewernb, True, True, 0)

        sbs_box = gtk.HBox()
        sbs_box.add(self.tip_statusbar)
        sbs_box.add(self.std_statusbar)
        main_vbox.pack_end(sbs_box, False, False, 0)

        self.add(main_vbox)

    def get_write_tips(self):
        """
        Returns True if tips should be written at statusbar, otherwise, False.
        """
        return self.__wtips

    def set_write_tips(self, write):
        """
        Sets to write tips or not at statusbar.
        """
        self.__wtips = write
        if not self.write_tips:
            self.tip_statusbar.pop(self.tip_sb)

    def get_invnb_cbtn(self):
        """
        Returns True if Inventory tabs have a close button placed on them.
        """
        return self.__invcbtn

    def set_invnb_cbtn(self, enable):
        """
        Sets to show or not close button in each Inventory tab.
        """
        self.__invcbtn = enable

    # Properties
    write_tips = property(get_write_tips, set_write_tips)
    invnb_close_btn = property(get_invnb_cbtn, set_invnb_cbtn)
예제 #4
0
class InventoryViewer(HIGMainWindow):
    """
    UMIT Network Inventory, Main Window.
    """

    def __init__(self, daddy=None):
        HIGMainWindow.__init__(self)

        self.daddy = daddy

        self.tip_timer = -1
        self.main_accel_group = gtk.AccelGroup()
        self.add_accel_group(self.main_accel_group)

        self.viewernb = HIGNotebook()

        self.invsearch = SearchDB(umitdb)
        self.invdb = ConnectInventoryDB(umitdb)
        self.invtree = InventoryTree(self)
        self.invnb = HIGNotebook()
        self.invnb_close_btn = startup_options()["tabs_close_btn"]
        self.invnbpages_titles = [ ]
        self.invnbpages_objects = [ ]

        # timeline tab
        self.timeline = TLHolder()

        # statusbars
        self.std_statusbar = gtk.Statusbar()
        self.std_sb = self.std_statusbar.get_context_id("stdbar")
        self._write_statusbar("                              ")

        self.tip_statusbar = gtk.Statusbar()
        self.tip_statusbar.set_has_resize_grip(False)
        self.tip_sb = self.tip_statusbar.get_context_id("tipbar")
        self.write_tips = startup_options()["tips"]

        # timer for checking updates
        self.db_stat = os.stat(umitdb).st_mtime
        self.schema_stat = os.stat(Path.sched_schemas).st_mtime
        self.timer_updater = gobject.timeout_add(4200, # nice number
            self.__check_for_updates)

        # gui scheduler controller
        self.schedctrl = SchedControl(self)


        self.inventories = self.invtree.fill_tree()
        self._create_ui_manager()
        self.__set_props()
        self.__do_layout()

        self.connect('inventory-activated', self._inventory_activated)
        self.connect('delete-event', self._exit_ni)
        self.connect('realize', self.on_realize)


    def on_realize(self, event):
        """
        After realizing, change to first page in main notebook (Historic page).
        """
        self.viewernb.set_current_page(0)


    def _close_current_hosttab(self, event):
        """
        Closes current active Host tab.
        """
        holder = self.invnb.get_nth_page(self.invnb.get_current_page())
        if holder:
            title = self.invnb.get_tab_label_text(holder)
            p = self.invnbpages_objects[self.invnbpages_titles.index(title)]

            if len(p.pages):
                tab_label = p.pages[p.get_current_page()]
                content = p.get_nth_page(p.get_current_page())
                p.remove_page(None, content, tab_label)


    def _close_current_invtab(self, event):
        """
        Closes current active inventory notebook.
        """
        holder = self.invnb.get_nth_page(self.invnb.get_current_page())
        if holder:
            label = self.invnb.get_tab_label_text(holder)
            del_index = self.invnbpages_titles.index(label)

            page = self.invnbpages_objects[del_index]
            for p in xrange(page.get_n_pages()-1):
                tab_label = page.pages[p]
                content = page.get_nth_page(p)
                page.remove_page(None, content, tab_label)
            self._delete_inventory_page()


    def _delete_inventory_page(self):
        """
        Removed last page from an Inventory notebook, now remove the Inventory
        notebook.
        """
        cur_page = self.invnb.get_current_page()
        page = self.invnb.get_nth_page(cur_page)
        page_label = self.invnb.get_tab_label_text(page)
        del_index = self.invnbpages_titles.index(page_label)
        del self.invnbpages_titles[del_index]
        del self.invnbpages_objects[del_index]

        self.invnb.remove_page(cur_page)


    def _inventory_activated(self, event, data):
        """
        Activated some item on Inventory tree.
        """
        if not data['host_addr']: # empty inventory was selected
            # clicked on Inventory title, dont do nothing
            return

        title = data["root"] # inventory name

        # check if activated inventory isnt open already
        if not title in self.invnbpages_titles:
            # create new notebook to hold this Inventory
            newinvnb = InventoryNB(self)
            self.invnb.append_page(newinvnb, gtk.Label(title))

            self.invnbpages_titles.append(title)
            self.invnbpages_objects.append(newinvnb)

            # check if a host was activated
            if data["host_addr"][0] != inventory_info:
                box = self._load_inventory_host_data(title,
                    data["host_addr"][1])
                newinvnb.append_host(data["host_addr"][0], box)
            else:
                # inventory info
                box = self._load_inventory_data(title)
                if box:
                    newinvnb.append_inv(box=box)

        else:
            # get Inventory notebook
            p = self.invnbpages_objects[self.invnbpages_titles.index(title)]

            # check if a host was activated
            if data["host_addr"][0] != inventory_info and \
                not p.page_exists(data["host_addr"]):
                box = self._load_inventory_host_data(title,
                    data["host_addr"][1])
                p.append_host(data["host_addr"][0], box)

            # if it is not a host, it was inventory info
            elif data["host_addr"][0] == inventory_info:
                box = self._load_inventory_data(title)
                if box:
                    p.append_inv(box=box)


    def _load_inventory_data(self, inventory_name):
        """
        Load data for activated Inventory.
        """
        fk_inventory = self.invdb.get_inventory_id_for_name(inventory_name)
        scans = self.invdb.get_scans_id_for_inventory(fk_inventory)
        last_scan_id = scans[len(scans)-1][0]
        hosts = self.invtree.invdata[inventory_name]

        box = gtk.VBox()

        # scan info
        hb = gtk.HBox()
        hb.pack_start(gtk.Label(_("Scan count:") + (" %d" % len(scans))),
                False, False, 0)
        box.pack_start(hb, False, False, 0)

        hb = gtk.HBox()
        last_scan_date = self.invdb.get_finish_timestamp_for_scan_from_db(
                last_scan_id)
        hb.pack_start(
                gtk.Label(_("Last scan date:") + (" %s" % last_scan_date)),
                False, False, 0)
        box.pack_start(hb, False, False, 0)

        details = self.invdb.get_scan_details_for_scan_from_db(scans[0][0])
        detc = [
            _("Scan args"), _("XML output version"), _("Verbose"),
            _("Debugging"), _("Scanner name"), _("Scanner version")
            ]
        for indx, item in enumerate(details):
            hb = gtk.HBox()
            hb.pack_start(gtk.Label("%s: %s" % (detc[indx], item)), False,
                False, 0)
            box.pack_start(hb, False, False, 0)

        # "separator"
        box.pack_start(gtk.HSeparator(), False, False, 6)

        # hosts info
        hb = gtk.HBox()
        hb.pack_start(gtk.Label(
            _("Hosts: ") +
            ', '.join([host[0] for host in hosts])), False, False, 0)
        box.pack_start(hb, False, False, 0)

        hb = gtk.HBox()
        hb.pack_start(gtk.Label(_("Host count:") + (" %s" % len(hosts))),
                False, False, 0)
        box.pack_start(hb, False, False, 0)

        sw = gtk.ScrolledWindow()
        sw.add_with_viewport(box)
        sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        sw.show_all()

        return sw


    def _load_inventory_host_data(self, inventory_name, host_addr):
        """
        Load data for a host in an inventory.
        """
        changesd = ChangesDiff(self.invdb)
        changesl = ChangesList(self, self.invdb, changesd,
            data={-1: (inventory_name, host_addr)})

        paned = gtk.VPaned()
        paned.add1(changesl)
        paned.add2(changesd)

        return paned


    def _write_statusbar(self, msg):
        """
        Write some message at standard statusbar.
        """
        self.std_statusbar.pop(self.std_sb)
        self.std_statusbar.push(self.std_sb, msg)


    def _clear_statusbar(self):
        """
        Clear standard statusbar.
        """
        self.std_statusbar.pop(self.std_sb)


    def _write_statusbar_tip(self, tipmsg, timeout=True):
        """
        Write a tip at statusbar.
        """
        if not self.write_tips:
            # tips disabled
            return

        if not self.window: # widget being destroyed
            gobject.source_remove(self.tip_timer)
            return

        self.tip_statusbar.pop(self.tip_sb)
        self.tip_statusbar.push(self.tip_sb, tipmsg)

        if self.tip_timer != -1:
            gobject.source_remove(self.tip_timer)

        if timeout:
            # remove tip after 15 seconds
            self.tip_timer = gobject.timeout_add(15000,
                self._clear_tip_statusbar)


    def _clear_tip_statusbar(self):
        """
        Clear statusbar.
        """
        self.tip_statusbar.pop(self.tip_sb)
        gobject.source_remove(self.tip_timer)
        self.tip_timer = -1

        return False


    def _update_viewer(self):
        """
        Call this when needed to update Network Inventory Viewer.
        """
        inventories = self.invtree.fill_tree()

        # update Edit Inventories menu
        new_inventories = set(inventories).difference(set(self.inventories))
        self.inventories = inventories

        for inv in new_inventories:
            self.main_action_group.add_actions([("_" + inv, None, inv,
                None, inv, self._edit_inventory)])

            self.ui_manager.add_ui(self.ui_manager.new_merge_id(),
                "/menubar/Edit/Inventory", inv, "_" + inv, "menuitem", False)


    def _build_results_view(self, query, returned, buildnow=False):
        """
        Build a TreeView with returned results from search or perform search
        and then Build TreeView.
        """
        def close_page(widget, data):
            """
            Close search page.
            """
            page_num = self.viewernb.page_num(data)
            self.viewernb.remove_page(page_num)


        def row_activated(tv, path, tvcolumn):
            """
            Activated some row in results treeview.
            """
            # create page at "Inventory"
            model = tv.get_model()
            tv_iter = model.get_iter(path)
            host_addr = model.get_value(tv_iter, 1)

            data = { }
            root = model[path[0]][0]
            data["root"] = root
            data["host_addr"] = (host_addr, host_addr) # new format

            self.emit('inventory-activated', data)

            # write tip to statusbar
            self._write_statusbar_tip(_("A page was open at Historic tab!"))


        def buildtv(*args):
            """
            Build treeview and append results.
            """
            lsmodel = gtk.ListStore(str, str, str, str)
            tview = gtk.TreeView(lsmodel)
            columns = 4
            columns_text = (
                _("Inventory"), _("Host"),
                _("First Matched Category"), _("Entry Date")
                )
            tview.columns = [None]*columns

            for n in range(columns):
                tview.columns[n] = gtk.TreeViewColumn(columns_text[n])
                tview.columns[n].cell = gtk.CellRendererText()
                tview.columns[n].pack_start(tview.columns[n].cell, True)
                tview.columns[n].set_attributes(tview.columns[n].cell, text=n)

                tview.append_column(tview.columns[n])

            tview.connect('row-activated', row_activated)

            # layout
            matches = len(args[1])
            word = append_s(_("result"), matches)
            upper_title = gtk.Label()
            upper_title.set_markup(
                    ("<b>%d " % matches) + word + _(" found.") +
                    "</b>")
            hutbox = HIGHBox()
            hutbox._pack_noexpand_nofill(upper_title)
            sw = HIGScrolledWindow()
            sw.add(tview)
            vbox = gtk.VBox()
            vbox.pack_start(hutbox, False, False, 3)
            vbox.pack_start(sw, True, True, 0)

            vbox.show_all()

            # append results to treeview
            results = args[1]

            for res, date in results.items():
                invname = self.invdb.get_inventory_name_for_id(res[0])
                lsmodel.append((invname, res[1], res[2], str(date)))

            title = _("Search: %s") % args[0]
            tab_label = HIGAnimatedTabLabel(title)
            tab_label.connect("close-clicked", close_page, vbox)
            self.viewernb.append_page(vbox, tab_label)


        if buildnow:
            # uhm.. nice, we already have the results.
            buildtv(query, returned)

        else:
            # no results yet, will get them now!
            host_matches = { }

            for key, values in returned.items():
                host_id = key
                inv_id = values[0]
                host_addr = values[1]
                date = values[2]
                res = self.invsearch.search(host_id, query)
                if res: # res = category returned
                    host_matches[(inv_id, host_addr, res)] = date

            # build treeview now
            buildtv(query, host_matches)


    def _create_ui_manager(self):
        """
        Set up user interface.
        """
        self.ui_manager = gtk.UIManager()
        self.main_action_group = gtk.ActionGroup("MainActionGroup")

        about_icon = None
        try: about_icon = gtk.STOCK_ABOUT
        except: pass

        main_actions = [ \
            ("File", None, _("_File"), None),
            ("Edit", None, _("_Edit"), None),
            ("Scheduler", None, _("_Scheduler"), None),
            ("Help", None, _("_Help"), None),

            ("ToolbarOpt", None, _("Tool_bar"), None),
            ("InvTabs", None, _("Devices _tab"), None),
            ("Inventory", None, _("_Inventories"), None),

            # File
            ("New",
             gtk.STOCK_NEW, _("_New Inventory"),
             None, _("Create new Inventory"),
             self._create_new),

            ("CloseHostTab",
             None, _("Close Device tab"),
             "<Control>W", _("Close current host tab"),
             self._close_current_hosttab),

            ("CloseInvTab",
             None, _("Close Inventory tab"),
             "<Shift><Control>W", _("Close current Inventory tab"),
             self._close_current_invtab),

            ("Quit",
             gtk.STOCK_QUIT, _("_Quit"),
             None, _("Close Network Inventory"),
             self._exit_ni),

            # Edit
            ('Control Data Removal',
             None, _("Data Removal"),
             None, _("Data Removal"),
             self._edit_data_removal),

            ('Startup',
             None, _("Startup settings"),
             None, _("Startup settings"),
             self._edit_startup),

            # Scheduler
            # Scheduler log
            ('Sched Log',
             None, _("Scheduler log"),
             None, _("View Scheduler log"),
             self._open_sched_log),

            # Scheduler status
            ('Sched Control',
             self.schedctrl.stock_icon, self.schedctrl.status_text,
             None, self.schedctrl.status_text,
             self.schedctrl._scheduler_control),

            # Help
            ('Show Help',
             gtk.STOCK_HELP, _('_Help'),
             None, _('Shows the application help'),
             self._show_help),

            ('Report a bug',
             gtk.STOCK_DIALOG_INFO, _('_Report a bug'),
             '<Control>b', _("Report a bug"),
             self._show_bug_report),

            ('About',
             about_icon, _("_About"),
             '<Control>a', _("About UMIT Network Inventory"),
             self._show_about)

        ]

        toggle_actions = [
            # Edit
            ("ShowTips",
             None, _("Show ti_ps"),
             None, _("Enable disable tip showing"),
             lambda c: self.set_write_tips(not self.write_tips),
             self.write_tips),

            # Edit/Notebooks
            ("TabCloseBtn",
             None, _("Place close button"),
             None, _("Place a close button upon tab creation"),
             lambda c: self.set_invnb_cbtn(not self.invnb_close_btn),
             self.invnb_close_btn)
        ]

        default_ui = """
            <menubar>
                <menu action='File'>
                    <menuitem action='New' />
                    <separator />
                    <menuitem action='CloseHostTab' />
                    <menuitem action='CloseInvTab' />
                    <menuitem action='Quit' />
                </menu>
                <menu action='Edit'>
                    <menu action='InvTabs'>
                         <menuitem action='TabCloseBtn' />
                    </menu>
                    <menu action='Inventory'>
                    </menu>
                    <menuitem action='ShowTips' />
                    <menuitem action='Control Data Removal' />
                    <menuitem action='Startup' />
                </menu>
                <menu action='Scheduler'>
                    <menuitem action='Sched Log' />
                    <menuitem action='Sched Control' />
                </menu>
                <menu action='Help'>
                    <menuitem action='Show Help' />
                    <menuitem action='Report a bug' />
                    <menuitem action='About' />
                </menu>
            </menubar>
            <toolbar>
                <toolitem action='Sched Control' />
                <separator />
            </toolbar>
            """

        self.main_action_group.add_actions(main_actions)
        self.main_action_group.add_toggle_actions(toggle_actions)

        for action in self.main_action_group.list_actions():
            action.set_accel_group(self.main_accel_group)
            action.connect_accelerator()

        self.ui_manager.insert_action_group(self.main_action_group, 0)
        self.ui_manager.add_ui_from_string(default_ui)

        self.schedctrl.ui_action = self.main_action_group.get_action(
            'Sched Control')

        # add Inventories to edit menu
        for inv in self.inventories:
            self.main_action_group.add_actions([("_" + inv, None, inv,
                None, inv, self._edit_inventory)])

            self.ui_manager.add_ui(self.ui_manager.new_merge_id(),
                "/menubar/Edit/Inventory", inv, "_" + inv, "menuitem", False)


    def _edit_data_removal(self, event):
        """
        Open window for editing data removal for Inventories.
        """
        w = ConfigureDataRemoval()
        w.show_all()


    def _edit_inventory(self, event):
        """
        Open inventory for editing.
        """
        try:
            w = NewInventory(event.get_name()[1:], edit_mode=True)
        except NoInventory:
            return
        w.show_all()


    def _create_new(self, event):
        """
        Open dialog for creating New Inventory.
        """
        w = NewInventory()
        w.show_all()


    def _open_sched_log(self, event):
        """
        Open scheduler log.
        """
        winlog = SchedLog()
        winlog.show_all()


    def _edit_startup(self, event):
        """
        Open window for editing startup settings.
        """
        settwin = NISettings()
        settwin.show_all()


    def _show_help(self, event):
        """
        Open help manual.
        """
        show_help(self,"index.html")


    def _show_bug_report(self, event):
        """
        Open bug report window.
        """
        bug = BugReport()
        bug.show_all()


    def _show_about(self, event):
        """
        Open about window.
        """
        awin = About()
        awin.show_all()


    def _exit_ni(self, *args):
        """
        Do necessary cleanup before destroying window.
        """
        # remove tip timer if still running
        if self.tip_timer != -1:
            gobject.source_remove(self.tip_timer)
            self._clear_tip_statusbar()

        # stop updater timer
        gobject.source_remove(self.timer_updater)

        self.hide()

        # remove old data
        if get_decays()[1]:
            win = RemoveOldData()
            win.show_all()
            win.connect('destroy', self.__leave_ni)
        else:
            self.__leave_ni()


    def __leave_ni(self, *args):
        """
        If you are here, Network Inventory has been cleanedup succesfully.
        """
        if self.daddy:
            self.daddy.running_ni = False
            self.destroy()
        else:
            gtk.main_quit()


    def __check_for_updates(self):
        """
        Check for some possible visual update needed to be done.
        """
        do_update = False

        # check for database changes
        prev_stat = self.db_stat
        try:
            possibly_new = os.stat(umitdb).st_mtime
        except OSError:
            return True

        if prev_stat != possibly_new:
            do_update = True
            self.db_stat = possibly_new

        # check for changes in schemas
        prev_state = self.schema_stat
        try:
            possibly_new = os.stat(Path.sched_schemas).st_mtime
        except OSError:
            return True

        if prev_state != possibly_new:
            do_update = True
            self.schema_stat = possibly_new


        if do_update:
            self._update_viewer()

        return True


    def __set_props(self):
        """
        Set window properties.
        """
        self.set_title(_("UMIT Network Inventory"))
        self.invnb.set_scrollable(True)
        self.viewernb.set_scrollable(True)
        # a size for testing
        self.set_position(gtk.WIN_POS_CENTER)
        width, height = gtk.gdk.get_default_root_window().get_size()
        self.set_default_size((width * 3) / 4, (height * 3) / 4)


    def __do_layout(self):
        """
        Layout window widgets.
        """
        main_vbox = gtk.VBox()
        main_hpaned = gtk.HPaned()
        nb_tl_hpaned = gtk.HPaned()

        menubar = self.ui_manager.get_widget('/menubar')
        main_vbox.pack_start(menubar, False, False, 0)

        schedbar = self.ui_manager.get_widget('/toolbar')
        schedbar.set_show_arrow(False)
        schedbar.set_style(gtk.TOOLBAR_ICONS)
        toolbar = SearchBar(self, self.invdb)

        toolbar_box = gtk.HBox()
        toolbar_box.pack_start(schedbar, False, False, 0)
        toolbar_box.pack_start(toolbar, True, True, 0)
        main_vbox.pack_start(toolbar_box, False, False, 0)

        left_pane_box = gtk.VBox()
        left_pane_box.pack_start(self.invtree, True, True, 0)

        main_hpaned.add1(left_pane_box)
        main_hpaned.add2(nb_tl_hpaned)

        # inventories notebook
        nb_tl_hpaned.pack1(self.invnb, True, False)

        self.viewernb.append_page(main_hpaned, gtk.Label(_(" Historic ")))
        self.viewernb.append_page(self.timeline, gtk.Label(_(" Timeline ")))

        self.timeline.show_all()

        main_vbox.pack_start(self.viewernb, True, True, 0)

        sbs_box = gtk.HBox()
        sbs_box.add(self.tip_statusbar)
        sbs_box.add(self.std_statusbar)
        main_vbox.pack_end(sbs_box, False, False, 0)

        self.add(main_vbox)


    def get_write_tips(self):
        """
        Returns True if tips should be written at statusbar, otherwise, False.
        """
        return self.__wtips


    def set_write_tips(self, write):
        """
        Sets to write tips or not at statusbar.
        """
        self.__wtips = write
        if not self.write_tips:
            self.tip_statusbar.pop(self.tip_sb)


    def get_invnb_cbtn(self):
        """
        Returns True if Inventory tabs have a close button placed on them.
        """
        return self.__invcbtn


    def set_invnb_cbtn(self, enable):
        """
        Sets to show or not close button in each Inventory tab.
        """
        self.__invcbtn = enable


    # Properties
    write_tips = property(get_write_tips, set_write_tips)
    invnb_close_btn = property(get_invnb_cbtn, set_invnb_cbtn)