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 agregarEmpresa():
if request.method == 'POST' and "ID" in session and session['Tipo'] == 1:
objEmp = Empresa()
objEmpDep = EmpresaDepartamento()
nombre = request.form['nom']
direccion = request.form['dir']
telefono = str("+506"+request.form['tel'])
tipo = int(request.form['tipo'])
departamentos = []
if tipo == 1:
departamentos = request.form.getlist('deps[]')
try:
objEmp.setEmpresa(nombre,direccion,telefono,tipo)
if objEmp.agregarEmpresa():
ID = int(objEmp.getUltimoID())
objEmpDep.setIDEmpresa(ID)
for dep in departamentos:
objEmpDep.setIDDepartamento(int(dep))
objEmpDep.agregarDepartamentosEmp()
return jsonify({'result':'success'})
except:
return jsonify({'result':'error'})