msdev-kit 0.1.0__py3-none-any.whl

msdev_kit/fabric/notebook.py

@@ -0,0 +1,88 @@
+import time
+import requests
+from typing import Dict
+
+
+class Notebook:
+
+    def __init__(self, token: str):
+        """
+        Initialize variables.
+        """
+        self.fabric_api_base_url = 'https://api.fabric.microsoft.com'
+        self.token = token
+        self.headers = {'Authorization': f'Bearer {self.token}'}
+
+
+    def _request_with_retry(self, method: str, url: str, max_retries: int = 3, **kwargs) -> requests.Response:
+        """
+        Makes an HTTP request with automatic retry on 429 (Too Many Requests).
+        Respects the Retry-After header when present.
+        """
+        for attempt in range(max_retries + 1):
+            response = requests.request(method, url, **kwargs)
+            if response.status_code != 429:
+                return response
+
+            retry_after = int(response.headers.get('Retry-After', 5))
+            print(f"  Rate limited (429). Retrying in {retry_after}s... (attempt {attempt + 1}/{max_retries})")
+            time.sleep(retry_after)
+
+        return response
+
+
+    def list_notebooks(self, workspace_id: str) -> Dict:
+        """
+        Lists all notebooks in a workspace.
+
+        Args:
+            workspace_id (str): The ID of the workspace.
+
+        Returns:
+            Dict: 'message' and 'content' (list of notebook dicts with id, displayName, description).
+        """
+        if workspace_id == '':
+            return {'message': 'Missing workspace id, please check.', 'content': ''}
+
+        api_url = f'{self.fabric_api_base_url}/v1/workspaces/{workspace_id}/notebooks'
+        notebooks = []
+
+        while api_url:
+            response = self._request_with_retry('GET', api_url, headers=self.headers)
+            if response.status_code != 200:
+                error_data = response.json() if response.headers.get('content-type', '').startswith('application/json') else {}
+                error_message = error_data.get('message', error_data.get('error', {}).get('message', response.text))
+                return {'message': {'error': error_message, 'status_code': response.status_code}}
+
+            data = response.json()
+            notebooks.extend(data.get('value', []))
+            api_url = data.get('continuationUri', None)
+
+        return {'message': 'Success', 'content': notebooks}
+
+
+    def get_notebook(self, workspace_id: str, notebook_id: str) -> Dict:
+        """
+        Gets the metadata of a specific notebook.
+
+        Args:
+            workspace_id (str): The ID of the workspace.
+            notebook_id (str): The ID of the notebook.
+
+        Returns:
+            Dict: 'message' and 'content' (notebook dict with id, displayName, description, etc.).
+        """
+        if workspace_id == '':
+            return {'message': 'Missing workspace id, please check.', 'content': ''}
+        if notebook_id == '':
+            return {'message': 'Missing notebook id, please check.', 'content': ''}
+
+        api_url = f'{self.fabric_api_base_url}/v1/workspaces/{workspace_id}/notebooks/{notebook_id}'
+        response = self._request_with_retry('GET', api_url, headers=self.headers)
+
+        if response.status_code == 200:
+            return {'message': 'Success', 'content': response.json()}
+        else:
+            error_data = response.json() if response.headers.get('content-type', '').startswith('application/json') else {}
+            error_message = error_data.get('message', error_data.get('error', {}).get('message', response.text))
+            return {'message': {'error': error_message, 'status_code': response.status_code}}

msdev_kit/fabric/operations.py

@@ -0,0 +1,108 @@
+import json
+import requests
+from typing import Dict
+from .utilities import create_directory
+
+
+class Operations:
+
+    def __init__(self, token: str):
+        """
+        Initialize variables.
+        """
+        self.main_url = 'https://api.fabric.microsoft.com/v1'
+        self.token = token
+        self.headers = {'Authorization': f'Bearer {self.token}'}
+        self.data_dir = './data/operations'
+
+        create_directory(self.data_dir)
+
+
+    def get_operation_state(
+                self, 
+                operation_id: str = '') -> str:
+        """
+        Get the operation state for a long running operation.
+
+        Args:
+            operation_id (str, optional): operation id to check state.
+
+        Returns:
+            Dict: message and operation state.
+        """
+
+        DEFAULT_STATE = 'unknown'
+
+        # Main URL
+        request_url = f'{self.main_url}/operations/{operation_id}'
+
+        # If operation ID was not informed, return error message...
+        if operation_id == '':
+            return {'message': 'Missing operation id, please check.', 'operation_state': DEFAULT_STATE}
+
+        # If workspace ID was informed...
+        else:
+
+            # Make the request
+            r = requests.get(url=request_url, headers=self.headers)
+
+            # Get HTTP status and content
+            status = r.status_code
+            response = json.loads(r.content)
+
+            # If success...
+            if status == 200:
+                # Get the state
+                operation_state = response.get('status', DEFAULT_STATE)
+                
+                return {'message': 'Success', 'operation_state': operation_state}
+
+            else:                
+                # If any error happens, return message.
+                response = json.loads(r.content)
+                error_message = response['error']['message']
+
+                return {'message': {'error': error_message}, 'operation_state': DEFAULT_STATE}
+
+
+    def get_operation_result(
+                self, 
+                operation_id: str = '') -> Dict:
+        """
+        Get the operation result for a long running operation.
+
+        Args:
+            operation_id (str, optional): operation id to get result from.
+
+        Returns:
+            Dict: json with it's contents.
+        """
+
+        # Main URL
+        request_url = f'{self.main_url}/operations/{operation_id}/result'
+
+        # If operation ID was not informed, return error message...
+        if operation_id == '':
+            return {'message': 'Missing operation id, please check.', 'content': ''}
+
+        # If workspace ID was informed...
+        else:
+
+            # Make the request
+            r = requests.get(url=request_url, headers=self.headers)
+
+            # Get HTTP status and content
+            status = r.status_code
+            response = json.loads(r.content)
+
+            # If success...
+            if status == 200:
+                
+                return {'message': 'Success', 'content': response}
+
+            else:                
+                # If any error happens, return message.
+                response = json.loads(r.content)
+                error_message = response['error']['message']
+
+                return {'message': {'error': error_message}, 'content': ''}