#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import pygtk
pygtk.require('2.0')
import gtk
import os
import getpass

class cb_exit:
  def disable_buttons(self):
    self.cancel.set_sensitive(False)
    self.logout.set_sensitive(False)
    self.suspend.set_sensitive(False)
    self.reboot.set_sensitive(False)
    self.shutdown.set_sensitive(False)

  def cancel_action(self,btn):
    self.disable_buttons()
    gtk.main_quit()

  def logout_action(self,btn):
    self.disable_buttons()
    self.status.set_label("Kijelentkezés, kérem várjon...")
    os.system("openbox --exit")

  def suspend_action(self,btn):
    self.disable_buttons()
    self.status.set_label("Felfüggesztés folyamatban, kérem várjon...")
    os.system("cb-lock")
    os.system("dbus-send --system --print-reply --dest=\"org.freedesktop.UPower\" /org/freedesktop/UPower org.freedesktop.UPower.Suspend")
    gtk.main_quit()

  def reboot_action(self,btn):
    self.disable_buttons()
    self.status.set_label("Újraindítás, kérem várjon...")
    os.system("dbus-send --system --print-reply --dest=\"org.freedesktop.ConsoleKit\" /org/freedesktop/ConsoleKit/Manager org.freedesktop.ConsoleKit.Manager.Restart")

  def shutdown_action(self,btn):
    self.disable_buttons()
    self.status.set_label("Leállítás, kérem várjon...")
    os.system("dbus-send --system --print-reply --dest=\"org.freedesktop.ConsoleKit\" /org/freedesktop/ConsoleKit/Manager org.freedesktop.ConsoleKit.Manager.Stop")

  def create_window(self):
    self.window = gtk.Window()
    title =  "Válasszon az alábbi lehetőségek közül:"
    self.window.set_title(title)
    self.window.set_border_width(5)
    self.window.set_size_request(500, 80)
    self.window.set_resizable(False)
    self.window.set_keep_above(True)
    self.window.stick
    self.window.set_position(1)
    self.window.connect("delete_event", gtk.main_quit)
    windowicon = self.window.render_icon(gtk.STOCK_QUIT, gtk.ICON_SIZE_MENU)
    self.window.set_icon(windowicon)


    #Create HBox for buttons
    self.button_box = gtk.HBox()
    self.button_box.show()

    #Cancel button
    self.cancel = gtk.Button(stock = gtk.STOCK_CANCEL)
    self.cancel.set_border_width(4)
    self.cancel.connect("clicked", self.cancel_action)
    self.button_box.pack_start(self.cancel)
    self.cancel.show()

    #Logout button
    self.logout = gtk.Button("_Kijelentkezés")
    self.logout.set_border_width(4)
    self.logout.connect("clicked", self.logout_action)
    self.button_box.pack_start(self.logout)
    self.logout.show()

    #Suspend button
    self.suspend = gtk.Button("_Felfüggesztés")
    self.suspend.set_border_width(4)
    self.suspend.connect("clicked", self.suspend_action)
    self.button_box.pack_start(self.suspend)
    self.suspend.show()

    #Reboot button
    self.reboot = gtk.Button("_Újraindítás")
    self.reboot.set_border_width(4)
    self.reboot.connect("clicked", self.reboot_action)
    self.button_box.pack_start(self.reboot)
    self.reboot.show()

    #Shutdown button
    self.shutdown = gtk.Button("_Leállítás")
    self.shutdown.set_border_width(4)
    self.shutdown.connect("clicked", self.shutdown_action)
    self.button_box.pack_start(self.shutdown)
    self.shutdown.show()

    #Create HBox for status label
    self.label_box = gtk.HBox()
    self.label_box.show()
    self.status = gtk.Label()
    self.status.show()
    self.label_box.pack_start(self.status)

    #Create VBox and pack the above HBox's
    self.vbox = gtk.VBox()
    self.vbox.pack_start(self.button_box)
    self.vbox.pack_start(self.label_box)
    self.vbox.show()

    self.window.add(self.vbox)
    self.window.show()

  def __init__(self):
    self.create_window()


def main():
    gtk.main()

if __name__ == "__main__":
    go = cb_exit()
    main()