Merge branch 'master' of https://github.com/lightbase/WSCacicNeo

Macieski
2 parents 26e10d0f faab0ce8
Showing 4 changed files with 33 additions and 21 deletions Show diff stats
wscacicneo/doc/classes_wscacicneo.png
wscacicneo/doc/packages_wscacicneo.png
wscacicneo/utils/utils.py
wscacicneo/views.py
@@ -8,3 +8,9 @@ class Utils:
     def to_url(*args):
         return '/'.join(list(args))
+
+    def verifica_email_institucional(email):
+    	if("gov.br" in email):
+    		return True
+    	else:
+    		return False
 \ No newline at end of file
@@ -9,6 +9,7 @@ from .models import (
     DBSession,
     SistemaOperacional,
     )
+from wscacicneo.utils.utils import Utils
 from wscacicneo.model.orgao import Orgao
 from wscacicneo.model.orgao import OrgaoBase
 from wscacicneo.model.user import User
@@ -328,27 +329,32 @@ def post_user(request):
     rest_url = REST_URL
     userbase = UserBase().lbbase
     doc = request.params
-    document = doc['favoritos']
-    favoritos = [document]
-    itens = [doc['lista_orgao'], doc['cadastro_orgao'], doc['lista_user'], doc['cadastro_user'], doc['relatorios'], doc['coleta'], doc['notify']]
-    user_obj = User(
-        nome = doc['nome'],
-        matricula = doc['matricula'],
-        email = doc['email'],
-        orgao = doc['orgao'],
-        telefone = doc['telefone'],
-        cargo = doc['cargo'],
-        setor = doc['setor'],
-        permissao = doc['permissao'],
-        senha = doc['senha'],
-        favoritos = favoritos,
-        itens = itens
-    )
-    print(user_obj)
-    id_doc = user_obj.create_user()
-    print(id_doc)
-
-    return Response(str(id_doc))
+    email_user = doc['email']
+    email_is_institucional = Utils.verifica_email_institucional(email_user)
+    if(email_is_institucional):
+        document = doc['favoritos']
+        favoritos = [document]
+        itens = [doc['lista_orgao'], doc['cadastro_orgao'], doc['lista_user'], doc['cadastro_user'], doc['relatorios'], doc['coleta'], doc['notify']]
+        user_obj = User(
+            nome = doc['nome'],
+            matricula = doc['matricula'],
+            email = doc['email'],
+            orgao = doc['orgao'],
+            telefone = doc['telefone'],
+            cargo = doc['cargo'],
+            setor = doc['setor'],
+            permissao = doc['permissao'],
+            senha = doc['senha'],
+            favoritos = favoritos,
+            itens = itens
+        )
+        print(user_obj)
+        id_doc = user_obj.create_user()
+        print(id_doc)
+
+        return Response(str(id_doc))
+    else:
+        return {"yololo":"yololo"}
 @view_config(route_name='edituser', renderer='templates/editaruser.pt')
 def edituser(request):
	@@ -8,3 +8,9 @@ class Utils:		@@ -8,3 +8,9 @@ class Utils:
8		8
9	def to_url(*args):	9	def to_url(*args):
10	return '/'.join(list(args))	10	return '/'.join(list(args))
		11	+
		12	+ def verifica_email_institucional(email):
		13	+ if("gov.br" in email):
		14	+ return True
		15	+ else:
		16	+ return False
11	\ No newline at end of file	17	\ No newline at end of file