def ocorrencia_del():
"""Função para exclusão de ocorrência na imagem
Faz update no fs.files, excluindo do array a id_ocorrencia
Args:
_id: ObjectId do arquivo
id_ocorrencia: String (texto)
Returns:
image_ocorrencia, passando
data['success']: True ou False
"""
_id = request.values.get('_id')
id_ocorrencia = request.values.get('id_ocorrencia')
try:
db = app.config['mongodb']
ocorrencias = Ocorrencias(db)
success = ocorrencias.delete(_id=ObjectId(_id),
id_ocorrencia=id_ocorrencia)
except Exception as err:
logger.error(err, exc_info=True)
return jsonify({'erro': str(err)})
return image_ocorrencia(_id, success)
def ocorrencia_add():
"""Função para inserção de ocorrência na imagem
Faz update no fs.files, inserindo em um array o nome do usuário ativo
e o texto da ocorrência passada.
Args:
_id: ObjectId do arquivo
texto: String (texto)
Returns:
json['success']: True ou False
"""
_id = request.values.get('_id')
texto = request.values.get('texto')
try:
db = app.config['mongodb']
ocorrencias = Ocorrencias(db)
success = ocorrencias.add(_id=ObjectId(_id),
usuario=current_user.id,
texto=texto)
except Exception as err:
logger.error(err, exc_info=True)
return jsonify({'erro': str(err)})
return image_ocorrencia(_id, success)
def test3_delete_ocorrencias(self):
ocorrencias = Ocorrencias(mongodb)
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 1')
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 2')
ocorrencias.add(self.inserted_id, 'ajna', 'Ocorrencia 3')
lista_ocorrencias = ocorrencias.list_usuario(self.inserted_id, 'ajna')
id_ocorrencia = lista_ocorrencias[0]['id_ocorrencia']
print(id_ocorrencia)
sucesso = ocorrencias.delete(self.inserted_id, id_ocorrencia)
assert sucesso is True
lista_ocorrencias = ocorrencias.list_usuario(self.inserted_id, 'ajna')
assert lista_ocorrencias == []
def file(_id=None):
"""Tela para exibição de um 'arquivo' do GridFS.
Exibe o arquivo e os metadados associados a ele.
"""
db = app.config['mongodb']
fs = GridFS(db)
tags_object = Tags(db)
form_tags = TagsForm()
form_tags.tags.choices = tags_object.tags_text
if request.args.get('filename'):
filename = mongo_sanitizar(request.args.get('filename'))
logger.warn('Filename %s ' % filename)
grid_data = fs.find_one({'filename': filename})
else:
if not _id:
_id = request.args.get('_id')
grid_data = fs.get(ObjectId(_id))
# print(grid_data)
if grid_data:
summary_ = dict_to_html(summary(grid_data=grid_data))
summary_carga = dict_to_html(carga.summary(grid_data=grid_data))
tags = tags_object.list(_id)
ocorrencias = Ocorrencias(db).list(_id)
else:
summary_ = summary_carga = 'Arquivo não encontrado.'
ocorrencias = []
return render_template('view_file.html',
myfile=grid_data,
summary=summary_,
summary_carga=summary_carga,
form_tags=form_tags,
tags=tags,
ocorrencias=ocorrencias)
def test2_list_ocorrencias_usuario(self):
ocorrencias = Ocorrencias(mongodb)
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 1')
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 2')
ocorrencias.add(self.inserted_id, 'ajna', 'Ocorrencia 3')
lista_ocorrencias = ocorrencias.list_usuario(self.inserted_id, 'ivan')
assert len(lista_ocorrencias) == 2
lista_ocorrencias = ocorrencias.list_usuario(self.inserted_id, 'ajna')
assert len(lista_ocorrencias) == 1
assert lista_ocorrencias[0]['texto'] == 'Ocorrencia 3'
assert lista_ocorrencias[0]['usuario'] == 'ajna'
def test1_add_ocorrencia(self):
ocorrencias = Ocorrencias(mongodb)
assert self.inserted_id is not None
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 1')
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 2')
ocorrencias.add(self.inserted_id, 'ajna', 'Ocorrencia 3')
lista_ocorrencias = ocorrencias.list(self.inserted_id)
assert type(lista_ocorrencias) is list
assert len(lista_ocorrencias) == 3
def image_ocorrencia(_id, success=True):
"""Função para listar ocorrências na imagem
Args:
_id: ObjectId do arquivo
success: Falso se houve erro em operação anterior
Returns:
json['success']: True ou False
"""
data = {'success': success}
try:
db = app.config['mongodb']
ocorrencias = Ocorrencias(db)
data['ocorrencias'] = ocorrencias.list(ObjectId(_id))
except Exception as err:
logger.error(err, exc_info=True)
data['error'] = str(err)
data['success'] = False
# raise
return jsonify(data)