AzureBaseResourceManagerAction

Complexity

Total Complexity

1

Size/Duplication

Total Lines	9
Duplicated Lines	0 %

1 Method

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	8	1

from st2actions.runners.pythonrunner import Action

from libcloud.compute.providers import Provider as ComputeProvider
from libcloud.compute.providers import get_driver as get_compute_driver
from libcloud.storage.providers import Provider as StorageProvider
from libcloud.storage.providers import get_driver as get_storage_driver

from msrestazure.azure_active_directory import ServicePrincipalCredentials

__all__ = [
    'AzureBaseComputeAction',
    'AzureBaseStorageAction',
    'AzureBaseResourceManagerAction'
]


class AzureBaseComputeAction(Action):
    def __init__(self, config):
        super(AzureBaseComputeAction, self).__init__(config=config)

        config = self.config['compute']
        subscription_id = config['subscription_id']
        key_file = config['cert_file']
        self._driver = self._get_driver(subscription_id=subscription_id,
                                        key_file=key_file)

    def _get_driver(self, subscription_id, key_file):
        cls = get_compute_driver(ComputeProvider.AZURE)
        driver = cls(subscription_id=subscription_id, key_file=key_file)
        return driver


class AzureBaseStorageAction(Action):
    def __init__(self, config):
        super(AzureBaseStorageAction, self).__init__(config=config)

        config = self.config['storage']
        name = config['name']
        access_key = config['access_key']
        self._driver = self._get_driver(name=name, access_key=access_key)

    def _get_driver(self, name, access_key):
        cls = get_storage_driver(StorageProvider.AZURE_BLOBS)
        driver = cls(key=name, secret=access_key)
        return driver


class AzureBaseResourceManagerAction(Action):
    def __init__(self, config):
        super(AzureBaseResourceManagerAction, self).__init__(config=config)

        resource_config = self.config['resource_manager']
        self.credentials = ServicePrincipalCredentials(
            client_id=resource_config['client_id'],
            secret=resource_config['secret'],
            tenant=resource_config['tenant']
        )