From d45e74234c043f0adb6b71da0c322e20e6f6eff6 Mon Sep 17 00:00:00 2001
From: Eduardo Calil <dudusaid@gmail.com>
Date: Fri, 12 Aug 2016 12:54:33 -0300
Subject: [PATCH] Atualiza codigo para o nome dos novos grupos

---
 sapl/utils.py                                 |  8 +++---
 scripts/inicializa_grupos_autorizacoes.py     | 27 ++++++++++---------
 .../test_inicializa_grupos_autorizacoes.py    | 12 ---------
 3 files changed, 19 insertions(+), 28 deletions(-)

diff --git a/sapl/utils.py b/sapl/utils.py
index 088319563..5b1bee282 100644
--- a/sapl/utils.py
+++ b/sapl/utils.py
@@ -261,7 +261,7 @@ def permissoes_parlamentares():
 def permissoes_protocoloadm():
     lista_permissoes = []
     perms_protocolo = Permission.objects.filter(
-        group__name='Operador de Protocolo')
+        group__name='Operador de Protocolo Administrativo')
     for p in perms_protocolo:
         lista_permissoes.append('protocoloadm.' + p.codename)
     return set(lista_permissoes)
@@ -270,7 +270,7 @@ def permissoes_protocoloadm():
 def permissoes_adm():
     lista_permissoes = []
     perms_adm = Permission.objects.filter(
-        group__name='Operador de Administração')
+        group__name='Operador Administrativo')
     for p in perms_adm:
         lista_permissoes.append('protocoloadm.' + p.codename)
     return set(lista_permissoes)
@@ -279,7 +279,7 @@ def permissoes_adm():
 def permissoes_sessao():
     lista_permissoes = []
     perms_sessao = list(Permission.objects.filter(
-        group__name='Operador de Sessão'))
+        group__name='Operador de Sessão Plenária'))
     for p in perms_sessao:
         lista_permissoes.append('sessao.' + p.codename)
     return set(lista_permissoes)
@@ -288,7 +288,7 @@ def permissoes_sessao():
 def permissoes_painel():
     lista_permissoes = []
     perms_painel = list(Permission.objects.filter(
-        group__name='Operador de Painel'))
+        group__name='Operador de Painel Eletrônico'))
     for p in perms_painel:
         lista_permissoes.append('painel.' + p.codename)
     return set(lista_permissoes)
diff --git a/scripts/inicializa_grupos_autorizacoes.py b/scripts/inicializa_grupos_autorizacoes.py
index 42e6cb778..65f27f880 100644
--- a/scripts/inicializa_grupos_autorizacoes.py
+++ b/scripts/inicializa_grupos_autorizacoes.py
@@ -10,6 +10,14 @@ def cria_ou_reseta_grupo(nome):
     return grupo
 
 
+def cria_usuario(nome, grupo):
+    nome_usuario = nome
+    usuario = User.objects.get_or_create(username=nome_usuario)[0]
+    usuario.set_password('interlegis')
+    usuario.save()
+    grupo.user_set.add(usuario)
+
+
 def cria_grupos_permissoes():
 
     nomes_apps = ['base', 'parlamentares', 'comissoes',
@@ -31,10 +39,7 @@ def cria_grupos_permissoes():
         grupo.permissions.add(p)
 
     nome_usuario = 'operador_administrativo'
-    usuario = User.objects.get_or_create(username=nome_usuario)[0]
-    usuario.set_password('interlegis')
-    usuario.save()
-    grupo.user_set.add(usuario)
+    cria_usuario(nome_usuario, grupo)
 
     # prolocolo administrativo
     cts = cts.exclude(model__icontains='tramitacao').exclude(
@@ -47,10 +52,7 @@ def cria_grupos_permissoes():
         grupo.permissions.add(p)
 
     nome_usuario = 'operador_protocoloadm'
-    usuario = User.objects.get_or_create(username=nome_usuario)[0]
-    usuario.set_password('interlegis')
-    usuario.save()
-    grupo.user_set.add(usuario)
+    cria_usuario(nome_usuario, grupo)
 
     # permissoes do base
     cts = ContentType.objects.filter(app_label='base')
@@ -91,6 +93,9 @@ def cria_grupos_permissoes():
         for p in lista:
             grupo_geral.permissions.add(p)
 
+    nome_usuario = 'operador_geral'
+    cria_usuario(nome_usuario, grupo_geral)
+
     # Autor
     perms_autor = []
     perms_autor.append(Permission.objects.get(name='Can add Proposição'))
@@ -101,11 +106,9 @@ def cria_grupos_permissoes():
     grupo = cria_ou_reseta_grupo('Autor')
     for p in perms_autor:
         grupo.permissions.add(p)
+
     nome_usuario = 'operador_autor'
-    usuario = User.objects.get_or_create(username=nome_usuario)[0]
-    usuario.set_password('interlegis')
-    usuario.save()
-    grupo.user_set.add(usuario)
+    cria_usuario(nome_usuario, grupo_geral)
 
 if __name__ == '__main__':
     cria_grupos_permissoes()
diff --git a/scripts/test_inicializa_grupos_autorizacoes.py b/scripts/test_inicializa_grupos_autorizacoes.py
index 401854991..a8e9064a0 100644
--- a/scripts/test_inicializa_grupos_autorizacoes.py
+++ b/scripts/test_inicializa_grupos_autorizacoes.py
@@ -45,15 +45,3 @@ def test_permissoes_extras_sao_apagadas(app_label):
     cria_grupos_permissoes()
 
     assert not grupo.permissions.filter(id=permissao_errada.id).exists()
-
-
-# Operador de Comissões
-# Operador de Matéria
-# Operador de Norma Jurídica
-# Operador de Sessão Plenária
-
-# Operador de Protocolo Administrativo
-# Operador de Documento Administrativo
-
-
-# Operador Geral