runUtils.py

import glob
import subprocess as sp
import shutil
import shlex
import json
import os

def main():
    checkTokens()
    # checkProxy()
    prepareTokens()
    useToken("hephy")
    # getHeplxPublicFolder()

def checkTokens():
        prepareTokens()
        return True


def prepareTokens():
        on_system = getSystem()

        p = sp.Popen( shlex.split("klist"), stdout=sp.PIPE, stderr=sp.PIPE )
        (out,err) =  p.communicate()    
        for line in out.splitlines():

            if "Ticket cache" in line:
                token = line.split("FILE:")[1]

        if not os.path.exists("kerberos"):
            os.mkdir("kerberos")
        shutil.copyfile(token, "kerberos/krb5_token_"+on_system )

        if on_system == "cern.ch":
            if os.path.exists("kerberos/username"):
                with open("kerberos/username","r") as FSO:
                    user = FSO.read()
            else:
                user = raw_input("Give username for cell hephy.at: ")
                save = raw_input("You want to save '{0}' as your username for later? (Y/n)".format(user))
                if save != "n":
                    with open("kerberos/username","w") as FSO:
                        FSO.write(user)

            p = sp.Popen( shlex.split("kinit {0}@HEPHY.AT -c kerberos/krb5_token_hephy.at".format(user)) )
            p.communicate()
            os.environ["KRB5CCNAME"] = "kerberos/krb5_token_hephy.at"
            os.system("aklog -d hephy.at")
            os.environ["KRB5CCNAME"] = "kerberos/krb5_token_cern.ch"


def useToken(cell):

    if cell == "hephy":
        os.environ["KRB5CCNAME"] = "kerberos/krb5_token_hephy.at"
        # os.system("aklog -d hephy.at")
    if cell == "cern":
        os.environ["KRB5CCNAME"] = "kerberos/krb5_token_cern.ch"
        # os.system("aklog -d cern.ch")



def checkProxy():

        p = sp.Popen( shlex.split("voms-proxy-info"), stdout=sp.PIPE, stderr=sp.PIPE )
        (out,err) =  p.communicate()

        if err:
            print err
            return False

        info = {"time":-1,"path":""}
        for line in out.splitlines():
            if "timeleft" in line: info['time'] = int( line.replace(" ","").split("timeleft:")[1].replace(":","") )
            if "path" in line: info["path"] = line.split(":")[1].replace(" ","")

        if info['time'] > 0:
            if not os.path.exists("proxy"):
                os.mkdir("proxy")
            shutil.copyfile(info["path"], "proxy/x509_proxy")
            return True
        else:
            print "Proxy expired! Get a new one with 'voms-proxy-init --voms cms'"
            return False

def getHeplxPublicFolder():

    user = os.environ["USER"]
    first = user[0]
    if user == 'jaandrej' and getSystem()=="hephy.at":
        second = 'jandrejkovic'
    else :
        second = user[:8]
    return glob.glob( "/afs/{0}//user/{1}/{2}*/public".format(getSystem(),first,second) )[0]

def getSystem(inverse = False):
    host = os.environ["HOSTNAME"]
    if "heplx" in host: 
      if inverse: return "cern.ch"
      else: return "hephy.at"

    elif "lxplus" in host: 
      if inverse: return "hephy.at"
      else: return "cern.ch"

if __name__ == '__main__':
    main()