pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Import CSP deps in uploader implementations

Browse Source
This commit is contained in:
richard-dds 2019-08-06 16:58:57 -04:00
parent 5a1c6b2b39
commit f1560d5d7f
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
atst/domain/csp/file_uploads.py
View File

@ -1,10 +1,6 @@
from azure.storage.common import CloudStorageAccount
from azure.storage.blob import BlobPermissions
from datetime import datetime, timedelta from datetime import datetime, timedelta
from uuid import uuid4 from uuid import uuid4
import boto3
def build_uploader(config): def build_uploader(config):
csp = config.get("CSP") csp = config.get("CSP")
@ -39,6 +35,11 @@ class AzureUploader(Uploader):
self.container_name = config["AZURE_TO_BUCKET_NAME"] self.container_name = config["AZURE_TO_BUCKET_NAME"]
self.timeout = timedelta(seconds=config["PERMANENT_SESSION_LIFETIME"]) self.timeout = timedelta(seconds=config["PERMANENT_SESSION_LIFETIME"])
from azure.storage.common import CloudStorageAccount
from azure.storage.blob import BlobPermissions
self.CloudStorageAccount = CloudStorageAccount
self.BlobPermissions = BlobPermissions
def get_token(self): def get_token(self):
""" """
Generates an Azure SAS token for pre-authorizing a file upload. Generates an Azure SAS token for pre-authorizing a file upload.
@ -47,7 +48,7 @@ class AzureUploader(Uploader):
- token_dict has a `token` key which contains the SAS token as a string - token_dict has a `token` key which contains the SAS token as a string
- object_name is a string - object_name is a string
""" """
account = CloudStorageAccount( account = self.CloudStorageAccount(
account_name=self.account_name, account_key=self.storage_key account_name=self.account_name, account_key=self.storage_key
) )
bbs = account.create_block_blob_service() bbs = account.create_block_blob_service()
@ -55,7 +56,7 @@ class AzureUploader(Uploader):
sas_token = bbs.generate_blob_shared_access_signature( sas_token = bbs.generate_blob_shared_access_signature(
self.container_name, self.container_name,
object_name, object_name,
permission=BlobPermissions.CREATE, permission=self.BlobPermissions.CREATE,
expiry=datetime.utcnow() + self.timeout, expiry=datetime.utcnow() + self.timeout,
protocol="https", protocol="https",
) )
@ -70,6 +71,9 @@ class AwsUploader(Uploader):
self.bucket_name = config["AWS_BUCKET_NAME"] self.bucket_name = config["AWS_BUCKET_NAME"]
self.timeout_secs = config["PERMANENT_SESSION_LIFETIME"] self.timeout_secs = config["PERMANENT_SESSION_LIFETIME"]
import boto3
self.boto3 = boto3
def get_token(self): def get_token(self):
""" """
Generates an AWS presigned post for pre-authorizing a file upload. Generates an AWS presigned post for pre-authorizing a file upload.
@ -79,7 +83,7 @@ class AwsUploader(Uploader):
form before being sent to S3 form before being sent to S3
- object_name is a string - object_name is a string
""" """
s3_client = boto3.client( s3_client = self.boto3.client(
"s3", "s3",
aws_access_key_id=self.access_key_id, aws_access_key_id=self.access_key_id,
aws_secret_access_key=self.secret_key, aws_secret_access_key=self.secret_key,