qbraid-cli 0.9.2__tar.gz → 0.9.3__tar.gz

{qbraid_cli-0.9.2/qbraid_cli.egg-info → qbraid_cli-0.9.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qbraid-cli
-Version: 0.9.2
+Version: 0.9.3
 Summary: Command Line Interface for interacting with all parts of the qBraid platform.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -30,7 +30,7 @@ License-File: LICENSE
 Requires-Dist: typer>=0.12.1
 Requires-Dist: rich>=10.11.0
 Requires-Dist: click
-Requires-Dist: qbraid-core>=0.1.31
+Requires-Dist: qbraid-core>=0.1.32
 Provides-Extra: jobs
 Requires-Dist: amazon-braket-sdk>=1.48.1; extra == "jobs"
 Provides-Extra: envs

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/docs/index.rst

@@ -121,4 +121,5 @@ by disabling quantum jobs:
    tree/qbraid_jobs
    tree/qbraid_kernels
    tree/qbraid_pip
-   tree/qbraid_chat
+   tree/qbraid_chat
+   tree/qbraid_files

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "qbraid-cli"
-version = "0.9.2"
+version = "0.9.3"
 description = "Command Line Interface for interacting with all parts of the qBraid platform."
 readme = "README.md"
 authors = [{ name = "qBraid Development Team", email = "contact@qbraid.com" }]
@@ -31,7 +31,7 @@ dependencies = [
     "typer>=0.12.1",
     "rich>=10.11.0",
     "click",
-    "qbraid-core>=0.1.31",
+    "qbraid-core>=0.1.32",
 ]
 requires-python = ">= 3.9"
 

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/qbraid_cli/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.dev1+gd39b277'
-__version_tuple__ = version_tuple = (0, 1, 'dev1', 'gd39b277')
+__version__ = version = '0.1.dev1+gb960e1f'
+__version_tuple__ = version_tuple = (0, 1, 'dev1', 'gb960e1f')

qbraid_cli-0.9.3/qbraid_cli/files/__init__.py

@@ -0,0 +1,11 @@
+# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+
+"""
+Module defining the qbraid files namespace
+
+"""
+
+from .app import files_app
+
+__all__ = ["files_app"]

qbraid_cli-0.9.3/qbraid_cli/files/app.py

@@ -0,0 +1,118 @@
+# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+
+"""
+Module defining commands in the 'qbraid files' namespace.
+
+"""
+
+from pathlib import Path
+from typing import Any
+
+import rich
+import typer
+
+from qbraid_cli.handlers import run_progress_task
+
+files_app = typer.Typer(help="Manage qBraid cloud storage files.", no_args_is_help=True)
+
+
+@files_app.command(name="upload")
+def files_upload(
+    filepath: Path = typer.Argument(
+        ...,
+        exists=True,
+        dir_okay=False,
+        resolve_path=True,
+        help="Local path to the file to upload.",
+    ),
+    namespace: str = typer.Option(
+        "user",
+        "--namespace",
+        "-n",
+        help="Target qBraid namespace for the upload.",
+    ),
+    object_path: str = typer.Option(
+        None,
+        "--object-path",
+        "-p",
+        help=("Target object path. " "Defaults to original filename in namespace root."),
+    ),
+    overwrite: bool = typer.Option(
+        False,
+        "--overwrite",
+        "-o",
+        help="Overwrite existing file if it already exists in the target location.",
+    ),
+):
+    """Upload a local file to qBraid storage."""
+
+    def upload_file() -> dict[str, Any]:
+        from qbraid_core.services.files import FileManagerClient
+
+        client = FileManagerClient()
+        data = client.upload_file(
+            filepath, namespace=namespace, object_path=object_path, overwrite=overwrite
+        )
+        return data
+
+    data: dict = run_progress_task(
+        upload_file, description="Uploading file...", include_error_traceback=False
+    )
+
+    rich.print("File uploaded successfully!")
+    namespace = data.get("namespace")
+    object_path = data.get("objectPath")
+
+    if namespace and object_path:
+        rich.print(f"\nNamespace: '{namespace}'")
+        rich.print(f"Object path: '{object_path}'")
+
+
+@files_app.command(name="download")
+def files_download(
+    object_path: str = typer.Argument(
+        ...,
+        help="The folder + filename describing the file to download.",
+    ),
+    namespace: str = typer.Option(
+        "user",
+        "--namespace",
+        "-n",
+        help="Source qBraid namespace for the download.",
+    ),
+    save_path: Path = typer.Option(
+        Path.cwd(),
+        "--save-path",
+        "-s",
+        resolve_path=True,
+        help="Local directory to save the downloaded file.",
+    ),
+    overwrite: bool = typer.Option(
+        False,
+        "--overwrite",
+        "-o",
+        help="Overwrite existing file if it already exists in the target location.",
+    ),
+):
+    """Download a file from qBraid storage."""
+
+    def download_file() -> Path:
+        from qbraid_core.services.files import FileManagerClient
+
+        client = FileManagerClient()
+        file_path = client.download_file(
+            object_path, namespace=namespace, save_path=save_path, overwrite=overwrite
+        )
+        return file_path
+
+    file_path: Path = run_progress_task(
+        download_file, description="Downloading file...", include_error_traceback=False
+    )
+
+    rich.print("File downloaded successfully!")
+    rich.print(f"Saved to: '{str(file_path)}'")
+
+
+if __name__ == "__main__":
+    files_app()

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/qbraid_cli/main.py

@@ -15,6 +15,7 @@ from qbraid_cli.admin import admin_app
 from qbraid_cli.chat import ChatFormat, list_models_callback, prompt_callback
 from qbraid_cli.configure import configure_app
 from qbraid_cli.devices import devices_app
+from qbraid_cli.files import files_app
 from qbraid_cli.jobs import jobs_app
 
 try:
@@ -32,6 +33,7 @@ app.add_typer(admin_app, name="admin")
 app.add_typer(configure_app, name="configure")
 app.add_typer(account_app, name="account")
 app.add_typer(devices_app, name="devices")
+app.add_typer(files_app, name="files")
 app.add_typer(jobs_app, name="jobs")
 
 if ENVS_COMMANDS is True:

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3/qbraid_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qbraid-cli
-Version: 0.9.2
+Version: 0.9.3
 Summary: Command Line Interface for interacting with all parts of the qBraid platform.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -30,7 +30,7 @@ License-File: LICENSE
 Requires-Dist: typer>=0.12.1
 Requires-Dist: rich>=10.11.0
 Requires-Dist: click
-Requires-Dist: qbraid-core>=0.1.31
+Requires-Dist: qbraid-core>=0.1.32
 Provides-Extra: jobs
 Requires-Dist: amazon-braket-sdk>=1.48.1; extra == "jobs"
 Provides-Extra: envs

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/qbraid_cli.egg-info/SOURCES.txt

@@ -58,6 +58,8 @@ qbraid_cli/envs/activate.py
 qbraid_cli/envs/app.py
 qbraid_cli/envs/create.py
 qbraid_cli/envs/data_handling.py
+qbraid_cli/files/__init__.py
+qbraid_cli/files/app.py
 qbraid_cli/jobs/__init__.py
 qbraid_cli/jobs/app.py
 qbraid_cli/jobs/toggle_braket.py
@@ -86,6 +88,8 @@ tests/envs/test_envs_activate_print_command.py
 tests/envs/test_envs_activate_pyenv.py
 tests/envs/test_envs_list.py
 tests/envs/test_envs_remove.py
+tests/files/__init__.py
+tests/files/test_files_commands.py
 tests/jobs/__init__.py
 tests/jobs/test_jobs_disable.py
 tests/jobs/test_jobs_enable.py

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/qbraid_cli.egg-info/requires.txt

@@ -1,7 +1,7 @@
 typer>=0.12.1
 rich>=10.11.0
 click
-qbraid-core>=0.1.31
+qbraid-core>=0.1.32
 
 [dev]
 isort

qbraid_cli-0.9.3/tests/files/test_files_commands.py

@@ -0,0 +1,109 @@
+# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+
+# pylint: disable=redefined-outer-name
+
+"""
+Unit tests for the `qbraid_cli.files.app` module.
+
+"""
+
+from unittest.mock import patch
+
+import pytest
+from typer.testing import CliRunner
+
+from qbraid_cli.files.app import files_app
+
+
+@pytest.fixture
+def runner():
+    """Fixture for invoking CLI commands."""
+    return CliRunner()
+
+
+def test_files_upload(runner, tmp_path):
+    """Test the `files upload` command."""
+    test_file = tmp_path / "test_file.txt"
+    test_file.write_text("Test content")
+
+    with patch("qbraid_core.services.files.FileManagerClient") as mock_client:
+        mock_client.return_value.upload_file.return_value = {
+            "namespace": "user",
+            "objectPath": "test_file.txt",
+        }
+
+        result = runner.invoke(files_app, ["upload", str(test_file)])
+
+    assert result.exit_code == 0
+    assert "File uploaded successfully!" in result.stdout
+    assert "Namespace: 'user'" in result.stdout
+    assert "Object path: 'test_file.txt'" in result.stdout
+
+
+def test_files_upload_with_options(runner, tmp_path):
+    """Test the `files upload` command with options."""
+    test_file = tmp_path / "test_file.txt"
+    test_file.write_text("Test content")
+
+    with patch("qbraid_core.services.files.FileManagerClient") as mock_client:
+        mock_client.return_value.upload_file.return_value = {
+            "namespace": "custom",
+            "objectPath": "folder/test_file.txt",
+        }
+
+        result = runner.invoke(
+            files_app,
+            [
+                "upload",
+                str(test_file),
+                "--namespace",
+                "custom",
+                "--object-path",
+                "folder/test_file.txt",
+                "--overwrite",
+            ],
+        )
+
+    assert result.exit_code == 0
+    assert "File uploaded successfully!" in result.stdout
+    assert "Namespace: 'custom'" in result.stdout
+    assert "Object path: 'folder/test_file.txt'" in result.stdout
+
+
+def test_files_download(runner, tmp_path):
+    """Test the `files download` command."""
+    with patch("qbraid_core.services.files.FileManagerClient") as mock_client:
+        mock_client.return_value.download_file.return_value = tmp_path / "downloaded_file.txt"
+
+        result = runner.invoke(files_app, ["download", "test_file.txt"])
+
+    assert result.exit_code == 0
+    assert "File downloaded successfully!" in result.stdout
+    assert f"Saved to: '{(tmp_path / 'downloaded_file.txt')}'" in result.stdout.replace("\n", "")
+
+
+def test_files_download_with_options(runner, tmp_path):
+    """Test the `files download` command with options."""
+    save_path = tmp_path / "custom_folder"
+    save_path.mkdir()
+
+    with patch("qbraid_core.services.files.FileManagerClient") as mock_client:
+        mock_client.return_value.download_file.return_value = save_path / "downloaded_file.txt"
+
+        result = runner.invoke(
+            files_app,
+            [
+                "download",
+                "folder/test_file.txt",
+                "--namespace",
+                "custom",
+                "--save-path",
+                str(save_path),
+                "--overwrite",
+            ],
+        )
+
+    assert result.exit_code == 0
+    assert "File downloaded successfully!" in result.stdout
+    assert f"Saved to: '{(save_path / 'downloaded_file.txt')}'" in result.stdout.replace("\n", "")

{qbraid_cli-0.9.2 → qbraid_cli-0.9.3}/tools/split_md.py

@@ -137,6 +137,8 @@ def markdown_to_mdx(file_path: str) -> None:
             )
             continue
 
+        line = line.replace("'", "'")
+
         new_lines.append(line)
 
     # Construct the new file path with .mdx extension