def newr():
if request.method == 'GET':
return render_template('newr.j2')
new_role = Roles(**request.form)
try:
new_role.save()
except peewee.IntegrityError:
return abort(403, 'Role exists')
return render_template('newr.j2')
def home():
if not google.authorized:
return redirect(url_for("google.login"))
resp = google.get("/oauth2/v2/userinfo")
emailUser = resp.json()["email"]
if emailUser == "*****@*****.**":
user = User.query.filter_by(email=emailUser).first()
if user is None:
user = User(email=emailUser)
db.session.add(user)
db.session.flush()
role = Roles(user_id=user.id, role="Admin")
db.session.add(role)
db.session.commit()
login_user(user, remember=True)
return redirect(url_for("admin"))
else:
login_user(user)
return redirect(url_for("admin"))
else:
user = User.query.filter_by(email=emailUser).first()
if user is None:
return "You are not allowed to do staging"
else:
login_user(user, remember=True)
return redirect(url_for("instance"))
def addUser():
response = {}
if current_user.is_authenticated:
email = request.form["email"]
if email.strip():
userEmail = email+"@ebizontek.com"
u = User.query.filter_by(email=userEmail).first()
if u is None:
user = User(email=userEmail)
db.session.add(user)
db.session.flush()
role = Roles(user_id=user.id, role="Lead")
db.session.add(role)
db.session.commit()
response["success"] = True
response["message"] = "You have added a user by email "+userEmail
return jsonify(**response)
else:
response["success"] = False
response["message"] = userEmail+" Already Present"
return jsonify(**response)
else:
response["success"] = False
response["message"] = "You are not allowed to do that"
return jsonify(**response)
def loadroles():
role = Roles()
role.rolename = "admin"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "customer"
db.session.add(role)
db.session.commit()
print("Roles creados")
def createroles(id=None):
if request.method == 'GET':
if id is not None:
roles = Roles.query.get(id)
if roles:
return jsonify(roles.serialize()), 200
else:
return jsonify({"msg": "Rol no encontrado :("}), 404
else:
roles = Roles.query.all()
roles = list(map(lambda role: role.serialize(), roles))
return jsonify(users), 200
if request.method == 'POST':
rolename = request.json.get('rolename', None)
roles = Roles()
roles.rolename = rolename
db.session.add(roles)
db.session.commit()
return jsonify(roles.serialize()), 201
if request.method == 'PUT':
rolename = request.json.get('rolename', None)
if not rolename or rolename == "":
return jsonify({"msg": "Ingresa el nombre del rol!"}), 400
roles = Roles.query.get(id)
if not roles:
return jsonify({"msg": "No encontrado"}), 404
roles.rolename = rolename
db.session.commit()
return jsonify(roles.serialize()), 201
if request.method == 'DELETE':
roles = Roles.query.get(id)
if not roles:
return jsonify({"msg": "Rol no encontrado"}), 404
db.session.delete(roles)
db.session.commit()
return jsonify({"msg": "Rol borrado!"}), 200
def load_globales():
role = Roles()
role.name = "admin"
role.status = True
role.save()
role = Roles()
role.name = "dj"
role.status = True
role.save()
role = Roles()
role.name = "client"
role.status = True
role.save()
og = ObjetosGlobales()
og.save()
print("Objetos globales cargados")
def loadroles():
role = Roles()
role.rolename = "admin"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "finanzas"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "comercial"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "staff general"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "alimentacion"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "seguridad"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "protagonista"
db.session.add(role)
db.session.commit()
role = Roles()
role.rolename = "cliente"
db.session.add(role)
db.session.commit()
print("Roles creados")