Skip to content

Commit a9a91c2

Browse files
committed
Moved up iris_current_user
1 parent 0f05312 commit a9a91c2

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

source/app/blueprints/rest/datastore_routes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def datastore_update_file(cur_id: int, caseid: int):
141141
msg_added_as += 'and added in IOC'
142142

143143
if dsf.file_is_evidence:
144-
datastore_add_file_as_evidence(dsf, caseid)
144+
datastore_add_file_as_evidence(iris_current_user.id, dsf, caseid)
145145
msg_added_as += ' and evidence' if len(msg_added_as) > 0 else 'and added in evidence'
146146

147147
track_activity(f'File \"{dsf.file_original_name}\" updated in DS', caseid=caseid)
@@ -266,7 +266,7 @@ def datastore_add_file(cur_id: int, caseid: int):
266266
msg_added_as += 'and added in IOC'
267267

268268
if dsf_sc.file_is_evidence:
269-
datastore_add_file_as_evidence(dsf_sc, caseid)
269+
datastore_add_file_as_evidence(iris_current_user.id, dsf_sc, caseid)
270270
msg_added_as += ' and evidence' if len(msg_added_as) > 0 else 'and added in evidence'
271271

272272
track_activity(f"File \"{dsf_sc.file_original_name}\" added to DS", caseid=caseid)

source/app/datamgmt/datastore/datastore_db.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ def datastore_add_file_as_ioc(dsf):
366366
db_create(ioc)
367367

368368

369-
def datastore_add_file_as_evidence(dsf, caseid):
369+
def datastore_add_file_as_evidence(user_identifier, dsf, caseid):
370370
crf = CaseReceivedFile.query.filter(
371371
CaseReceivedFile.file_hash == dsf.file_sha256
372372
).first()
@@ -379,7 +379,7 @@ def datastore_add_file_as_evidence(dsf, caseid):
379379
crf.date_added = datetime.datetime.now()
380380
crf.filename = dsf.file_original_name
381381
crf.file_size = dsf.file_size
382-
crf.user_id = iris_current_user.id
382+
crf.user_id = user_identifier
383383

384384
db_create(crf)
385385

0 commit comments

Comments
 (0)