Ubuntu Sverige

Kod: Markera allt

'''
Created on 6 okt 2012

@author: andersm
'''
import gtk
import appindicator
import sage_server
import os
import time

buf = ["Starta server", "Stoppa server", "Avsluta"]
running_icon = os.getenv("HOME") + "/.sage/sagemath-running.png"
stopped_icon = os.getenv("HOME") + "/.sage/sagemath-stopped.png"
        
class SageControll(appindicator.Indicator):

    def menuitem_response(self, w, itmtxt):
        if itmtxt == buf[2] :
            print "Avslutar"
            dialog = gtk.MessageDialog(None, gtk.DIALOG_MODAL,
                                       gtk.MESSAGE_INFO, gtk.BUTTONS_YES_NO,
                                       "Avsluta?")
            dialog.set_title(":(")
    
            response = dialog.run()
            dialog.destroy()
            if response == gtk.RESPONSE_YES:
                gtk.main_quit()
        elif itmtxt == buf[1] :
            print "Stoppar server"
            sage_server.stop_server()
            self.set_status(appindicator.STATUS_ACTIVE)
        elif itmtxt == buf[0] :
            print "Startar server"
            sage_server.start_server()
	    rng = False
            while not rng :
                self.set_status(appindicator.STATUS_ACTIVE)
                time.sleep(0.5)
                rng = sage_server.get_server_status()
                self.set_status(appindicator.STATUS_ATTENTION)
                time.sleep(0.5)
        else :
            print "confused"
            
            
    def __init__(self):
        appindicator.Indicator.__init__(self, "sage control",
                              stopped_icon,
                              appindicator.CATEGORY_APPLICATION_STATUS)
        self.set_attention_icon(running_icon)
        self.set_status (appindicator.STATUS_ACTIVE)
                    # create a menu
        menu = gtk.Menu()
    
        # create items
        for i in range(len(buf)) :
            menu_items = gtk.MenuItem(buf[i])
            menu.append(menu_items)
            menu_items.connect("activate", self.menuitem_response, buf[i])
            menu_items.show()
    
        self.set_menu(menu)


if __name__ == "__main__":
    ind = SageControll()
    gtk.main()

Kod: Markera allt

"""
Sage server starting and stopping script

AUTHORS:

    - Kwankyu Lee (2010-10-17): initial version

    """

    #*****************************************************************************
    #       Copyright (C) 2010 Kwankyu Lee <ekwankyu@gmail.com>
    #
    #  Distributed under the terms of the GNU General Public License (GPL)
    #                  http://www.gnu.org/licenses/
    #*****************************************************************************
#Modifierat av Anders Martinsson 2012

import os,sys
import pexpect
import time              
import subprocess,signal

from time import strftime


sage_server_root = os.environ['HOME'] + '/.sage'
conf_file_name = sage_server_root + '/sage_server.conf'
log_file_name = sage_server_root + '/sage_server.log'

UPGRADE_CONFIRM = r'Do you want to continue [y/N]?'

MAX_WAITING_TIME_BEFORE_KILLING = 30 # in seconds

def print_help():
    """
    Prints a short usage description.
    """

    print "Usage: sage-server [command]"
    print "Commands: start, stop, restart, upgrade, status"
    print ""
    print "The server configurations should be listed in the file `sage-server.conf'"
    print "in `.sage' directory under your home directory."

def get_config():
    arg=[]
    with open(conf_file_name) as f:
        for line in f: arg.append(line.strip())
    while arg.count('') > 0: arg.remove('')
    return arg

    
def get_server_status():
    """
    Returns the status of the Sage server.
    """

    try:
        # if sage notebook server is running, "twist.pid" was created in the notebook directory.
        pid_file_name = sage_server_root + '/sage_notebook.sagenb/twistd.pid'
        pid_file = open(pid_file_name, 'r')
        pid = int(pid_file.read()) 
        pid_file.close()
    except IOError:                        
        return False
    except ValueError: # remove the invalid pid file
        os.remove(pid_file_name)
        return False

    # check validity of the pid by checking the pid belongs 
    # to the current user's sage twistd server process.
    c1 = subprocess.Popen(['ps','-p',str(pid),'-o','user='],stdout=subprocess.PIPE)
    c2 = subprocess.Popen(['ps','-p',str(pid),'-o','cmd='],stdout=subprocess.PIPE)
    if os.environ['USER'] in c1.communicate()[0] and 'twistd' in c2.communicate()[0]:
        return True
    else:
        return False

def start_server():
    """
    Starts the Sage server.
    """
    if get_server_status():
        print "The Sage server is already running."
        return

    try:
        print "Starting the Sage server..."
	proc = ['sage', '-notebook']
	proc.extend(get_config()) 
        subprocess.Popen(proc)
    except IOError:
        print "Failed to start the Sage server. Check the configuration file."
        sys.exit(os.EX_CONFIG)

def stop_server():                               
    """
    Stops the Sage server.
    """
    if not get_server_status():
        print "There is no Sage server running." 
        return

    pid_file_name = sage_server_root + '/sage_notebook.sagenb/twistd.pid'
    pid_file = open(pid_file_name, 'r')
    pid = int(pid_file.readline()) 
    pid_file.close()

    print "Stopping the Sage server..."
    os.kill(pid,signal.SIGTERM)
    elapsed_time = 0
    while os.path.isfile(pid_file) and elapsed_time < MAX_WAITING_TIME_BEFORE_KILLING:
        time.sleep(.1)
        elapsed_time += .1

    if os.path.isfile(pid_file): # assume the server process is malbehaving.
        print "Killing the errorneous Sage server process..." 
        os.kill(pid,signal.SIGKILL)
        os.remove(pid_file)

def upgrade_sage():
    """
    Upgrades the Sage skipping the upgrade confirmation.
    """
    print "Upgrading Sage..."
    child = pexpect.spawn(sage_root + '/sage -upgrade')
    child.logfile = sys.stdout

    while True:
        i = child.expect([pexpect.EOF,UPGRADE_CONFIRM,pexpect.TIMEOUT],timeout=1)
        if i == 0:
            print "Upgrading finished."
            return 
        elif i == 1: # confirm to upgrade
            child.sendline('y')
        elif i == 2: # upgrading is going on
            continue 
        else:
            print "Something wrong..."
            sys.exit(os.EX_SOFTWARE)

def log(log_file,message):
    """
    A function for logging.
    """
    current_time = strftime("%Y-%m-%d %H:%M:%S")
    log_file.write(current_time + ': ' + message + '\n')

if __name__ == "__main__":
    if len(sys.argv) > 1:
        if os.environ.has_key('SAGE_ROOT'):
            sage_root = os.environ['SAGE_ROOT']
        else:
            print "The SAGE_ROOT environment variable is undefined."
            sys.exit(os.EX_CONFIG)

        try:
            log_file = open(log_file_name, 'a')
        except IOError:         
            print "Failed to open the log file."
            sys.exit(os.EX_IOERR)

        cmd = sys.argv[1]
        if cmd == 'start':
            start_server()
            log(file,'sage server started')
        elif cmd == 'stop':
            stop_server()
            log(file,'sage server stopped')
        elif cmd == 'restart':
            stop_server()
            start_server()
            log(file,'The sage server restarted')
        elif cmd == 'upgrade':
            running = get_server_status()
            if running:
                stop_server()
            log(file,'Upgrading the sage started')
            upgrade_sage()
            log(file,'Upgrading the sage finished')
            if running:
                start_server()
        elif cmd == 'status':
            status = get_server_status()
            if status:
                print "The Sage server is running."
            else:
                print "There is no Sage server running."
        else:
            print_help()
        file.close()
    else:
        print_help()