msdev-kit 0.1.8__tar.gz → 0.2.0__tar.gz

{msdev_kit-0.1.8 → msdev_kit-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: msdev-kit
-Version: 0.1.8
+Version: 0.2.0
 Summary: Microsoft developer toolkit: Fabric, MS Graph, and SharePoint
 License: MIT
 License-File: LICENSE
@@ -14,17 +14,19 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
-Requires-Dist: azure-identity (>=1.24.0,<2)
-Requires-Dist: azure-kusto-data (>=6.0.1,<7.0.0)
-Requires-Dist: cryptography (>=46.0.6,<47.0.0)
+Requires-Dist: azure-identity (>=1.25.3,<2)
+Requires-Dist: azure-kusto-data (>=6.0.3,<7.0.0)
+Requires-Dist: cryptography (>=48.0.1,<49.0.0)
+Requires-Dist: ipykernel (>=7.3.0,<8.0.0)
 Requires-Dist: numpy (==2.1.3)
 Requires-Dist: openpyxl (==3.1.5)
 Requires-Dist: pandas (==2.2.3)
 Requires-Dist: pydantic (>=2.13.4,<3.0.0)
+Requires-Dist: pyjwt (>=2.13.0,<3.0.0)
 Requires-Dist: pyodbc (==5.2.0)
 Requires-Dist: python-dotenv (>=1.2.2,<2)
 Requires-Dist: pyyaml (>=6.0.3,<7.0.0)
-Requires-Dist: requests (>=2.33.0)
+Requires-Dist: requests (>=2.34.2)
 Requires-Dist: sqlalchemy (==2.0.44)
 Project-URL: Homepage, https://github.com/Bernardo-Rufino/msdev-kit
 Project-URL: Repository, https://github.com/Bernardo-Rufino/msdev-kit
@@ -374,7 +376,7 @@ sp.create_folder('/Reports/2026/Q1')
 | Method | Description |
 |---|---|
 | `download_file(file_path, local_dir)` | Download a file from the default document library. Returns local file path. |
-| `upload_file(remote_path, source, content_type?)` | Upload/overwrite a file. `source` is a local file path (str) or raw bytes. |
+| `upload_file(remote_path, source, content_type?)` | Upload/overwrite a file up to 250 MB. `source` is a local file path or raw bytes. |
 | `create_folder(folder_path)` | Create a folder and all intermediate folders. |
 
 Hostname and site path inputs are normalized automatically:

{msdev_kit-0.1.8 → msdev_kit-0.2.0}/README.md

@@ -342,7 +342,7 @@ sp.create_folder('/Reports/2026/Q1')
 | Method | Description |
 |---|---|
 | `download_file(file_path, local_dir)` | Download a file from the default document library. Returns local file path. |
-| `upload_file(remote_path, source, content_type?)` | Upload/overwrite a file. `source` is a local file path (str) or raw bytes. |
+| `upload_file(remote_path, source, content_type?)` | Upload/overwrite a file up to 250 MB. `source` is a local file path or raw bytes. |
 | `create_folder(folder_path)` | Create a folder and all intermediate folders. |
 
 Hostname and site path inputs are normalized automatically:

msdev_kit-0.2.0/msdev_kit/__init__.py

@@ -0,0 +1,4 @@
+from .auth import Auth
+
+__version__ = "0.2.0"
+__all__ = ["Auth"]

msdev_kit-0.2.0/msdev_kit/auth.py

@@ -0,0 +1,52 @@
+from azure.identity import (
+    ClientSecretCredential,
+    InteractiveBrowserCredential,
+    TokenCachePersistenceOptions,
+)
+
+_SCOPES = {
+    "pbi": "https://analysis.windows.net/powerbi/api/.default",
+    "fabric": "https://api.fabric.microsoft.com/.default",
+    "azure": "https://management.azure.com/.default",
+    "graph": "https://graph.microsoft.com/.default",
+}
+
+
+class Auth:
+    def __init__(
+        self, tenant_id: str = "", client_id: str = "", client_secret: str = ""
+    ):
+        self.tenant_id = tenant_id
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self._credential = (
+            ClientSecretCredential(
+                authority="https://login.microsoftonline.com/",
+                tenant_id=tenant_id,
+                client_id=client_id,
+                client_secret=client_secret,
+            )
+            if all((tenant_id, client_id, client_secret))
+            else InteractiveBrowserCredential(
+                cache_persistence_options=TokenCachePersistenceOptions()
+            )
+        )
+
+    def get_token(self, service: str = "pbi") -> str:
+        scope = _SCOPES.get(service)
+        if not scope:
+            raise ValueError(
+                f"Invalid service specified. Choose one of: {', '.join(_SCOPES)}"
+            )
+        return self._credential.get_token(scope).token
+
+    def get_token_for_user(self, service: str = "pbi") -> str:
+        scope = _SCOPES.get(service)
+        if not scope:
+            raise ValueError(
+                f"Invalid service specified. Choose one of: {', '.join(_SCOPES)}"
+            )
+        credential = InteractiveBrowserCredential(
+            cache_persistence_options=TokenCachePersistenceOptions()
+        )
+        return credential.get_token(scope).token

{msdev_kit-0.1.8 → msdev_kit-0.2.0}/msdev_kit/fabric/__init__.py

@@ -9,3 +9,17 @@ from .kql import KQLDatabase
 from .database import Database
 from .pipeline import Pipeline
 from .notebook import Notebook
+
+__all__ = [
+    "Workspace",
+    "Dataset",
+    "Report",
+    "Dataflow",
+    "Capacity",
+    "Admin",
+    "Operations",
+    "KQLDatabase",
+    "Database",
+    "Pipeline",
+    "Notebook",
+]

{msdev_kit-0.1.8 → msdev_kit-0.2.0}/msdev_kit/fabric/capacity.py

@@ -1,4 +1,3 @@
-import os
 import json
 import requests
 import pandas as pd
@@ -9,31 +8,32 @@ from .workspace import Workspace
 
 class Capacity:
 
-    def __init__(self, pbi_token: str, fabric_token: str = None, azure_token: str = None):
+    def __init__(
+        self, pbi_token: str, fabric_token: str = None, azure_token: str = None
+    ):
         """
         Initialize variables.
         """
         # Power BI Capacity parameters
-        self.main_url = 'https://api.powerbi.com/v1.0/myorg'
+        self.main_url = "https://api.powerbi.com/v1.0/myorg"
 
         # Fabric Capacity parameters
-        self.fabric_api_base_url = 'https://management.azure.com'
+        self.fabric_api_base_url = "https://management.azure.com"
         self.azure_subscription_id = None
         self.azure_resource_group = None
 
         # General parameters
         self.token = pbi_token
-        self.headers = {'Authorization': f'Bearer {self.token}'}
+        self.headers = {"Authorization": f"Bearer {self.token}"}
         self.workspace = Workspace(self.token)
 
         # Directories
-        self.capacities_dir = './data/capacities'
+        self.capacities_dir = "./data/capacities"
         self.directories = [self.capacities_dir]
 
         for dir in self.directories:
             create_directory(dir)
 
-
     def list_powerbi_capacities(self) -> Dict:
         """
         List all Power BI capacities that the user has access to.
@@ -44,44 +44,45 @@ class Capacity:
         Returns:
             Dict: status message and content.
         """
-        
+
         # Main URL
-        request_url = f'{self.main_url}/capacities'
+        request_url = f"{self.main_url}/capacities"
 
-        filename = f'capacities_powerbi.xlsx'
+        filename = "capacities_powerbi.xlsx"
 
         # 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).get('value', '')
+        response = json.loads(r.content).get("value", "")
 
         # If success...
         if status == 200:
-            if type == 'fabric':
+            if type == "fabric":  # noqa: E721
                 df = pd.json_normalize(response)
-                df['name'] = df['displayName']
-                df.drop(columns=['displayName'], inplace=True)
+                df["name"] = df["displayName"]
+                df.drop(columns=["displayName"], inplace=True)
             else:
                 df = pd.DataFrame(response)
-            df.to_excel(f'{self.capacities_dir}/{filename}', index=False)
-            result = json.loads(df.to_json(orient='records'))
+            df.to_excel(f"{self.capacities_dir}/{filename}", index=False)
+            result = json.loads(df.to_json(orient="records"))
 
-            return {'message': 'Success', 'content': result}
+            return {"message": "Success", "content": result}
 
-        else:                
+        else:
             # If any error happens, return message.
             response = json.loads(r.content)
-            error_message = response['error']['message']
+            error_message = response["error"]["message"]
 
-            return {'message': {'error': error_message, 'content': response}}
+            return {"message": {"error": error_message, "content": response}}
 
-
-    def list_fabric_capacities(self, azure_subscription_id: str, azure_resource_group: str = None) -> Dict:
+    def list_fabric_capacities(
+        self, azure_subscription_id: str, azure_resource_group: str = None
+    ) -> Dict:
         """
         List all Fabric capacities that the user has access to, to a given subscription.
-        
+
         If a resource group is provided, only capacities within that resource group will be listed.
 
         Args:
@@ -91,47 +92,45 @@ class Capacity:
         Returns:
             Dict: status message and content.
         """
-        
+
         # Main URL
         if azure_resource_group:
-            request_url = f'{self.fabric_api_base_url}/subscriptions/{azure_subscription_id}/resourceGroups/{azure_resource_group}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01'
+            request_url = f"{self.fabric_api_base_url}/subscriptions/{azure_subscription_id}/resourceGroups/{azure_resource_group}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01"
         else:
-            request_url = f'{self.fabric_api_base_url}/subscriptions/{azure_subscription_id}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01'
+            request_url = f"{self.fabric_api_base_url}/subscriptions/{azure_subscription_id}/providers/Microsoft.Fabric/capacities?api-version=2023-11-01"
 
-        filename = f'capacities_powerbi.xlsx'
+        filename = "capacities_powerbi.xlsx"
 
         # 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).get('value', '')
+        response = json.loads(r.content).get("value", "")
 
         # If success...
         if status == 200:
-            if type == 'fabric':
+            if type == "fabric":  # noqa: E721
                 df = pd.json_normalize(response)
-                df['name'] = df['displayName']
-                df.drop(columns=['displayName'], inplace=True)
+                df["name"] = df["displayName"]
+                df.drop(columns=["displayName"], inplace=True)
             else:
                 df = pd.DataFrame(response)
-            df.to_excel(f'{self.capacities_dir}/{filename}', index=False)
-            result = json.loads(df.to_json(orient='records'))
+            df.to_excel(f"{self.capacities_dir}/{filename}", index=False)
+            result = json.loads(df.to_json(orient="records"))
 
-            return {'message': 'Success', 'content': result}
+            return {"message": "Success", "content": result}
 
-        else:                
+        else:
             # If any error happens, return message.
             response = json.loads(r.content)
-            error_message = response['error']['message']
-
-            return {'message': {'error': error_message, 'content': response}}
+            error_message = response["error"]["message"]
 
+            return {"message": {"error": error_message, "content": response}}
 
     def assign_workspace_to_capacity(
-                self, 
-                workspace_id: str = '',
-                capacity_id: str = '') -> Dict:
+        self, workspace_id: str = "", capacity_id: str = ""
+    ) -> Dict:
         """
         Assign a workspace to a specific capacity.
 
@@ -144,23 +143,22 @@ class Capacity:
         """
 
         # If both workspace and capacity were provided...
-        if (workspace_id != '') & (capacity_id != ''):
+        if (workspace_id != "") & (capacity_id != ""):
 
             ws = self.workspace.get_worspace_details(workspace_id)
-            workspace_name = ws.get('content', {}).get('name', None)
-            current_capacity_id = ws.get('content', {}).get('capacityId', None)
+            current_capacity_id = ws.get("content", {}).get("capacityId", None)
 
             if current_capacity_id.lower() == capacity_id.lower():
-                return {'message': 'Workspace is already assigned to the specified capacity.'}
+                return {
+                    "message": "Workspace is already assigned to the specified capacity."
+                }
 
-            request_url = self.main_url + f'/groups/{workspace_id}/AssignToCapacity'
+            request_url = self.main_url + f"/groups/{workspace_id}/AssignToCapacity"
 
-            headers = {'Authorization': f'Bearer {self.token}'}
+            headers = {"Authorization": f"Bearer {self.token}"}
 
             # https://learn.microsoft.com/en-us/rest/api/power-bi/capacities/groups-assign-to-capacity
-            data = {
-                "capacityId": capacity_id
-            }
+            data = {"capacityId": capacity_id}
 
             # Make the request
             r = requests.post(url=request_url, headers=headers, json=data)
@@ -170,17 +168,24 @@ class Capacity:
 
             # If success...
             if status == 200:
-                return {'message': 'Success'}
-            
+                return {"message": "Success"}
+
             elif status == 401:
-                return {'message': 'Unauthorized. Please check your access level. Workspace administration rights are required to perform this action.'}
-            
-            else:                
+                return {
+                    "message": "Unauthorized. Please check your access level. Workspace administration rights are required to perform this action."
+                }
+
+            else:
                 # If any error happens, return message.
                 response = json.loads(r.content)
-                error_message = response['error']['code']
+                error_code = response.get("error", {}).get("code", "Unknown error")
 
-                return {'message': {'error': {'status': status, 'description': ''}, 'content': response.content}}
+                return {
+                    "message": {
+                        "error": {"status": status, "description": error_code},
+                        "content": response,
+                    }
+                }
 
         else:
-            return {'message': 'Missing parameters, please check.', 'content': ''}
+            return {"message": "Missing parameters, please check.", "content": ""}

{msdev_kit-0.1.8 → msdev_kit-0.2.0}/msdev_kit/fabric/dataflow.py

@@ -274,7 +274,7 @@ class Dataflow:
                     response = json.loads(r.content)
                     error_message = response['error']
 
-                except:
+                except Exception:
                     return {'message': 'Error reading JSON response'}
                 
                 return {'message': {'error': error_message, 'content': response}}
@@ -337,7 +337,7 @@ class Dataflow:
                     response = json.loads(r.content)
                     error_message = response['error']
 
-                except:
+                except Exception:
                     return {'message': 'Error reading JSON response'}
                 
                 return {'message': {'error': error_message, 'content': response}}
@@ -926,7 +926,6 @@ class Dataflow:
             return {'message': 'Could not parse mashup.pq.', 'content': ''}
 
         data_queries = parsed['data_queries']
-        source_queries = parsed['source_queries']
         dest_queries = parsed['dest_queries']
 
         if not data_queries: