source: lliurex-mirror/trunk/fuentes/lliurex-mirror-cli/usr/bin/lliurex-mirror @ 2516

#!/usr/bin/env python
import xmlrpclib as x
import sys
import getpass
from clint import arguments
from clint.textui import puts, indent, colored
import os

class LliurexMirror(object):
        """docstring for LliurexMirror"""
        def __init__(self,credentials,host):
                super(LliurexMirror, self).__init__()
                self.ip = host if host != None else "localhost"
                self.credentials = credentials
                self.client = x.ServerProxy("https://"+self.ip+":9779")

        def setCname(self):
                result = self.client.set_cname(self.credentials,"MirrorManager")
                print result['msg']
                sys.exit( 0 if result['status'] else 1 )

        def runDebmirror(self,distro):
                result = self.client.update(self.credentials,"MirrorManager",'',distro,None)
                print result['msg']
                sys.exit( 0 if result['status'] else 1 )

        def isAlive(self):
                result = self.client.is_alive(self.credentials,"MirrorManager")
                print "True" if result['status'] else "False"
                sys.exit( 0 if result['status'] else 1 )

        def getMirrorArchitecture(self,distro):
                result = self.client.get_mirror_architecture(self.credentials,"MirrorManager",distro)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def setMirrorArchitecture(self,distro,archs):
                result = self.client.set_mirror_architecture(self.credentials,"MirrorManager",distro,archs)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def getMirrorOrig(self,distro):
                result = self.client.get_mirror_orig(self.credentials,"MirrorManager",distro)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def setMirrorOrig(self,distro,url):
                result = self.client.set_mirror_orig(self.credentials,"MirrorManager",distro,url)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def getChecksumValidation(self,distro):
                result = self.client.get_checksum_validation(self.credentials,"MirrorManager",distro)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def setChecksumValidation(self,distro,value):
                check = 0 if not value else 1
                result = self.client.set_checksum_validation(self.credentials,"MirrorManager",distro,check)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def getPercentage(self,distro):
                result = self.client.get_percentage(self.credentials,"MirrorManager",distro)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def enableWebserverForPath(self,path):
                result = self.client.enable_webserver_into_folder(self.credentials,"MirrorManager",path)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def stopWebserver(self,port):
                result = self.client.stop_webserver(self.credentials,"MirrorManager",port)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def stopUpdate(self):
                result = self.client.stopupdate(self.credentials,"MirrorManager")
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def getAvailableDistros(self):
                result = self.client.get_available_mirrors(self.credentials,"MirrorManager")
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )

        def isMirrorUpdate(self,distro):
                result = self.client.is_update_available(self.credentials,"MirrorManager",distro)
                print str(result['msg'])
                sys.exit( 0 if result['status'] else 1 )        

def usage():
        puts("Usage")
        with indent(4):
                puts("lliurex-mirror [FLAGS...] ACTION")
                puts("Actions")
                with indent(4):
                        puts("setcname")
                        puts("rundebmirror DISTRO")
                        puts("isalive")
                        puts("getmirrorarchitecture")
                        puts("setmirrorarchitecture [ARCH ARCH ...]")
                        puts("getmirrororig")
                        puts("setmirrororig URL")
                        puts("getchecksumvalidation")
                        puts("setchecksumvalidation [0|1]")
                        puts("getpercentage")
                        puts("enablewebserverforpath PATH")
                        puts("stopwebserver PORT")
                        puts("stopupdate")
                        puts("getavailabledistros")
                        puts("ismirrorupdate")
                puts("Flags")
                with indent(4):
                        puts("-h --help : Show help")
                        puts("-H --host : Remove ip to launch mirror actions")
                        puts("-k --key : Magic key to run lliurex-mirror")
                        puts("-u --user : User to validation")
                        puts("-p --password : Password to validation")

                        
        sys.exit(1)

if __name__ == '__main__':
        args = arguments.Args().copy
        host = None
        credentials = ""
        user = None
        password = None
        if args.contains(["-h","--help"]) or len(args.all) == 0 :
                usage()
        
        if args.contains(["-H","--host"]):
                #host = args.value_after()
                host = args.value_after("-H") if args.value_after("-H") != None else args.value_after("--host")
                if host == None:
                        usage()


        if not args.contains(["-k","--key","-u","--user","-p","--password","-P","--intpassword"]):
                if os.access('/etc/n4d/key',os.R_OK):
                        f = open("/etc/n4d/key","r")
                        credentials = f.readlines()[0].strip()

        if args.contains(["-u","--user"]):
                user = args.value_after("-u") if args.value_after("-u") != None else args.value_after("--user")
                if user == None:
                        usage()
                index = args.first(["-u","--user"])
                args.pop(index)
                args.pop(index)

        if args.contains(["-p","--password"]):
                password = args.value_after("-p") if args.value_after("-p") != None else args.value_after("--password")
                if password == None:
                        usage()

                index = args.first(["-p","--password"])
                args.pop(index)
                args.pop(index)

        if args.contains(["-P","--intpassword"]):
                if args.contains(["-u","--user"]):
                        password = getpass.getpass("Password: ")
                        index = args.first(["-P","--intpassword"])
                        args.pop(index)
                else:
                        usage()

        if user != None and password != None:
                credentials = (user,password)

        if args.contains(["-k","--key"]):
                credentials = args.value_after("-k") if args.value_after("-k") != None else args.value_after("--key")
                if credentials == None:
                        usage()
                index = args.first(["-k","--key"])
                args.pop(index)
                args.pop(index)

        action = args.pop(0)
        lliurexmirror = LliurexMirror(credentials,host)
        if action == "setcname":
                lliurexmirror.setCname()
        elif action == "rundebmirror":
                lliurexmirror.runDebmirror(args.get(0))
        elif action == "isalive":
                lliurexmirror.isAlive()
        elif action == "getmirrorarchitecture":
                lliurexmirror.getMirrorArchitecture(args.get(0))
        elif action == "setmirrorarchitecture":
                lliurexmirror.setMirrorArchitecture(args.pop(0),args.all)
        elif action == "getmirrororig":
                lliurexmirror.getMirrorOrig(args.get(0))
        elif action == "setmirrororig":
                lliurexmirror.setMirrorOrig(args.get(0),args.get(1))
        elif action == "getchecksumvalidation":
                lliurexmirror.getChecksumValidation(args.get(0))
        elif action == "setchecksumvalidation":
                checkvalue = 1 if args.get(1) == "True" else 0
                lliurexmirror.setChecksumValidation(args.get(0),checkvalue)
        elif action == "getpercentage":
                lliurexmirror.getPercentage(args.get(0))
        elif action == "enablewebserverforpath":
                lliurexmirror.enableWebserverForPath(args.get(0))
        elif action == "stopwebserver":
                lliurexmirror.stopWebserver(args.get(0))
        elif action == "getavailabledistros":
                lliurexmirror.getAvailableDistros()
        elif action == "stopupdate":
                lliurexmirror.stopUpdate()
        elif action == "ismirrorupdate":
                lliurexmirror.isMirrorUpdate(args.get(0))