openreplay/ee/api/chalicelib/utils/storage/extra.py
Amirouche ba8a875394 wip
2024-02-06 14:18:50 +01:00

16 lines
700 B
Python

from decouple import config
from chalicelib.utils.storage import StorageClient
def tag_session(file_key, tag_key='retention', tag_value='vault'):
bucket = config("sessions_bucket")
if not await StorageClient.exists(bucket=bucket, key=file_key):
return None
return await StorageClient.tag_file(file_key=file_key, bucket=bucket, tag_key=tag_key, tag_value=tag_value)
def tag_record(file_key, tag_key='retention', tag_value='vault'):
bucket = config('ASSIST_RECORDS_BUCKET')
if not await StorageClient.exists(bucket=bucket, key=file_key):
return None
return await StorageClient.tag_file(file_key=file_key, bucket=bucket, tag_key=tag_key, tag_value=tag_value)