diff options
Diffstat (limited to 'tcosmonitor/extensions/reboot.py')
-rw-r--r-- | tcosmonitor/extensions/reboot.py | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/tcosmonitor/extensions/reboot.py b/tcosmonitor/extensions/reboot.py new file mode 100644 index 0000000..be943cb --- /dev/null +++ b/tcosmonitor/extensions/reboot.py @@ -0,0 +1,120 @@ +# -*- coding: UTF-8 -*- +# TcosMonitor version __VERSION__ +# +# Copyright (c) 2006-2011 Mario Izquierdo <mariodebian@gmail.com> +# +# This package is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# +# This package 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. +""" template extension """ + +from gettext import gettext as _ +import sys + +from tcosmonitor import shared +from tcosmonitor.TcosExtensions import TcosExtension + + +def print_debug(txt): + if shared.debug: + print >> sys.stderr, "%s::%s" % (__name__, txt) + #print("%s::%s" % (__name__, txt), file=sys.stderr) + + +class RebootPoweroff(TcosExtension): + def register(self): + self.main.menus.register_simple(_("Reboot"), "menu_reboot.png", 0, self.reboot, "reboot") + self.main.menus.register_all( _("Reboot all clients"), "menu_reboot.png", 0, self.reboot_all, "reboot") + self.main.menus.register_simple( _("Poweroff"), "menu_poweroff.png", 0, self.poweroff, "reboot") + self.main.menus.register_all( _("Poweroff all clients"), "menu_poweroff.png", 0, self.poweroff_all, "reboot") + + def reboot(self, w, ip): + if not self.get_client(): + return + msg=_( _("Do you want to reboot: %s?" ) %(self.connected_users_txt_all) ) + if shared.ask_msg ( msg ): + timeout=self.main.config.GetVar("actions_timeout") + msg=(_("Pc will reboot in %s seconds") %timeout) + self.exe_app_in_client("reboot", timeout, msg, users=[ip], connected_users=self.connected_users) + + + def reboot_all(self, *args): + if not self.get_all_clients(): + return + msg=_( _("Do you want to reboot the following clients: %s?" ) %(self.connected_users_txt_all) ) + if shared.ask_msg ( msg ): + timeout=self.main.config.GetVar("actions_timeout") + msg=(_("Pc will reboot in %s seconds") %timeout) + self.exe_app_in_client("reboot", timeout, msg, users=self.allclients, connected_users=self.connected_users) + + def poweroff(self, w, ip): + if not self.get_client(): + return + msg=_( _("Do you want to poweroff: %s?" ) %(self.connected_users_txt_all) ) + if shared.ask_msg ( msg ): + timeout=self.main.config.GetVar("actions_timeout") + msg=(_("Pc will shutdown in %s seconds") %timeout) + self.exe_app_in_client("poweroff", timeout, msg, users=[ip], connected_users=self.connected_users) + + + def poweroff_all(self, *args): + if not self.get_all_clients(): + return + msg=_( _("Do you want to poweroff the following clients: %s?" )%(self.connected_users_txt_all) ) + if shared.ask_msg ( msg ): + timeout=self.main.config.GetVar("actions_timeout") + msg=(_("Pc will shutdown in %s seconds") %timeout) + self.exe_app_in_client("poweroff", timeout, msg, users=self.allclients, connected_users=self.connected_users) + + def real_action(self, ip, action): + print_debug("real_action() ip=%s action='%s'"%(ip, action) ) + self.main.xmlrpc.Exe(action) + + + def exe_app_in_client(self, mode, timeout=0, msg="", users=[], connected_users=[]): + remote_cmd=("/usr/lib/tcos/session-cmd-send %s %s %s" %(mode.upper(), timeout, msg.replace("'", "ยด"))) + action="down-controller %s %s" %(mode, timeout) + print_debug("exe_app_in_client() usernames=%s" %users) + + if len(connected_users) != 0 and connected_users[0] != shared.NO_LOGIN_MSG: + newusernames=[] + for user in connected_users: + if user.find(":") != -1: + # we have a standalone user... + usern, ip = user.split(":") + self.main.xmlrpc.newhost(ip) + self.main.xmlrpc.DBus("exec", remote_cmd ) + else: + # we have a thin client + newusernames.append(user) + + result = self.main.dbus_action.do_exec( newusernames, remote_cmd ) + + if not result: + shared.error_msg ( _("Error while exec remote app:\nReason:%s") %( self.main.dbus_action.get_error_msg() ) ) + + self.main.worker=shared.Workers(self.main, None, None) + self.main.worker.set_for_all_action(self.action_for_clients, users, action ) + return + + + + +__extclass__=RebootPoweroff + + + + + + |