hcornet
506716e703
Some checks failed
Deployment Verification / deploy-and-test (push) Failing after 29s
146 lines
4.3 KiB
Python
146 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# IRIS Source Code
|
|
# Copyright (C) 2021 - Airbus CyberSecurity (SAS)
|
|
# ir@cyberactionlab.net
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 3 of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
from flask import Blueprint
|
|
# IMPORTS ------------------------------------------------
|
|
from flask import redirect
|
|
from flask import request
|
|
from flask_login import current_user
|
|
|
|
from app import app
|
|
from app import cache
|
|
from app import db
|
|
from app.datamgmt.context.context_db import ctx_get_user_cases
|
|
from app.datamgmt.context.context_db import ctx_search_user_cases
|
|
from app.models.authorization import Permissions
|
|
from app.models.cases import Cases
|
|
from app.models.models import Client
|
|
from app.util import ac_api_requires
|
|
from app.util import not_authenticated_redirection_url
|
|
from app.util import response_success
|
|
|
|
# CONTENT ------------------------------------------------
|
|
ctx_blueprint = Blueprint(
|
|
'context',
|
|
__name__,
|
|
template_folder='templates'
|
|
)
|
|
|
|
|
|
@ctx_blueprint.route('/context/set', methods=['POST'])
|
|
def set_ctx():
|
|
"""
|
|
Set the context elements of a user i.e the current case
|
|
:return: Page
|
|
"""
|
|
if not current_user.is_authenticated:
|
|
return redirect(not_authenticated_redirection_url())
|
|
|
|
ctx = request.form.get('ctx')
|
|
ctx_h = request.form.get('ctx_h')
|
|
|
|
current_user.ctx_case = ctx
|
|
current_user.ctx_human_case = ctx_h
|
|
|
|
db.session.commit()
|
|
|
|
update_user_case_ctx()
|
|
|
|
return response_success(msg="Saved")
|
|
|
|
|
|
@app.context_processor
|
|
def iris_version():
|
|
return dict(iris_version=app.config.get('IRIS_VERSION'),
|
|
organisation_name=app.config.get('ORGANISATION_NAME'),
|
|
std_permissions=Permissions,
|
|
demo_domain=app.config.get('DEMO_DOMAIN', None))
|
|
|
|
|
|
@app.context_processor
|
|
@cache.cached(timeout=3600, key_prefix='iris_has_updates')
|
|
def has_updates():
|
|
|
|
return dict(has_updates=False)
|
|
|
|
|
|
@ctx_blueprint.route('/context/get-cases/<int:max_results>', methods=['GET'])
|
|
@ac_api_requires(no_cid_required=True)
|
|
def cases_context(max_results, caseid):
|
|
# Get all investigations not closed
|
|
datao = ctx_get_user_cases(current_user.id, max_results=max_results)
|
|
|
|
return response_success(data=datao)
|
|
|
|
|
|
@ctx_blueprint.route('/context/search-cases', methods=['GET'])
|
|
@ac_api_requires(no_cid_required=True)
|
|
def cases_context_search(caseid):
|
|
search = request.args.get('q')
|
|
if not search:
|
|
return response_success(data=[])
|
|
|
|
# Get all investigations not closed
|
|
datao = ctx_search_user_cases(search, current_user.id, max_results=100)
|
|
|
|
return response_success(data=datao)
|
|
|
|
|
|
def update_user_case_ctx():
|
|
"""
|
|
Retrieve a list of cases for the case selector
|
|
:return:
|
|
"""
|
|
# Get all investigations not closed
|
|
res = Cases.query.with_entities(
|
|
Cases.name,
|
|
Client.name,
|
|
Cases.case_id,
|
|
Cases.close_date) \
|
|
.join(Cases.client) \
|
|
.order_by(Cases.open_date) \
|
|
.all()
|
|
|
|
data = [row for row in res]
|
|
|
|
if current_user and current_user.ctx_case:
|
|
# If the current user have a current case,
|
|
# Look for it in the fresh list. If not
|
|
# exists then remove from the user context
|
|
is_found = False
|
|
for row in data:
|
|
if row[2] == current_user.ctx_case:
|
|
is_found = True
|
|
break
|
|
|
|
if not is_found:
|
|
# The case does not exist,
|
|
# Removes it from the context
|
|
current_user.ctx_case = None
|
|
current_user.ctx_human_case = "Not set"
|
|
db.session.commit()
|
|
# current_user.save()
|
|
|
|
app.jinja_env.globals.update({
|
|
'cases_context_selector': data
|
|
})
|
|
|
|
return data
|