root/trunk/main.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import sys
import pkg_resources
sys.modules['simplepanels'] = __import__(__name__)

ENTRYPOINT = 'simplepanels.plugins'
PLUGIN_DIR = os.path.join(os.path.dirname(__file__), 'plugins')
import gtk

from threading import Thread
from misc.SimpleGladeApp import SimpleGladeApp
from misc.SimpleGladeApp import bindtextdomain
from misc.pexpect import pexpect, pxssh
from misc.utils import Treeview
import config

try:
    import gtkmozembed #A mozilla embeded!
except:
    gtkmozembed = None

gtk.gdk.threads_init()

app_name = "simplepanels"
app_version = "0.0.1"

class MainWnd(SimpleGladeApp):
    def new(self):
        self.config = config
        self.settings = {}
        self.attach_plugins()
        self.show_hosts()
        #self.init_treeview()

    def attach_plugins(self):
        # threading plugins
        for plugin in get_plugins_by_capability('thread'):
            plugin.start()
            
        # attach plugins
        for plugin in get_plugins_by_capability('attach'):
            print '%s: %s' % (plugin, plugin.attach(self))
        
        #self.plugin = {}
        #for item in config.PLUGINS:
        #    self.plugin[item] = __import__("plugins/%s" % item)
        #    if self.plugin[item].attach(self.info_vbox, self):
        #        print "Plugin \"%s\" loaded" % item
        #    else:
        #        print "Plugin \"%s\" disabled" % item

    def show_hosts(self):
        """Show host from the config file"""
        data = []
        columns = [('visible', 'str', 'Name'),
                   ('visible', 'str', 'OS'),
                   ('visible', 'str', 'IP Address'),
                   ('novisible', 'str', 'Port')]
        for i in config.HOSTS:
            j = config.HOSTS[i]
            try:
                row = (i, j["os"], j["ip"], j["port"])
            except:
                row = (i, j["os"], j["ip"], "22")
            data.append(row)
        self.t = Treeview("list", columns, data, self.treeview)

    def show_users(self):
        """Show host from the config file"""
        data = []
        columns = [('visible', 'str', 'Nick'),
                   ('visible', 'str', 'Name'),
                   ('visible', 'str', 'Status')]
        for i in config.USERS:
            j = config.USERS[i]
            row = (i, j["name"], j["status"])
            data.append(row)
        self.t = Treeview("list", columns, data, self.treeview)

    def on_connect_btn_clicked(self, button):
        self.s = pxssh.pxssh()
        user = "root" #self.settings["user"]
        ip = self.settings["ip"]
        port = self.settings["port"]
        os = self.settings["os"]
        if self.s.login(ip, user, port=port):
            print "login!"
        else:
            print "no login!"

    def on_treeview_button_release_event(self, widget, event):
        #Get the path at the specific mouse position
        model, iter = widget.get_selection().get_selected()
        self.settings["ip"] = model.get_value(iter, 2)
        self.settings["port"] = model.get_value(iter, 3)
        self.settings["os"] = model.get_value(iter, 1)

    def on_table_combo_changed(self, widget):
        """Prepare data to show in treeview"""
        data = []
        notebook_wnd = None
        table = widget.get_active_text().lower()
        if table == "users":
            self.show_users()
        elif table == "hosts":
            self.show_hosts()

    def get_settings(self, key):
        return self.settings[key]

    def set_username(self, username):
        self.username = username

    def on_quit_activate(self, object):
        gtk.main_quit()

    def on_about_activate(self, object):
        about_wnd = AboutWnd("%s/main.glade" % config.GLADEDIR, "about_wnd")
        about_wnd.run()

    def on_main_wnd_destroy(self, window):
        gtk.main_quit()


class LoginWnd(SimpleGladeApp):
    def on_ok_btn_clicked(self, button):
        username = self.username_entry.get_text()
        self.main_widget.destroy()
        main_wnd = MainWnd("%s/main.glade" % config.GLADEDIR, "main_wnd")
        main_wnd.set_username(username)
        main_wnd.run()

    def on_login_wnd_destroy(self, window):
        gtk.main_quit()


class AboutWnd(SimpleGladeApp):
    def on_about_wnd_close(self, widget):
        gtk.main_quit()

    def on_about_wnd_destroy(self, window):
        gtk.main_quit()

# plugins base from http://lucumr.pocoo.org/blogarchive/setuptools-plugins
def init_plugins():
    pkg_resources.working_set.add_entry(PLUGIN_DIR)
    pkg_env = pkg_resources.Environment([PLUGIN_DIR])
    plugins = {}
    for name in pkg_env:
        egg = pkg_env[name][0]
        egg.activate()
        modules = []
        for name in egg.get_entry_map(ENTRYPOINT):
            entry_point = egg.get_entry_info(ENTRYPOINT, name)
            cls = entry_point.load()
            if not hasattr(cls, 'capabilities'):
                cls.capabilities = []
            instance = cls()
            for c in cls.capabilities:
                plugins.setdefault(c, []).append(instance)
    return plugins

plugins = init_plugins() #load plugins

def get_plugins_by_capability(capability):
    return plugins.get(capability, [])

def get_all_plugins():
    result = set()
    for p in plugins.itervalues():
        for plugin in p:
            result.add(plugin)
    return list(result)


def main():
    """Main loop"""
    aplicacion = LoginWnd("%s/main.glade" % config.GLADEDIR, "login_wnd")
    aplicacion.run()
    
if __name__ == "__main__":
    main()