datacrunch 1.15.0__py3-none-any.whl → 1.17.1__py3-none-any.whl

datacrunch/authentication/authentication.py

@@ -1,112 +0,0 @@
-import requests
-import time
-
-from datacrunch.http_client.http_client import handle_error
-
-TOKEN_ENDPOINT = '/oauth2/token'
-
-CLIENT_CREDENTIALS = 'client_credentials'
-REFRESH_TOKEN = 'refresh_token'
-
-
-class AuthenticationService:
-    """A service for client authentication"""
-
-    def __init__(self, client_id: str, client_secret: str, base_url: str) -> None:
-        self._base_url = base_url
-        self._client_id = client_id
-        self._client_secret = client_secret
-
-    def authenticate(self) -> dict:
-        """Authenticate the client and store the access & refresh tokens
-
-        returns an authentication data dictionary with the following schema:
-        {
-            "access_token": token str,
-            "refresh_token": token str,
-            "scope": scope str,
-            "token_type": token type str,
-            "expires_in": duration until expires in seconds
-        }
-
-        :return: authentication data (tokens, scope, token type, expires in)
-        :rtype: dict
-        """
-        url = self._base_url + TOKEN_ENDPOINT
-        payload = {
-            "grant_type": CLIENT_CREDENTIALS,
-            "client_id": self._client_id,
-            "client_secret": self._client_secret
-        }
-
-        response = requests.post(
-            url, json=payload, headers=self._generate_headers())
-        handle_error(response)
-
-        auth_data = response.json()
-
-        self._access_token = auth_data['access_token']
-        self._refresh_token = auth_data['refresh_token']
-        self._scope = auth_data['scope']
-        self._token_type = auth_data['token_type']
-        self._expires_at = time.time() + auth_data['expires_in']
-
-        return auth_data
-
-    def refresh(self) -> dict:
-        """Authenticate the client using the refresh token - refresh the access token.
-
-        updates the object's tokens, and:
-        returns an authentication data dictionary with the following schema:
-        {
-            "access_token": token str,
-            "refresh_token": token str,
-            "scope": scope str,
-            "token_type": token type str,
-            "expires_in": duration until expires in seconds
-        }
-
-        :return: authentication data (tokens, scope, token type, expires in)
-        :rtype: dict
-        """
-        url = self._base_url + TOKEN_ENDPOINT
-
-        payload = {
-            "grant_type": REFRESH_TOKEN,
-            "refresh_token": self._refresh_token
-        }
-
-        response = requests.post(
-            url, json=payload, headers=self._generate_headers())
-
-        # if refresh token is also expired, authenticate again:
-        if response.status_code == 401 or response.status_code == 400:
-            return self.authenticate()
-        else:
-            handle_error(response)
-
-        auth_data = response.json()
-
-        self._access_token = auth_data['access_token']
-        self._refresh_token = auth_data['refresh_token']
-        self._scope = auth_data['scope']
-        self._token_type = auth_data['token_type']
-        self._expires_at = time.time() + auth_data['expires_in']
-
-        return auth_data
-
-    def _generate_headers(self):
-        # get the first 10 chars of the client id
-        client_id_truncated = self._client_id[:10]
-        headers = {
-            'User-Agent': 'datacrunch-python-' + client_id_truncated
-        }
-        return headers
-
-    def is_expired(self) -> bool:
-        """Returns true if the access token is expired.
-
-        :return: True if the access token is expired, otherwise False.
-        :rtype: bool
-        """
-        return time.time() >= self._expires_at

datacrunch/balance/balance.py

@@ -1,52 +0,0 @@
-from typing import Dict
-
-BALANCE_ENDPOINT = '/balance'
-
-
-class Balance:
-    """A balance model class"""
-
-    def __init__(self, amount: float, currency: str) -> None:
-        """Initialize a new Balance object
-
-        :param amount: Balance amount
-        :type amount: float
-        :param currency: currency code
-        :type currency: str
-        """
-        self._amount = amount
-        self._currency = currency
-
-    @property
-    def amount(self) -> float:
-        """Get the balance amount
-
-        :return: amount
-        :rtype: float
-        """
-        return self._amount
-
-    @property
-    def currency(self) -> str:
-        """Get the currency code
-
-        :return: currency code
-        :rtype: str
-        """
-        return self._currency
-
-
-class BalanceService:
-    """A service for interacting with the balance endpoint"""
-
-    def __init__(self, http_client) -> None:
-        self._http_client = http_client
-
-    def get(self) -> Balance:
-        """Get the client's current balance
-
-        :return: Balance object containing the amount and currency.
-        :rtype: Balance
-        """
-        balance = self._http_client.get(BALANCE_ENDPOINT).json()
-        return Balance(balance["amount"], balance["currency"])

datacrunch/constants.py

@@ -1,107 +0,0 @@
-class Actions:
-    START = 'start'
-    SHUTDOWN = 'shutdown'
-    DELETE = 'delete'
-    HIBERNATE = 'hibernate'
-    RESTORE = 'restore'
-
-    def __init__(self):
-        return
-
-
-class VolumeActions:
-    ATTACH = 'attach'
-    DETACH = 'detach'
-    RENAME = 'rename'
-    INCREASE_SIZE = 'resize'
-    DELETE = 'delete'
-    CLONE = 'clone'
-
-    def __init__(self):
-        return
-
-
-class InstanceStatus:
-    ORDERED = 'ordered'
-    RUNNING = 'running'
-    PROVISIONING = 'provisioning'
-    OFFLINE = 'offline'
-    STARTING_HIBERNATION = 'starting_hibernation'
-    HIBERNATING = 'hibernating'
-    RESTORING = 'restoring'
-    ERROR = 'error'
-
-    def __init__(self):
-        return
-
-
-class VolumeStatus:
-    ORDERED = "ordered"
-    CREATING = "creating"
-    ATTACHED = "attached"
-    DETACHED = "detached"
-    DELETING = "deleting"
-    DELETED = "deleted"
-    CLONING = 'cloning'
-
-    def __init__(self):
-        return
-
-
-class VolumeTypes:
-    NVMe = "NVMe"
-    HDD = "HDD"
-
-    def __init__(self):
-        return
-
-
-class Locations:
-    FIN_01: str = "FIN-01"
-    ICE_01: str = "ICE-01"
-
-    def __init__(self):
-        return
-
-
-class ErrorCodes:
-    INVALID_REQUEST = "invalid_request"
-    UNAUTHORIZED_REQUEST = "unauthorized_request"
-    INSUFFICIENT_FUNDS = "insufficient_funds"
-    FORBIDDEN_ACTION = "forbidden_action"
-    NOT_FOUND = "not_found"
-    SERVER_ERROR = "server_error"
-    SERVICE_UNAVAILABLE = "service_unavailable"
-
-    def __init__(self):
-        return
-
-
-class Constants:
-    def __init__(self, base_url, version):
-        self.instance_actions: Actions = Actions()
-        """Available actions to perform on an instance"""
-
-        self.volume_actions: VolumeActions = VolumeActions()
-        """Available actions to perform on a volume"""
-
-        self.instance_status: InstanceStatus = InstanceStatus()
-        """Possible instance statuses"""
-
-        self.volume_status: VolumeStatus = VolumeStatus()
-        """Possible volume statuses"""
-
-        self.volume_types: VolumeTypes = VolumeTypes()
-        """Available volume types"""
-
-        self.locations: Locations = Locations()
-        """Available locations"""
-
-        self.error_codes: ErrorCodes = ErrorCodes()
-        """Available error codes"""
-
-        self.base_url: str = base_url
-        """DataCrunch's Public API URL"""
-
-        self.version: str = version
-        """Current SDK Version"""

datacrunch/containers/__init__.py

@@ -1,33 +0,0 @@
-from .containers import (
-    EnvVar,
-    EnvVarType,
-    ContainerRegistryType,
-    ContainerDeploymentStatus,
-    HealthcheckSettings,
-    EntrypointOverridesSettings,
-    VolumeMount,
-    SecretMount,
-    SharedFileSystemMount,
-    GeneralStorageMount,
-    VolumeMountType,
-    Container,
-    ContainerRegistryCredentials,
-    ContainerRegistrySettings,
-    ComputeResource,
-    ScalingPolicy,
-    QueueLoadScalingTrigger,
-    UtilizationScalingTrigger,
-    ScalingTriggers,
-    ScalingOptions,
-    Deployment,
-    ReplicaInfo,
-    Secret,
-    RegistryCredential,
-    ContainersService,
-    BaseRegistryCredentials,
-    DockerHubCredentials,
-    GithubCredentials,
-    GCRCredentials,
-    AWSECRCredentials,
-    CustomRegistryCredentials,
-)