43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from decouple import config
|
|
|
|
from chalicelib.utils import s3
|
|
from chalicelib.utils.s3 import client
|
|
|
|
|
|
def __get_mob_keys(project_id, session_id):
|
|
params = {
|
|
"sessionId": session_id,
|
|
"projectId": project_id
|
|
}
|
|
return [
|
|
config("SESSION_MOB_PATTERN_S", default="%(sessionId)s") % params,
|
|
config("SESSION_MOB_PATTERN_E", default="%(sessionId)se") % params
|
|
]
|
|
|
|
|
|
def get_urls(project_id, session_id):
|
|
results = []
|
|
for k in __get_mob_keys(project_id=project_id, session_id=session_id):
|
|
results.append(client.generate_presigned_url(
|
|
'get_object',
|
|
Params={'Bucket': config("sessions_bucket"), 'Key': k},
|
|
ExpiresIn=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
|
|
))
|
|
return results
|
|
|
|
|
|
def get_ios(session_id):
|
|
return client.generate_presigned_url(
|
|
'get_object',
|
|
Params={
|
|
'Bucket': config("ios_bucket"),
|
|
'Key': str(session_id)
|
|
},
|
|
ExpiresIn=config("PRESIGNED_URL_EXPIRATION", cast=int, default=900)
|
|
)
|
|
|
|
|
|
def delete_mobs(project_id, session_ids):
|
|
for session_id in session_ids:
|
|
for k in __get_mob_keys(project_id=project_id, session_id=session_id):
|
|
s3.schedule_for_deletion(config("sessions_bucket"), k)
|