def set_avatar(name, image):
if not image:
return False
entry = Avatar.objects(username=name).first()
try:
if entry:
entry.image.replace(image)
entry.last_modified = datetime.utcnow().replace(microsecond=0)
entry.save()
else:
entry = Avatar(username=name)
entry.image.put(image)
entry.save()
except mongoengine.ValidationError:
return False
return True
def set_avatar(name, image):
if not image:
return False
entry = Avatar.objects(username=name).first()
try:
if entry:
entry.image.replace(image)
entry.last_modified = datetime.utcnow().replace(microsecond=0)
entry.save()
else:
entry = Avatar(username=name)
entry.image.put(image)
entry.save()
except mongoengine.ValidationError:
return False
return True
def get_avatar(name):
avatar = Avatar.objects(username=re.compile(name, re.IGNORECASE)).first()
if avatar is None or avatar.image is None:
return send_file(open('blueprints/avatar/char_face.png', 'r'), mimetype='image/png',
cache_timeout=3600 * 24 * 30, add_etags=False)
else:
if not avatar.last_modified or not request.if_modified_since or avatar.last_modified > request.if_modified_since:
image = StringIO.StringIO(avatar.image.read())
image.seek(0)
response = send_file(image, mimetype='image/png')
response.last_modified = avatar.last_modified
response.expires = None
return response
else:
return flask.Response(status=304)
def get_avatar(name):
avatar = Avatar.objects(username=re.compile(name, re.IGNORECASE)).first()
if avatar is None or avatar.image is None:
return send_file(open('blueprints/avatar/char_face.png', 'r'),
mimetype='image/png',
cache_timeout=3600 * 24 * 30,
add_etags=False)
else:
if not avatar.last_modified or not request.if_modified_since or avatar.last_modified > request.if_modified_since:
image = StringIO.StringIO(avatar.image.read())
image.seek(0)
response = send_file(image, mimetype='image/png')
response.last_modified = avatar.last_modified
response.expires = None
return response
else:
return flask.Response(status=304)
def create_user():
name = request.json.get('name')
email = request.json.get('email')
password = request.json.get('password')
confirm_password = request.json.get('confirm_password')
account_type = request.json.get('account_type')
mentor_id = request.json.get('mentor_id')
phone_number = request.json.get('phone_number')
portfolio = request.json.get('portfolio')
status = check_client(account_type, "pending")
user_id = request.json.get('id')
mentor_check = request.json.get('is_mentor')
rg = request.json.get('rg')
cpf = request.json.get('cpf')
services = request.json.get('services')
if not name:
return jsonify({"data": {"msg": "Nome é obrigatório"}}), 400
elif not email:
return jsonify({"data": {"msg": "E-mail obrigatório"}}), 400
elif not password:
return jsonify({"data": {"msg": "Senha é obrigatório"}}), 400
elif not confirm_password:
return jsonify(
{"data": {
"msg": "Confirmação de senha é obrigatório"
}}), 400
elif not account_type:
return jsonify({"data": {
"msg": "Tipo de conta é obrigatório"
}}), 400
elif not phone_number:
return jsonify({"data": {"msg": "Telefone é obrigatório"}}), 400
elif not mentor_id and account_type == "pro" and status == "pending":
return jsonify({"data": {"msg": "Selecione um mentor"}}), 400
elif account_type == "pro" and services == "[]":
return jsonify(
{"data": {
"msg": "Lista de serviços é obrigatório"
}}), 400
#validation
#name
email_pattern = r"(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)"
url_pattern = r'(http:\/\/www\.|https:\/\/www\.|http:\/\/|https:\/\/)?[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,5}(:[0-9]{1,5})?(\/.*)?$'
if len(name) < 7:
return jsonify(
{"data": {
"msg": "Seu nome deve ter mais de 6 letras"
}}), 400
elif not re.match(email_pattern, email):
return jsonify({"data": {"msg": "E-mail inválido"}}), 400
elif phone_number and len(phone_number) < 11:
return jsonify({"data": {
"msg": "Número de telefone inválido"
}}), 400
elif len(password) < 8:
return jsonify(
{"data": {
"msg": "Sua senha deve ter mais de 8 dígitos"
}}), 400
elif not re.search('[a-zA-Z]', password):
return jsonify({
"data": {
"msg": "Sua senha deve conter pelo menos uma letra"
}
}), 400
elif not re.search('[0-9]', password):
return jsonify({
"data": {
"msg": "Sua senha deve conter pelo menos um número"
}
}), 400
elif password != confirm_password:
return jsonify({"data": {"msg": "Senhas não coincidem"}}), 400
elif portfolio and not re.match(url_pattern, portfolio):
return jsonify({"data": {
"msg": "Link do portfólio inválido"
}}), 400
elif cpf and len(cpf) != 11:
return jsonify({"data": {"msg": "CPF inválido"}}), 400
elif rg and len(rg) != 9:
return jsonify({"data": {"msg": "RG inválido"}}), 400
hashed_pass = bcrypt.hashpw(password.encode('utf-8'),
bcrypt.gensalt(10))
mentor_data = mentor_id
if user_id:
admin = db.session.query(User).filter_by(id=user_id).first()
mentor_data = admin.id
if admin.account_type == "admin" or admin.account_type == "mentor" and mentor_check == True:
account_type = "mentor"
status = "approved"
elif admin.account_type == "admin" or admin.account_type == "mentor" and mentor_check == False:
account_type = "pro"
status = "approved"
new_user = User(name, email, hashed_pass, account_type, mentor_data,
phone_number, check_client(account_type, portfolio),
check_client(account_type, "Amador"), 5,
check_client(account_type, 1),
check_client(account_type, 1),
check_client(account_type, 1),
check_client(account_type, 1),
check_client(account_type, 1),
check_client(account_type, 1),
check_client(account_type, 1), None, None,
request.json.get('rg'), request.json.get('cpf'),
time.time(), time.time(), None, status,
check_client(account_type, services))
try:
db.session.add(new_user)
db.session.commit()
except:
db.session.close()
return jsonify(
{"data": {
"msg": "Este E-mail já esta sendo utilizado"
}}), 400
json = UserSchema(strict=True).dump(new_user).data
images = Avatar(json['id'], None, None)
db.session.add(images)
db.session.commit()
data = {
"id": json['id'],
"name": json['name'],
"email": json['email'],
"account_type": json['account_type'],
"phone_number": json['phone_number'],
"mentor_id": json['mentor_id'],
"portfolio": json['portfolio'],
"phone_number": json['phone_number'],
"avatar_name": json['avatar_id'],
"banner_name": json['banner_id'],
"rg": json['rg'],
"cpf": json['cpf'],
"created_at": json['created_at'],
"updated_at": json['updated_at'],
"about_me": json['about_me'],
"status": json['status'],
"services": json['services'],
"ratings": {
"general":
json['rating_general'],
"pro":
json['rating_pro'],
"events": [
{
'rating': json['evt_rating_birthday'],
'label': "Aniversário"
},
{
'rating': json['evt_rating_civil'],
'label': "Civil"
},
{
'rating': json['evt_rating_baptism'],
'label': "Batizado"
},
{
'rating': json['evt_rating_essay'],
'label': "Ensaio"
},
{
'rating': json['evt_rating_wedding'],
'label': "Casamento"
},
{
'rating': json['evt_rating_corp'],
'label': "Corporativo"
},
{
'rating': json['evt_rating_debut'],
'label': "Debutante"
},
]
}
}
db.session.close()
return jsonify({"data": data})