Python Colors.Colors类代码示例

 def undefine(self):
     """
     Usuwa maszynę
     
     @param void
     @return void
     """
     name = Params().getArg(1)
     if name == "":
         print(Colors.setred(" * ") + u"Nie podano nazwy maszyny wirtualnej")
         sys.exit(0)
     
     try:
         domain = BasicConsole._getDomain()
         
         if not Params().isParameter("force"):
             print(Colors.setyellow(" * ") + u"Czy na pewno chcesz usunąć definicję maszyny " + domain.name + " ? [y|n]"),
             out = raw_input()
             if out != "y":
                 print(Colors.setred(" * ") + u"Anulowano") 
                 sys.exit(0)
                             
         # Czyszczenie uprawnień
         domain.clearOwnership()
         # Usuwanie
         domain.undefine()
         
         # Usuwanie tymczasowych haseł
         GraphicPasswords().deletePassword(name)
         
         print(Colors.setgreen(" * ") + u"Usunięto maszynę wirtualną " + domain.name)
     except Exception as e:
         print(Colors.setred(" * ") + u"Błąd podczas usuwania maszyny wirtualnej. "),
         print(str(e).decode('utf-8'))

 def migrate(self):
     """
     Migracja
     """        
     name = Params().getArg(1)
     agentname = Params().getArg(2)
     
     domain = Domain(name)
     # Szukanie obecnego agenta domeny, sprawdzanie czy vm istnieje
     try:
         domain.searchAgent()
     except Exception as e:
         print(Colors.setred(" * ") + u"Błąd. "),
         print(str(e))
         sys.exit(0)
     
     # Sprawdzanie czy maszyna już jest na tym agencie
     if domain.getAgent().getHostName() == agentname:
         print(Colors.setgreen(" * ") + u"Maszyna już jest na agencie " + domain.getAgent().getHostName())
         sys.exit(1)
     
     print(Colors.setgreen(" * ") + u"Rozpoczęto migrację maszyny " + name + " z agenta " + domain.getAgent().getHostName())
     
     if domain.getState() == 1:
         if Params().isParameter("force"):
             domain.destroy()
         else:
             print(Colors.setgreen(" * ") + u"Maszyna jest uruchomiona. Wyłączyć? [y|n] "),
             out = raw_input()
             if out == "y":
                 domain.destroy()
             else:
                 print(Colors.setred(" * ") + u"Anulowano")
                 sys.exit(1)
     
     agents = getAgents()
     for agent in agents:
         if agent.getHostName() == agentname:
             print(Colors.setgreen(" * ") + u"Sprawdzanie połączenia z agentem " + agentname)
             # Pingowanie agenta
             #r = agent.ping()
             r = True
             if r != False:
                 print(Colors.setgreen(" * ") + u"Definiowanie maszyny na docelowym agencie")
                 xmldesc = domain.getXML()
                 # Definiowanie maszyny na nowym hoście
                 agent.defineXML(xmldesc)
                 # Usuwanie maszyny ze starej lokalizacji
                 print(Colors.setgreen(" * ") + u"Usuwanie maszyny z poprzedniego agenta")
                 domain.undefine()
                 
                 sys.exit(0)
             else:
                 print(Colors.setred(" * ") + u"Brak połączenia z agentem.")
                 sys.exit(2)
             
             sys.exit(0)
             
     print(Colors.setred(" * ") + u"Nie znaleziono definicji agenta " + agentname)
     sys.exit(1)

    def list(self):
        """
        Wyświetla listę wszystkich uprawnień
        
        @param void
        @return void
        """
        # Tworzenie zapytania
        q = Query("ownership-list")
        
        # Sprawdzenie czy podano nazwę użytkownika
        if Params().argLen() == 3:
            # Dodawanie uid do zapytania
            q.setData(Params().getArg(2))

        # Wykonywanie zapytania i uzyskiwanie odpowiedzi
        response = self._getRemoteObject().execute(q)
        self._getRemoteObject().quit()
        
        records = response.getData()
        
        if len(records) > 0:
            print(Colors.setbold("Nazwa domeny".ljust(31)) + Colors.setbold(u"Użytkownik"))
        
            # Wyświetlanie tabelki
            for record in records:
                print(record[0].ljust(30)),
                print(str(record[1]))

 def createSnapshotVm(self):
     template = Params().getArg(2)
     baselv = Params().getArg(3)
     size = Params().getArg(4)
     
     try:
         int(size)
     except ValueError:
         print(Colors.setred(" * ") + u"Niepoprawna wielkość snapshotu. Musi być liczbą całkowitą")
         sys.exit(0)
         
     balancer = Params().getArg(5)   
     if balancer == "":
         balancer = "round"
     
     # Tworzenie obiektu laboratorium
     lab = Lab("_hidden_vmsnapshots", baselv)  
     
     try:
         # Tworzenie snapshota
         lab.createSnapshots(1, size) 
     except Exception as e:
         print(Colors.setred(" * ") + u"Błąd. "),
         print(str(e).decode('utf-8'))
         sys.exit(0)
     
     # Tworzenie maszyn
     lab.createSnapshotMachines(template, 1, balancer)

    def add(self):
        """
        Dodaje uprawnienie
        
        @param void
        @return void
        """
        if Params().argLen() == 4:
            # Pobieranie parametrów
            name = Params().getArg(2)
            users = Params().getArgList(3)

            for user in users:
                # Tworzenie zapytania
                q = Query()
                q.setCommand("ownership-add")
                q.setData([user, name])
                
                # Wykonywanie zapytania
                self._getRemoteObject().execute(q)
                
                print(Colors.setgreen(" * ") + u"Dodano maszynę " + Colors.setbold(name) + u" użytkownikowi " + Colors.setbold(user))

            
            self._getRemoteObject().quit()

    def createBase(self, vg, name, size):
        """
        Tworzy obraz dysku maszyny bazowej
        
        @param Grupa woluminowa
        @param Nazwa
        @param Wielkość (w GB)
        """
        size = str(size)
        
        # Sprawdzanie czy podano jednostki, jeśli nie, dodawanie gigabajtów
        if size[-1:].isdigit():
            size += "G"
        
        # Polecenie 
        command = "lvcreate -L "+str(size) + "G -n " + name + " " + vg
        # Wykonanie polecenia
        (status, output) = commands.getstatusoutput(command)
        if (status == 0):                        
            # Dodaje bazę do mysql
            self.attachBase(vg, name)
            
            # Uaktualnienie Aavahi
            try:
                DisksUpdater().sendUpdate()
            except Exception:
                print(Colors.setred(" * ") + u"Błąd podczas odświeżania Avahi.")

            return True
        else:
            print(Colors.setred(" * ") + u"Tworzenie systemu bazowego nie powiodło się na poziomie tworzenia partycji")
            print(Colors.setred(" * ") + u"Komenda: ".ljust(12) + command)
            print(Colors.setred(" * ") + u"Komunikat: ".ljust(12) + output)
            
            return False

    def _getDomain():
        """
        Zwraca domenę
        
        @param void
        @return folavirt.remove.Domain
        """
        if Params().argLen() < 2:
            print(Colors.setred(" * ") + u"Musisz podać nazwę maszyny wirtualnej")
            sys.exit(1)           
            
        domain = Domain()
        
        splitted = Params().getArg(1).split("@")
        if len(splitted) > 1:
            agent = splitted[1]
            domain.name = splitted[0]
            try:
                domain.setAgentByName(agent)
            except Exception as e:
                print(Colors.setred(" * ") + u"Błąd! "),
                print(str(e).decode('utf-8'))
                sys.exit(1)
        else:   
            domain.name = Params().getArg(1)

            try:
                domain.searchAgent()
            except Exception as e:
                print(Colors.setred(" * ") + u"Błąd! " + str(e).decode('utf-8'))
                sys.exit(1)
        
        return domain

 def reload(self):
     """
     Przeładowywuje TGT
     """
     (status, output) = commands.getstatusoutput("tgt-admin -c " + self.conffilepath + " --update ALL -f")
     if status == 0:
         print(Colors.setgreen(" * ") + u"Odświeżono stan iscsid")
     else:
         print(Colors.setred(" * ") + u"Błąd podczas odświeżania iscsid")
         print(Colors.setred(" * ") + u"Output: " + str(output).decode('utf-8'))

 def clean(self):
     """
     Czyści uprawnienia do domeny lub wszystkie użytkownikowi
     """
     # Czyszczenie domeny
     if Params().getArg(2) == "vm":
         owns = self._getRemoteObject().execute(Query("ownership-list")).getData()
         
         if not Params().getArg(3) in [own[0] for own in owns]:
             print(Colors.setred(" * ") + u"Nie ma zdefiniowanych praw dla takiej maszyny wirtualnej")
             sys.exit(0)
         
         # Tworzenie zapytania
         q = Query()
         q.setCommand('ownership-cleandomain')
         q.setData(Params().getArg(3))
             
         self._getRemoteObject().execute(q)
         self._getRemoteObject().quit()
             
         print (Colors.setgreen(" * ") + u"Wyczyszczono wszystkie uprawnienia do domeny " + Params().getArg(3))
         
         sys.exit(0)
         
     # Czyszczenie uid
     if Params().getArg(2) == "user":
         users = Params().getArgList(3)
         
         owns = self._getRemoteObject().execute(Query("ownership-list")).getData()
         
         if len([own[1] for own in owns if own[1] in users]) == 0:
             print(Colors.setred(" * ") + u"Nie ma żadnych przypisanych maszyn dla podanych użytkowników")
             sys.exit(0)
         
         for user in users:
             if not user in [own[1] for own in owns if own[1] in users]:
                 print(Colors.setred(" * ") + u"Użytkownik " + user + u" nie posiada żadnych maszyn wirtualnych")
                 continue
             
             # Tworzenie zapytania
             q = Query()
             q.setCommand('ownership-cleanuser')
             q.setData(user)
                 
             # Wykonywanie zapytania
             self._getRemoteObject().execute(q)
                 
             print(Colors.setgreen(" * ") + u"Usunięto wszystkie uprawnienia użytkownikowi " + user)
             
         self._getRemoteObject().quit()
         
         sys.exit(0)
         
     self.printhelp()

 def printhelp(ljust1 = 37, ljust2 = 15):
     """
     Wyświetla pomoc
     
     @param void
     @return void
     """
     print("\n " + u"Obsługa zarządców dyskami")
     print("  " + Colors.setbold("disk discover [--dry-run]".ljust(ljust1)) + u"poszukiwanie maszyn z uruchomioną usługą foladiskd [bez aktualizuji etc/disks]")
     print("  " + Colors.setbold("disk list".ljust(ljust1)) + u"wyświetla listę zarządców dyskami")
     print("  " + Colors.setbold("disk lv list".ljust(ljust1)) + u"lista woluminów bazowych")

 def printhelp(ljust1 = 37, ljust2 = 15):
     """
     Pomoc dotycząca poleceń instalujących system
     
     @param void
     @return void
     """
     print("\n " + "Instalacja grup woluminowych")
     print("  " + Colors.setbold("vg attach $VG [--force]".ljust(ljust1)) + u"dodaje grupę woluminową do folavirt [z pominięciem sprawdzenia czy istnieje]")
     print("  " + Colors.setbold("vg detach $VG".ljust(ljust1)) + u"usuwa grupę woluminową z folavirt")
     print("  " + Colors.setbold("vg details $VG".ljust(ljust1)) + u"szczegóły grupy woluminowej")
     print("  " + Colors.setbold("vg list".ljust(ljust1)) + u"wyświetla listę grup woluminowych zarządzanych przez foladisk")

 def _start(self, pid):
     try:
         if pid == None:
             raise Exception('fresh start')
         os.kill(int(pid), 0)
         print(Colors.setred(" * ") + Params().getScriptName() + u" jest uruchomiony")
         sys.exit(1)
     except OSError:
         print(Colors.setgreen(" * ") + Params().getScriptName() + u" uruchomiony")
         return 1
     except Exception:
         print(Colors.setgreen(" * ") + Params().getScriptName() + u" uruchomiony")
         return 1

 def printhelp(ljust1 = 27, ljust2 = 15):
     """
     Pomoc dotycząca tworzenia maszyn bazowych
     
     @param void
     @return void
     """
     print("\n " + u"Zarządzanie woluminami bazowymi")
     print("  " + Colors.setbold("lv attach $VG $LV".ljust(ljust1)) + u"dodaje do folavirt wolumin bazowy, bez jego tworzenia")
     print("  " + Colors.setbold("lv create $VG $LV".ljust(ljust1)) + u"tworzy wolumin bazowy")
     print("  " + Colors.setbold("lv detach $VG $LV".ljust(ljust1)) + u"usuwa wolumin bazowy z folavirt, nie usuwa go")
     print("  " + Colors.setbold("lv list".ljust(ljust1)) + u"lista woluminów bazowych")
     print("  " + Colors.setbold("lv remove $VG $LV".ljust(ljust1)) + u"usuwa wolumin bazowy")

 def printhelp(ljust1 = 37, ljust2 = 15):
     """
     Wypisuje pomoc
     
     @param void
     @return void
     """
     print("\n " + u"Uprawnienia do maszyn wirtualnych")
     print("  " + Colors.setbold("ownership list".ljust(ljust1)) + u"wypisuje wszystkie prawa własności do domen")
     print("  " + Colors.setbold("ownership add $VM $U".ljust(ljust1)) + u"dodaje prawo własności")
     print("  " + Colors.setbold("ownership delete $VM $U".ljust(ljust1)) + u"zabiera prawo własności")
     print("  " + Colors.setbold("ownership clean vm $VM".ljust(ljust1)) + u"czyści wszystkie prawa do domeny")
     print("  " + Colors.setbold("ownership clean user $U".ljust(ljust1)) + u"zabiera wszystkie prawa użytkownikowi")

 def _attachdetachParams(self):
     vg = Params().getArg(2)
     name = Params().getArg(3)
     
     if Params().getArg(2) == "":
         print(Colors.setred(" * ") + u"Nie podano grupy woluminowej")
         sys.exit(0)
         
     if Params().getArg(3) == "":
         print(Colors.setred(" * ") + u"Nie podano nazwy woluminu")
         sys.exit(0)
         
     return {'vg':vg, 'name': name}