jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(chalice): fixed generate_presigned_url for assist-records

Browse source
This commit is contained in:
Taha Yassine Kraiem 2023-06-22 16:58:18 +01:00
parent 2e363da443
commit 7f849dfab6
1 changed files with 9 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
ee/api/chalicelib/core/assist_records.py
View file

@ -33,10 +33,9 @@ def save_record(project_id, data: schemas_ee.AssistRecordSavePayloadSchema, cont
params)
cur.execute(query)
result = helper.dict_to_camel_case(cur.fetchone())
result["URL"] = StorageClient.generate_presigned_url(
'get_object',
Params={'Bucket': config("ASSIST_RECORDS_BUCKET"), 'Key': result.pop("fileKey")},
ExpiresIn=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
result["URL"] = StorageClient.get_presigned_url_for_sharing(
bucket=config("ASSIST_RECORDS_BUCKET"), key=result.pop("fileKey"),
expires_in=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
)
return result
@ -104,10 +103,9 @@ def get_record(project_id, record_id, context: schemas_ee.CurrentContext):
cur.execute(query)
result = helper.dict_to_camel_case(cur.fetchone())
if result:
result["URL"] = StorageClient.generate_presigned_url(
'get_object',
Params={'Bucket': config("ASSIST_RECORDS_BUCKET"), 'Key': result.pop("fileKey")},
ExpiresIn=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
result["URL"] = StorageClient.get_presigned_url_for_sharing(
bucket=config("ASSIST_RECORDS_BUCKET"), key=result.pop("fileKey"),
expires_in=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
)
return result
@ -131,10 +129,9 @@ def update_record(project_id, record_id, data: schemas_ee.AssistRecordUpdatePayl
result = helper.dict_to_camel_case(cur.fetchone())
if not result:
return {"errors": ["record not found"]}
result["URL"] = StorageClient.generate_presigned_url(
'get_object',
Params={'Bucket': config("ASSIST_RECORDS_BUCKET"), 'Key': result.pop("fileKey")},
ExpiresIn=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
result["URL"] = StorageClient.get_presigned_url_for_sharing(
bucket=config("ASSIST_RECORDS_BUCKET"), key=result.pop("fileKey"),
expires_in=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
)
return result