def registroEmpresa():
if "Tipo" in session:
return redireccionarUsuario()
if request.method == "GET":
return render_template('registroEmpresa.html')
elif request.method == "POST":
objEnc = Encriptador()
objUsuario = Usuario()
objEmp = Empresa()
nombreEmpresa = str(request.form['nombreEmp'])
direccion = str(request.form['direccion'])
telefono = str("+506" + request.form['telefono'])
nombreUsuario = str(request.form['usuario'])
cedula = str(request.form['cedula'])
email = str(request.form['email'])
contra = objEnc.Encriptar(str(request.form['contra']))
objEmp.setEmpresa(nombreEmpresa, direccion, telefono, 2)
if objEmp.agregarEmpresa():
idUltimo = int(objEmp.getUltimoID())
objUsuario.setUsuario(nombreUsuario, cedula, email, contra, 5,
idUltimo, 0)
if objUsuario.AgregarUsuario():
usuario = objUsuario.validarUsuario(2)
session["last_active"] = dt.datetime.now()
session["ID"] = usuario[0][0]
session["Nombre"] = usuario[0][1]
session["Cedula"] = usuario[0][2]
session["Email"] = usuario[0][3]
session["Contra"] = usuario[0][4]
session["Avatar"] = usuario[0][5]
session["Activo"] = usuario[0][6]
session["Empresa"] = usuario[0][8]
session["Tipo"] = usuario[0][9]
session["Departamento"] = usuario[0][10]
return "Exito"
return "Error"
def agregarUsuario():
if request.method == 'POST' and "ID" in session:
objUsuario = Usuario()
objEnc = Encriptador()
boolImg = True
imagen = ""
nombre = request.form['nombre']
cedula = request.form['cedula']
email = request.form['email']
contra = objEnc.Encriptar(str(request.form['contra']))
if session['Tipo'] == 1:
tipo = int(request.form['tipo'])
tipoEmp = int(request.form['tipoEmp'])
empresa = request.form['empresa']
dep = int(request.form['departamento']) if int(request.form['departamento'] != 0) else 0
elif session['Tipo'] == 5:
empresa = int(session['Empresa'])
dep = 0
tipo = 6
elif session['Tipo'] == 2:
tipo = int(request.form['tipo'])
empresa = int(session['Empresa'])
dep = int(session['Departamento'])
if int(session['Tipo']) == 2:
if tipo not in[3,4]:
return "Error"
elif int(session['Tipo']) == 5:
if tipo != 6:
return "Error"
if tipo in [2,3,4] and dep == 0:
return "Error"
if request.files:
imagen = request.files["avatar"]
if not imagenPermitida(imagen.filename):
boolImg = False
else:
ext = imagen.filename.rsplit(".",1)[1]
nombreArchivo = str(uuid.uuid4().hex+"."+ext)
objUsuario.setAvatar(str(nombreArchivo))
if boolImg:
if session['Tipo'] == 1:
if tipoEmp == 2 and (tipo != 5 and tipo != 6):
return "Error"
elif tipoEmp == 1 and (tipo == 5 or tipo == 6):
return "Error"
objUsuario.setUsuario(nombre,cedula,email,contra,tipo,empresa,dep)
if tipo == 2 and objUsuario.existeTipoUsuario():
return "Gerente Existe"
elif tipo == 5 and objUsuario.existeTipoUsuario():
return "ClienteM Existe"
if objUsuario.AgregarUsuario():
if imagen != "":
imagen.save(os.path.join(app.config['IMAGE_UPLOADS'],nombreArchivo))
return "Exito"
else:
return "Error"
else:
return "Error"
else:
return redirect(url_for('inicio.index'))