120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
from flask import Blueprint, request, jsonify
|
|
from flask_jwt_extended import jwt_required
|
|
from flasgger.utils import swag_from
|
|
from models.service import Service
|
|
from models.service_roles import ServiceRole
|
|
from models.ambiente import Ambiente
|
|
from models.user import db
|
|
|
|
ambiente_bp = Blueprint('ambiente', __name__, url_prefix='/ambiente')
|
|
|
|
@ambiente_bp.route('/', methods=['POST'])
|
|
@jwt_required()
|
|
@swag_from('../docs/ambiente/create.yml')
|
|
def create_ambiente():
|
|
data = request.get_json()
|
|
service_id = data.get('service_id')
|
|
service_role_id = data.get('service_role_id')
|
|
name = data.get('name')
|
|
description = data.get('description')
|
|
|
|
# Verifica se o serviço existe
|
|
service = Service.query.get(service_id)
|
|
if not service:
|
|
return jsonify({'error': 'Serviço não encontrado'}), 404
|
|
|
|
# Verifica se o ambiente já existe com o mesmo nome.
|
|
existing_role = ServiceRole.query.filter_by(service_id=service_id, name=name).first()
|
|
if existing_role:
|
|
return jsonify({'error': 'Função de serviço já existe para este serviço'}), 400
|
|
|
|
# Cria um novo ServiceRole
|
|
ambiente = Ambiente(
|
|
service_id=service_id,
|
|
service_role_id = service_role_id,
|
|
name=name,
|
|
description=description
|
|
)
|
|
|
|
# Adiciona e comita no banco de dados
|
|
db.session.add(ambiente)
|
|
db.session.commit()
|
|
|
|
return jsonify({'message': 'ambiente criado com sucesso', 'ambiente_id': ambiente.id}), 201
|
|
|
|
@ambiente_bp.route('/', methods=['GET'])
|
|
@jwt_required()
|
|
@swag_from('../docs/ambiente/list.yml')
|
|
def list_ambiente():
|
|
page = request.args.get('page', 1, type=int)
|
|
per_page = request.args.get('per_page', 10, type=int)
|
|
|
|
pagination = Ambiente.query.paginate(page=page, per_page=per_page, error_out=False)
|
|
ambientes = pagination.items
|
|
|
|
return jsonify({
|
|
'ambiente': [
|
|
{
|
|
'id': ambiente.id,
|
|
'name': ambiente.name,
|
|
'description': ambiente.description,
|
|
'service_id': ambiente.service_id,
|
|
'service_role_id': ambiente.service_role_id
|
|
} for ambiente in ambientes
|
|
],
|
|
'total': pagination.total,
|
|
'page': pagination.page,
|
|
'per_page': pagination.per_page
|
|
}), 200
|
|
|
|
@ambiente_bp.route('/<int:ambiente_id>', methods=['GET'])
|
|
@jwt_required()
|
|
@swag_from('../docs/ambiente/get.yml')
|
|
def get_ambiente(ambiente_id):
|
|
ambiente = Ambiente.query.get(ambiente_id)
|
|
if not ambiente:
|
|
return jsonify({'error': 'Camera não encontrada'}), 404
|
|
|
|
return jsonify({
|
|
'id': ambiente.id,
|
|
'name': ambiente.name,
|
|
'description': ambiente.description,
|
|
'service_id': ambiente.service_id,
|
|
'service_role_id': ambiente.service_role_id
|
|
}), 200
|
|
|
|
@ambiente_bp.route('/<int:ambiente_id>', methods=['PUT'])
|
|
@jwt_required()
|
|
@swag_from('../docs/ambiente/update.yml')
|
|
def update_camera(ambiente_id):
|
|
data = request.get_json()
|
|
name = data.get('name')
|
|
description = data.get('description')
|
|
|
|
ambiente = Ambiente.query.get(ambiente_id)
|
|
if not ambiente:
|
|
return jsonify({'error': 'Camera não encontrada'}), 404
|
|
|
|
# Atualiza os campos fornecidos
|
|
if name:
|
|
ambiente.name = name
|
|
if description:
|
|
ambiente.description = description
|
|
|
|
db.session.commit()
|
|
|
|
return jsonify({'message': 'Ambiente atualizada com sucesso', 'ambiente_id': ambiente.id}), 200
|
|
|
|
@ambiente_bp.route('/<int:ambiente_id>', methods=['DELETE'])
|
|
@jwt_required()
|
|
@swag_from('../docs/ambiente/delete.yml')
|
|
def delete_ambiente(ambiente_id):
|
|
ambiente = Ambiente.query.get(ambiente_id)
|
|
if not ambiente:
|
|
return jsonify({'error': 'Ambiente não encontrada'}), 404
|
|
|
|
db.session.delete(ambiente)
|
|
db.session.commit()
|
|
|
|
return jsonify({'message': 'Ambiente deletada com sucesso'}), 200
|