Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

black

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-02-24 15:37:30 +01:00
parent 36931850c3
commit d32828ad91
5 changed files with 4 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyinfra/storage/adapters/azure.py
View File

@ -12,9 +12,7 @@ logging.getLogger("urllib3").setLevel(logging.WARNING)
def _retry(exceptions=Exception):
def inner(func):
@retry(exceptions=exceptions, delay=5, jitter=(0, 3), max_delay=60)
@wraps(func)
def inner(*args, **kwargs):
@ -26,7 +24,6 @@ def _retry(exceptions=Exception):
class AzureStorageAdapter(StorageAdapter):
def __init__(self, client):
super().__init__(client=client)
self.__client: BlobServiceClient = self._StorageAdapter__client

2
pyinfra/storage/adapters/s3.py
View File

@ -7,8 +7,8 @@ from pyinfra.storage.adapters.adapter import StorageAdapter
logger = logging.getLogger(__name__)
class S3StorageAdapter(StorageAdapter):
class S3StorageAdapter(StorageAdapter):
def __init__(self, client):
super().__init__(client=client)
self.__client: Minio = self._StorageAdapter__client

28
pyinfra/storage/storage.py
View File

@ -21,34 +21,6 @@ class Storage:
return self.__adapter.clear_bucket(bucket_name)
# import abc
# import gzip
# import logging

2
pyinfra/storage/storages.py
View File

@ -11,4 +11,4 @@ def get_azure_storage(config=None):
def get_s3_storage(config=None):
return Storage(S3StorageAdapter(get_s3_client(config)))
return Storage(S3StorageAdapter(get_s3_client(config)))

4
pyinfra/test/unit_tests/storage_test.py
View File

@ -58,8 +58,8 @@ def storage(client_name, bucket_name, request):
def test_get_azure_storage_yields_storage():
assert isinstance(get_azure_storage(), Storage)
assert isinstance(get_azure_storage(), Storage)
def test_get_s3_storage_yields_storage():
assert isinstance(get_s3_storage(), Storage)
assert isinstance(get_s3_storage(), Storage)