feat(cli): add csv download functionality to device commands

Add download_activity_csv utility function to handle downloading activity reports
and integrate it with both assign_device and unassign_device commands to provide
users with downloaded CSV files when available.

Author: kaincenteno

pyaxm/cli.py

@@ -4,6 +4,7 @@
 from typing_extensions import Annotated
 from typing import List
 from pyaxm.client import Client
+from pyaxm.utils import download_activity_csv
 
 app = typer.Typer()
 
@@ -80,6 +81,10 @@ def assign_device(device_id: Annotated[List[str], typer.Argument()], server_id:
     df = pd.DataFrame([activity_data])
     df.to_csv(sys.stdout, index=False)
 
+    file_path = download_activity_csv(activity)
+    if file_path:
+        typer.echo(f"Report downloaded successfully to: {file_path}")
+
 @app.command()
 def unassign_device(device_id: Annotated[List[str], typer.Argument()], server_id: Annotated[str, typer.Argument()]):
     """Unassign one or more devices from an MDM server."""
@@ -89,6 +94,10 @@ def unassign_device(device_id: Annotated[List[str], typer.Argument()], server_id
     activity_data.update(activity.attributes.model_dump())
     df = pd.DataFrame([activity_data])
     df.to_csv(sys.stdout, index=False)
+    
+    file_path = download_activity_csv(activity)
+    if file_path:
+        typer.echo(f"Report downloaded successfully to: {file_path}")
 
 if __name__ == "__main__":
     app()

pyaxm/utils.py

@@ -0,0 +1,39 @@
+import os
+import re
+import requests
+from typing import Optional
+from urllib.parse import urlparse, parse_qs
+from pyaxm.models import OrgDeviceActivity
+
+
+def download_activity_csv(activity: OrgDeviceActivity, save_path: str = None) -> Optional[str]:
+    """
+    Download CSV file from activity if download URL is available.
+    
+    :param activity: OrgDeviceActivity object with attributes.downloadUrl property
+    :param save_path: Optional path to save file (defaults to current directory)
+    :return: Path to saved file or None if no download URL
+    """
+    if not save_path:
+        save_path = os.getcwd()
+
+    if not (activity.attributes and activity.attributes.downloadUrl):
+        return None
+
+    url = activity.attributes.downloadUrl
+    parsed_url = urlparse(url)
+    query_params = parse_qs(parsed_url.query)
+    
+    disposition = query_params['response-content-disposition'][0]
+    match = re.search(r'filename="([^"]+)"', disposition)
+    filename = match.group(1)
+    
+    file_path = os.path.join(save_path, filename)
+    
+    response = requests.get(url)
+    response.raise_for_status()
+    
+    with open(file_path, 'wb') as f:
+        f.write(response.content)
+    
+    return file_path