Skip to content

Commit

Permalink
Fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
estevao90 committed Dec 22, 2023
1 parent 877678f commit ffe0951
Show file tree
Hide file tree
Showing 10 changed files with 214 additions and 108 deletions.
24 changes: 15 additions & 9 deletions vmm_manager/entidade/acao.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,22 +90,27 @@ def was_executada_com_sucesso(self):
def get_resultado_execucao(self):
if self.__status_execucao is None:
raise AttributeError('Ação não executada.')

if self.__status_execucao:
return self.__retorno_execucao

return {'Msgs': self.__retorno_execucao}

def get_cmd_pos_execucao(self, agrupamento, servidor_acesso):
if self.is_criacao_vm():
cmd = Comando('criar_vm_pos',
descricao=f"Taguear VM { self.args['nome_vm']}",
servidor_vmm=servidor_acesso.servidor_vmm,
campo_agrupamento=CAMPO_AGRUPAMENTO[0],
campo_id=CAMPO_ID[0],
campo_imagem=CAMPO_IMAGEM[0],
campo_regiao=CAMPO_REGIAO[0],
campo_rede_principal=CAMPO_REDE_PRINCIPAL[0],
agrupamento=agrupamento)
cmd = Comando(
'criar_vm_pos',
descricao=f"Taguear VM { self.args['nome_vm']}",
servidor_vmm=servidor_acesso.servidor_vmm,
campo_agrupamento=CAMPO_AGRUPAMENTO[0],
campo_id=CAMPO_ID[0],
campo_imagem=CAMPO_IMAGEM[0],
campo_regiao=CAMPO_REGIAO[0],
campo_rede_principal=CAMPO_REDE_PRINCIPAL[0],
agrupamento=agrupamento
)
cmd.args.update(self.args)

return cmd

raise AttributeError(
Expand All @@ -114,6 +119,7 @@ def get_cmd_pos_execucao(self, agrupamento, servidor_acesso):
def get_str_impressao_inline(self):
cmd_args = ', '.join([f'{arg}={value}'
for arg, value in self.args.items()])

return f'{self.nome_comando} - [{cmd_args}]'

def __eq__(self, other):
Expand Down
18 changes: 10 additions & 8 deletions vmm_manager/entidade/inventario.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,14 +159,16 @@ def validar(self, servidor_acesso):

redes.update([rede.nome for rede in maquina_virtual.redes])

cmd = Comando('validar_inventario', imagens=imagens,
nuvem=self.nuvem,
redes=redes,
servidor_vmm=servidor_acesso.servidor_vmm,
qtde_minima_regioes=len(regioes),
agrupamento=self.agrupamento,
lista_nome_vms_str=self.lista_nome_vms_str(),
campo_agrupamento=CAMPO_AGRUPAMENTO[0])
cmd = Comando(
'validar_inventario', imagens=imagens,
nuvem=self.nuvem,
redes=redes,
servidor_vmm=servidor_acesso.servidor_vmm,
qtde_minima_regioes=len(regioes),
agrupamento=self.agrupamento,
lista_nome_vms_str=self.lista_nome_vms_str(),
campo_agrupamento=CAMPO_AGRUPAMENTO[0]
)
_, msg = cmd.executar(servidor_acesso)
if msg:
raise ValueError(msg)
Expand Down
8 changes: 5 additions & 3 deletions vmm_manager/entidade/plano_execucao.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,9 +161,11 @@ def __limpar_guids(self, servidor_acesso, ocultar_progresso):
imprimir_acao_corrente(
'Limpando objetos temporários', ocultar_progresso)

cmd = Comando('limpar_objs_criacao_vm',
servidor_vmm=servidor_acesso.servidor_vmm,
guids=self.__guids_a_limpar)
cmd = Comando(
'limpar_objs_criacao_vm',
servidor_vmm=servidor_acesso.servidor_vmm,
guids=self.__guids_a_limpar
)
status, retorno = cmd.executar(servidor_acesso)

if status:
Expand Down
98 changes: 60 additions & 38 deletions vmm_manager/entidade/vm.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,21 @@


class VM:
def __init__(self, nome, descricao,
imagem, regiao,
qtde_cpu, qtde_ram_mb, redes,
id_vmm=None,
virtualizacao_aninhada=False,
memoria_dinamica=True,
status=VMStatusEnum.EM_EXECUCAO,
no_regiao=None):
def __init__(
self,
nome,
descricao,
imagem,
regiao,
qtde_cpu,
qtde_ram_mb,
redes,
id_vmm=None,
virtualizacao_aninhada=False,
memoria_dinamica=True,
status=VMStatusEnum.EM_EXECUCAO,
no_regiao=None
):
self.nome = nome
self.descricao = descricao if not descricao is None else ''
self.imagem = imagem
Expand Down Expand Up @@ -76,8 +83,12 @@ def add_acoes_diferenca_discos_adicionais(self, vm_remota, plano_execucao):
data_disco.get_acoes_diferenca_disco(
vm_remota.discos_adicionais[nome_disco], vm_remota.id_vmm, self.nome))

def add_acoes_diferenca_regiao(self, vm_remota,
plano_execucao, inv_remoto):
def add_acoes_diferenca_regiao(
self,
vm_remota,
plano_execucao,
inv_remoto
):
if self.regiao == SCRegion.REGIAO_PADRAO:
return

Expand Down Expand Up @@ -127,44 +138,55 @@ def get_rede_principal(self):
return next((rede.nome for rede in self.redes if rede.principal), None)

def get_acao_criar_vm(self):
return Acao(Acao.ACAO_CRIAR_VM,
nome_vm=self.nome,
descricao=self.descricao,
imagem=self.imagem,
regiao=self.regiao,
qtde_cpu=self.qtde_cpu,
qtde_ram_mb=self.qtde_ram_mb,
memoria_dinamica=self.memoria_dinamica,
redes=[rede.nome for rede in self.redes],
rede_principal=self.get_rede_principal()
)
return Acao(
Acao.ACAO_CRIAR_VM,
nome_vm=self.nome,
descricao=self.descricao,
imagem=self.imagem,
regiao=self.regiao,
qtde_cpu=self.qtde_cpu,
qtde_ram_mb=self.qtde_ram_mb,
memoria_dinamica=self.memoria_dinamica,
redes=[rede.nome for rede in self.redes],
rede_principal=self.get_rede_principal()
)

def get_acao_excluir_vm(self):
return Acao(Acao.ACAO_EXCLUIR_VM,
id_vm=self.id_vmm)
return Acao(
Acao.ACAO_EXCLUIR_VM,
id_vm=self.id_vmm
)

def get_acao_mover_vm_regiao(self, id_no_regiao):
return Acao('mover_vm_regiao',
nome_vm=self.nome,
id_no_regiao=id_no_regiao,
regiao=self.regiao)
return Acao(
'mover_vm_regiao',
nome_vm=self.nome,
id_no_regiao=id_no_regiao,
regiao=self.regiao
)

def get_acao_atualizar_virtualizacao_aninhada(self):
return Acao('atualizar_virtualizacao_aninhada',
nome_vm=self.nome,
virtualizacao_aninhada=self.virtualizacao_aninhada)
return Acao(
'atualizar_virtualizacao_aninhada',
nome_vm=self.nome,
virtualizacao_aninhada=self.virtualizacao_aninhada
)

def get_acao_atualizar_memoria_dinamica(self):
return Acao('atualizar_memoria_dinamica',
nome_vm=self.nome,
memoria_dinamica=self.memoria_dinamica)
return Acao(
'atualizar_memoria_dinamica',
nome_vm=self.nome,
memoria_dinamica=self.memoria_dinamica
)

def get_acao_atualizar_vm(self, id_vm_remota):
return Acao('atualizar_vm',
id_vm=id_vm_remota,
descricao=self.descricao,
qtde_cpu=self.qtde_cpu,
qtde_ram_mb=self.qtde_ram_mb)
return Acao(
'atualizar_vm',
id_vm=id_vm_remota,
descricao=self.descricao,
qtde_cpu=self.qtde_cpu,
qtde_ram_mb=self.qtde_ram_mb
)

def __hash__(self):
return hash(self.nome)
Expand Down
54 changes: 32 additions & 22 deletions vmm_manager/entidade/vm_disco.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,36 +39,46 @@ def get_tamanho_tipo_create(self):
return self.tamanho_tipo.name

def get_acao_criar_disco(self, nome_vm):
return Acao('criar_disco_vm',
nome_vm=nome_vm,
tipo=self.tipo.value,
tamanho_mb=self.tamanho_mb,
tamanho_tipo=self.get_tamanho_tipo_create(),
arquivo=self.arquivo,
caminho=self.caminho)
return Acao(
'criar_disco_vm',
nome_vm=nome_vm,
tipo=self.tipo.value,
tamanho_mb=self.tamanho_mb,
tamanho_tipo=self.get_tamanho_tipo_create(),
arquivo=self.arquivo,
caminho=self.caminho
)

def get_acao_excluir_disco(self, id_vm):
return Acao(Acao.ACAO_EXCLUIR_DISCO_VM,
id_vm=id_vm,
id_drive=self.get_id_drive())
return Acao(
Acao.ACAO_EXCLUIR_DISCO_VM,
id_vm=id_vm,
id_drive=self.get_id_drive()
)

def get_acao_expandir_disco(self, id_vm, id_drive):
return Acao('expandir_disco_vm',
id_vm=id_vm,
id_drive=id_drive,
tamanho_mb=self.tamanho_mb)
return Acao(
'expandir_disco_vm',
id_vm=id_vm,
id_drive=id_drive,
tamanho_mb=self.tamanho_mb
)

def get_acao_mover_disco(self, id_vm, id_disco):
return Acao('mover_disco_vm',
id_vm=id_vm,
id_disco=id_disco,
caminho=self.caminho)
return Acao(
'mover_disco_vm',
id_vm=id_vm,
id_disco=id_disco,
caminho=self.caminho
)

def get_acao_converter_disco(self, id_vm, id_drive):
return Acao('converter_disco_vm',
id_vm=id_vm,
id_drive=id_drive,
tamanho_tipo=self.get_tamanho_tipo_create())
return Acao(
'converter_disco_vm',
id_vm=id_vm,
id_drive=id_drive,
tamanho_tipo=self.get_tamanho_tipo_create()
)

def get_acoes_diferenca_disco(self, disco_remoto, id_vm, nome_vm):
acoes = []
Expand Down
11 changes: 7 additions & 4 deletions vmm_manager/infra/servidor_acesso.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,10 +124,13 @@ def __enviar_arquivo(self, nome_arquivo):
return resultado

if self.__is_conexao_sftp_ok():
file_attrs = self.conexao_sftp.put(nome_arquivo,
ServidorAcesso.__get_caminho_arquivo(
nome_arquivo),
confirm=True)
file_attrs = self.conexao_sftp.put(
nome_arquivo,
ServidorAcesso.__get_caminho_arquivo(
nome_arquivo
),
confirm=True
)
return True, file_attrs

return False, self.get_msg_erro_conexao_sftp()
Expand Down
8 changes: 6 additions & 2 deletions vmm_manager/parser/parser_local.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,8 +55,12 @@ def __validar_arquivo_yaml(self):
if os.stat(self.__arquivo_inventario).st_size == 0:
raise ValueError('Arquivo de inventário vazio.')

def __montar_inventario(self, dados_inventario,
filtro_nome_vm=None, filtro_dados_completos=True):
def __montar_inventario(
self,
dados_inventario,
filtro_nome_vm=None,
filtro_dados_completos=True
):
nomes_vm = []
self.__inventario = Inventario(
dados_inventario['agrupamento'], dados_inventario['nuvem'])
Expand Down
17 changes: 13 additions & 4 deletions vmm_manager/parser/parser_remoto.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,13 @@
"""
import json
from vmm_manager.infra.comando import Comando
from vmm_manager.util.config import (CAMPO_AGRUPAMENTO, CAMPO_ID, CAMPO_IMAGEM,
CAMPO_REGIAO, CAMPO_REDE_PRINCIPAL)
from vmm_manager.util.config import (
CAMPO_AGRUPAMENTO,
CAMPO_ID,
CAMPO_IMAGEM,
CAMPO_REGIAO,
CAMPO_REDE_PRINCIPAL
)
from vmm_manager.entidade.inventario import Inventario
from vmm_manager.entidade.vm import VM
from vmm_manager.entidade.vm_rede import VMRede
Expand Down Expand Up @@ -100,8 +105,12 @@ def __get_discos_adicionais(self, servidor_acesso):

return discos_vms

def __montar_inventario(self, servidor_acesso,
filtro_nome_vm=None, filtro_dados_completos=True):
def __montar_inventario(
self,
servidor_acesso,
filtro_nome_vm=None,
filtro_dados_completos=True
):
self.__inventario = Inventario(self.agrupamento, self.nuvem)

vms_servidor = json.loads(
Expand Down
10 changes: 8 additions & 2 deletions vmm_manager/util/operacao.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,14 @@ def validar_retorno_operacao_sem_lock(status, msg, ocultar_progresso):
finalizar_com_erro(msg)


def validar_retorno_operacao_com_lock(status, msg, servidor_acesso,
agrupamento, nuvem, ocultar_progresso):
def validar_retorno_operacao_com_lock(
status,
msg,
servidor_acesso,
agrupamento,
nuvem,
ocultar_progresso
):
if status:
imprimir_ok(ocultar_progresso)
else:
Expand Down
Loading

0 comments on commit ffe0951

Please sign in to comment.