def saveCsvLogToBlob(self, fileName):
to_location_path = fileName
account_name = s().Data["blob"]["account_name"]
account_key = s().Data["blob"]["account_key"]
container_name = s().Data["blob"]["container"]
cloud_account = CloudStorageAccount(account_name=account_name,
account_key=account_key)
append_blob_service = cloud_account.create_append_blob_service()
append_blob_service.create_container(container_name)
append_blob_service.set_container_acl(
container_name, public_access=PublicAccess.Container)
if append_blob_service.exists(container_name, self.fileName):
append_blob_service.append_blob_from_path(
container_name,
self.fileName,
self.fileName,
progress_callback=self.progress_callback_w)
else:
cloud_account.create_block_blob_service().create_blob_from_path(
container_name,
self.fileName,
to_location_path,
progress_callback=self.progress_callback_w)
class BlobLogger():
def __init__(self,
account_name,
account_key,
container_name,
blob_name=None):
self.account_name = account_name
self.account_key = account_key
self.container_name = container_name
self.storage_client = CloudStorageAccount(account_name, account_key)
self.blob_service = self.storage_client.create_append_blob_service()
if not blob_name:
self.blob_name = socket.gethostname(
) + ".txt" # + "-" + datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + ".txt"
else:
self.blob_name = blob_name
self.blob_service.create_container(container_name)
if not self.blob_service.exists(self.container_name, self.blob_name):
self.blob_service.create_blob(self.container_name, self.blob_name)
def write(self, text):
now = datetime.datetime.now().strftime("[%Y-%m-%d %H:%M:%S] ")
self.blob_service.append_block(self.container_name, self.blob_name,
str.encode(now + text + "\n"))
print(now + text)
class StorageAccountTest(StorageTestCase):
def setUp(self):
super(StorageAccountTest, self).setUp()
self.account_name = self.settings.STORAGE_ACCOUNT_NAME
self.account_key = self.settings.STORAGE_ACCOUNT_KEY
self.sas_token = '?sv=2015-04-05&st=2015-04-29T22%3A18%3A26Z&se=2015-04-30T02%3A23%3A26Z&sr=b&sp=rw&sip=168.1.5.60-168.1.5.70&spr=https&sig=Z%2FRHIX5Xcg0Mq2rqI3OlWTjEg2tYkboXr1P9ZUXDtkk%3D'
self.account = CloudStorageAccount(self.account_name, self.account_key)
#--Helpers-----------------------------------------------------------------
def validate_service(self, service, type):
self.assertIsNotNone(service)
self.assertIsInstance(service, type)
self.assertEqual(service.account_name, self.account_name)
self.assertEqual(service.account_key, self.account_key)
#--Test cases --------------------------------------------------------
def test_create_block_blob_service(self):
# Arrange
# Act
service = self.account.create_block_blob_service()
# Assert
self.validate_service(service, BlockBlobService)
def test_create_page_blob_service(self):
# Arrange
# Act
service = self.account.create_page_blob_service()
# Assert
self.validate_service(service, PageBlobService)
def test_create_append_blob_service(self):
# Arrange
# Act
service = self.account.create_append_blob_service()
# Assert
self.validate_service(service, AppendBlobService)
def test_create_table_service(self):
# Arrange
# Act
service = self.account.create_table_service()
# Assert
self.validate_service(service, TableService)
def test_create_queue_service(self):
# Arrange
# Act
service = self.account.create_queue_service()
# Assert
self.validate_service(service, QueueService)
def test_create_file_service(self):
# Arrange
# Act
service = self.account.create_file_service()
# Assert
self.validate_service(service, FileService)
def test_create_service_no_key(self):
# Arrange
# Act
bad_account = CloudStorageAccount('', '')
with self.assertRaises(ValueError):
service = bad_account.create_block_blob_service()
# Assert
def test_create_account_sas(self):
# Arrange
# Act
sas_account = CloudStorageAccount(self.account_name, sas_token=self.sas_token)
service = sas_account.create_block_blob_service()
# Assert
self.assertIsNotNone(service)
self.assertEqual(service.account_name, self.account_name)
self.assertIsNone(service.account_key)
self.assertEqual(service.sas_token, self.sas_token)
def test_create_account_sas_and_key(self):
# Arrange
# Act
account = CloudStorageAccount(self.account_name, self.account_key, self.sas_token)
service = account.create_block_blob_service()
# Assert
self.validate_service(service, BlockBlobService)
def test_create_account_emulated(self):
# Arrange
# Act
account = CloudStorageAccount(is_emulated=True)
service = account.create_block_blob_service()
# Assert
self.assertIsNotNone(service)
self.assertEqual(service.account_name, 'devstoreaccount1')
self.assertIsNotNone(service.account_key)
@record
def test_generate_account_sas(self):
# SAS URL is calculated from storage key, so this test runs live only
if TestMode.need_recordingfile(self.test_mode):
return
# Arrange
token = self.account.generate_shared_access_signature(
Services.BLOB,
ResourceTypes.OBJECT,
AccountPermissions.READ,
datetime.utcnow() + timedelta(hours=1),
)
service = self.account.create_block_blob_service()
data = b'shared access signature with read permission on blob'
container_name='container1'
blob_name = 'blob1.txt'
try:
service.create_container(container_name)
service.create_blob_from_bytes(container_name, blob_name, data)
# Act
url = service.make_blob_url(
container_name,
blob_name,
sas_token=token,
)
response = requests.get(url)
# Assert
self.assertTrue(response.ok)
self.assertEqual(data, response.content)
finally:
service.delete_container(container_name)
