codex-manager 3.0.0__tar.gz → 4.0.0__tar.gz

codex_manager-4.0.0/PKG-INFO

@@ -0,0 +1,201 @@
+Metadata-Version: 2.4
+Name: codex-manager
+Version: 4.0.0
+Summary: Codex account snapshot manager
+Author-email: Dhruv <dhruv13x@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/dhruv13x/codex-manager
+Project-URL: Repository, https://github.com/dhruv13x/codex-manager
+Project-URL: Issues, https://github.com/dhruv13x/codex-manager/issues
+Keywords: codex,backup,cli,automation
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Environment :: Console
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: rich>=13.0.0
+Requires-Dist: boto3>=1.28.0
+Requires-Dist: b2sdk>=2.0.0
+Requires-Dist: relm>=7.1.1
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=5.0.0; extra == "dev"
+Requires-Dist: pytest-timeout>=2.2.0; extra == "dev"
+Requires-Dist: pytest-json-report>=1.5.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.10.0; extra == "dev"
+Requires-Dist: pyfakefs>=5.0.0; extra == "dev"
+Requires-Dist: ruff>=0.6.0; extra == "dev"
+Requires-Dist: black>=24.3.0; extra == "dev"
+Requires-Dist: mypy>=1.11.0; extra == "dev"
+Requires-Dist: types-PyYAML>=6.0; extra == "dev"
+
+# 📦 Codex Manager
+
+**The ultimate CLI tool for managing OpenAI Codex account snapshots, tracking quotas, and ensuring seamless workflow continuity.**
+
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
+[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)](https://www.python.org/downloads/)
+[![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![Linter: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+[![Maintenance Status](https://img.shields.io/badge/Maintenance-Active-success)](#)
+
+---
+
+## ⚡ Quick Start
+
+Get up and running in under 5 minutes.
+
+### Prerequisites
+- Python 3.10 or higher
+- `uv` (recommended for fast dependency management)
+
+### Install
+Clone the repository and install using `uv`:
+```bash
+git clone https://github.com/dhruv13x/codex-manager.git
+cd codex-manager
+uv pip install --system -e .
+```
+
+### Run
+Verify the installation by viewing the CLI help menu:
+```bash
+codex-manager --help
+# Or use the shorter alias:
+cm --help
+```
+
+### Demo
+Here's a quick 5-line workflow to backup your current account, view recommendations, and switch to a new one:
+```bash
+# 1. Take a live snapshot of your active Codex account state
+cm backup --cloud
+
+# 2. Check cooldown statuses for all accounts
+cm cooldown
+
+# 3. Get the smartest recommendation for the next account to use
+cm recommend
+
+# 4. Switch to a new account using the 'auth-only' method
+cm use --email new_user@example.com
+```
+
+> **[🖼️ Suggestion: Add an animated GIF here demonstrating the `cm use` command in action with the Rich UI]**
+
+---
+
+## ✨ Features
+
+**Core**
+- **Live Status Tracking:** Automatically parses live Codex `/status` output to capture account email, quota text, and weekly reset timestamps.
+- **Smart Recommendations:** Recommends the optimal account to use next based on calculated cooldowns and real-time metadata.
+- **Full State Recovery:** Backup and restore full Codex runtime states (`auth.json`, history, logs) via `*.tar.gz` archives and `*.metadata.json`.
+
+**Performance & Reliability**
+- **Offline & Fallback Mode:** Employs an emergency `--without-status-check` fallback that gracefully estimates cooldowns even when the live tracker is temporarily unavailable.
+- **Lightning Fast UI:** Employs the `rich` library for beautiful terminal output, tables, and status animations.
+
+**Security & Cloud**
+- **Cloud Synchronization:** First-class support for Backblaze B2 (and S3-compatible buckets) for remote backup metadata and archive storage.
+- **Safe Operations:** Every modifying command supports a `--dry-run` flag to safely simulate actions without touching your files.
+
+---
+
+## 🛠️ Configuration
+
+Codex Manager prioritizes configuration via environment variables and CLI arguments for flexibility.
+
+### Environment Variables
+
+| Name | Description | Default | Required |
+|---|---|---|---|
+| `CODEX_MANAGER_HOME` | The primary home directory for the manager config and backups. | `~/.codex-manager` | No |
+| `CODEX_HOME` | The target directory where Codex state resides. | `~/.codex` | No |
+
+*Note: You can also place a `config.json` inside your `CODEX_MANAGER_HOME` to persist configuration settings.*
+
+### Key CLI Arguments
+
+Most commands support these primary flags. Use `cm <command> --help` for a complete list.
+
+| Flag | Description |
+|---|---|
+| `--dry-run` | Safely preview the changes without modifying local or cloud state. |
+| `--cloud` | Enable Backblaze B2/S3 cloud capabilities for the command. |
+| `--email <email>` | Specify a target account email for use, restore, or listing. |
+| `--backup-dir <dir>` | Override the directory used for reading/writing backups. |
+| `--without-status-check`| Bypass live capture and calculate cooldowns statically (+7 days). |
+| `--auth-only` | During `use` or `backup`, target only identity/auth files instead of the full state. |
+
+---
+
+## 🏗️ Architecture
+
+### Directory Tree
+
+```
+~/.codex-manager/ (CODEX_MANAGER_HOME)
+├── backups/                # Local archives and metadata
+│   ├── backup_1.tar.gz
+│   └── backup_1.metadata.json
+├── config.json             # Persistent local configurations
+└── cooldown.json           # Registry caching overall account cooldowns
+```
+
+### High-Level Data Flow
+1. **Capture:** `cm backup` or `cm status` reads live text from a `tmux` session running Codex.
+2. **Process:** The CLI parses the text to build a state model (Quota, Cooldown, Email) and packages the `~/.codex` directory into an archive.
+3. **Store:** Archives and adjacent JSON metadata are saved locally and pushed to the cloud (if configured).
+4. **Evaluate:** When `cm recommend` or `cm cooldown` is invoked, local and cloud metadata are fetched, evaluated against real-time, and ranked to find the optimal active account.
+5. **Switch:** `cm use` restores the selected account's data into the `~/.codex` home, rotating your session seamlessly.
+
+---
+
+## 🐞 Troubleshooting
+
+### Common Issues
+
+| Error Message | Cause | Solution |
+|---|---|---|
+| `TokenExpiredError: TOKEN EXPIRED` | The active Codex session token has expired. | Re-authenticate in Codex manually, or run with `--without-status-check` to bypass. |
+| `Could not resolve Cloud (B2) credentials.` | Missing B2 credentials for cloud sync. | Pass `--b2-id` and `--b2-key` flags, or ensure your credentials are set up. |
+| `No backups found in Cloud for <email>.` | The requested account isn't backed up to the specified bucket. | Run `cm list-backups --cloud` to verify the email and backup availability. |
+
+### Debug Mode
+While the CLI does not have a single `--debug` flag, you can often reveal more information by viewing the full exception traces or by utilizing the built-in doctor command:
+```bash
+cm doctor
+```
+The `doctor` command verifies your dependencies, runtime directories, and validates the status parser setup.
+
+---
+
+## 🤝 Contributing
+
+We welcome contributions to Codex Manager! Please review our `CONTRIBUTING.md` (coming soon) before submitting PRs.
+
+### Dev Setup
+To set up your local development environment:
+```bash
+# 1. Install all development dependencies
+uv pip install --system -e .[dev]
+
+# 2. Run the tests (Ensure 90%+ coverage)
+python -m pytest tests --cov=src --cov-report=term-missing
+
+# 3. Format and lint the codebase
+uv run ruff check --fix src/ tests/
+uv run black src/ tests/
+```
+
+---
+
+## 🗺️ Roadmap
+
+- **Plugin Architecture:** Allow custom plugins to manage other CLI authentication tokens.
+- **Enhanced Cloud Coverage:** Add direct first-class integrations for Google Cloud Storage and Azure Blob.
+- **Automated Rotation Daemon:** A background worker to automatically rotate accounts when token limits are reached in real-time.

codex_manager-4.0.0/README.md

@@ -0,0 +1,167 @@
+# 📦 Codex Manager
+
+**The ultimate CLI tool for managing OpenAI Codex account snapshots, tracking quotas, and ensuring seamless workflow continuity.**
+
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
+[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)](https://www.python.org/downloads/)
+[![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![Linter: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+[![Maintenance Status](https://img.shields.io/badge/Maintenance-Active-success)](#)
+
+---
+
+## ⚡ Quick Start
+
+Get up and running in under 5 minutes.
+
+### Prerequisites
+- Python 3.10 or higher
+- `uv` (recommended for fast dependency management)
+
+### Install
+Clone the repository and install using `uv`:
+```bash
+git clone https://github.com/dhruv13x/codex-manager.git
+cd codex-manager
+uv pip install --system -e .
+```
+
+### Run
+Verify the installation by viewing the CLI help menu:
+```bash
+codex-manager --help
+# Or use the shorter alias:
+cm --help
+```
+
+### Demo
+Here's a quick 5-line workflow to backup your current account, view recommendations, and switch to a new one:
+```bash
+# 1. Take a live snapshot of your active Codex account state
+cm backup --cloud
+
+# 2. Check cooldown statuses for all accounts
+cm cooldown
+
+# 3. Get the smartest recommendation for the next account to use
+cm recommend
+
+# 4. Switch to a new account using the 'auth-only' method
+cm use --email new_user@example.com
+```
+
+> **[🖼️ Suggestion: Add an animated GIF here demonstrating the `cm use` command in action with the Rich UI]**
+
+---
+
+## ✨ Features
+
+**Core**
+- **Live Status Tracking:** Automatically parses live Codex `/status` output to capture account email, quota text, and weekly reset timestamps.
+- **Smart Recommendations:** Recommends the optimal account to use next based on calculated cooldowns and real-time metadata.
+- **Full State Recovery:** Backup and restore full Codex runtime states (`auth.json`, history, logs) via `*.tar.gz` archives and `*.metadata.json`.
+
+**Performance & Reliability**
+- **Offline & Fallback Mode:** Employs an emergency `--without-status-check` fallback that gracefully estimates cooldowns even when the live tracker is temporarily unavailable.
+- **Lightning Fast UI:** Employs the `rich` library for beautiful terminal output, tables, and status animations.
+
+**Security & Cloud**
+- **Cloud Synchronization:** First-class support for Backblaze B2 (and S3-compatible buckets) for remote backup metadata and archive storage.
+- **Safe Operations:** Every modifying command supports a `--dry-run` flag to safely simulate actions without touching your files.
+
+---
+
+## 🛠️ Configuration
+
+Codex Manager prioritizes configuration via environment variables and CLI arguments for flexibility.
+
+### Environment Variables
+
+| Name | Description | Default | Required |
+|---|---|---|---|
+| `CODEX_MANAGER_HOME` | The primary home directory for the manager config and backups. | `~/.codex-manager` | No |
+| `CODEX_HOME` | The target directory where Codex state resides. | `~/.codex` | No |
+
+*Note: You can also place a `config.json` inside your `CODEX_MANAGER_HOME` to persist configuration settings.*
+
+### Key CLI Arguments
+
+Most commands support these primary flags. Use `cm <command> --help` for a complete list.
+
+| Flag | Description |
+|---|---|
+| `--dry-run` | Safely preview the changes without modifying local or cloud state. |
+| `--cloud` | Enable Backblaze B2/S3 cloud capabilities for the command. |
+| `--email <email>` | Specify a target account email for use, restore, or listing. |
+| `--backup-dir <dir>` | Override the directory used for reading/writing backups. |
+| `--without-status-check`| Bypass live capture and calculate cooldowns statically (+7 days). |
+| `--auth-only` | During `use` or `backup`, target only identity/auth files instead of the full state. |
+
+---
+
+## 🏗️ Architecture
+
+### Directory Tree
+
+```
+~/.codex-manager/ (CODEX_MANAGER_HOME)
+├── backups/                # Local archives and metadata
+│   ├── backup_1.tar.gz
+│   └── backup_1.metadata.json
+├── config.json             # Persistent local configurations
+└── cooldown.json           # Registry caching overall account cooldowns
+```
+
+### High-Level Data Flow
+1. **Capture:** `cm backup` or `cm status` reads live text from a `tmux` session running Codex.
+2. **Process:** The CLI parses the text to build a state model (Quota, Cooldown, Email) and packages the `~/.codex` directory into an archive.
+3. **Store:** Archives and adjacent JSON metadata are saved locally and pushed to the cloud (if configured).
+4. **Evaluate:** When `cm recommend` or `cm cooldown` is invoked, local and cloud metadata are fetched, evaluated against real-time, and ranked to find the optimal active account.
+5. **Switch:** `cm use` restores the selected account's data into the `~/.codex` home, rotating your session seamlessly.
+
+---
+
+## 🐞 Troubleshooting
+
+### Common Issues
+
+| Error Message | Cause | Solution |
+|---|---|---|
+| `TokenExpiredError: TOKEN EXPIRED` | The active Codex session token has expired. | Re-authenticate in Codex manually, or run with `--without-status-check` to bypass. |
+| `Could not resolve Cloud (B2) credentials.` | Missing B2 credentials for cloud sync. | Pass `--b2-id` and `--b2-key` flags, or ensure your credentials are set up. |
+| `No backups found in Cloud for <email>.` | The requested account isn't backed up to the specified bucket. | Run `cm list-backups --cloud` to verify the email and backup availability. |
+
+### Debug Mode
+While the CLI does not have a single `--debug` flag, you can often reveal more information by viewing the full exception traces or by utilizing the built-in doctor command:
+```bash
+cm doctor
+```
+The `doctor` command verifies your dependencies, runtime directories, and validates the status parser setup.
+
+---
+
+## 🤝 Contributing
+
+We welcome contributions to Codex Manager! Please review our `CONTRIBUTING.md` (coming soon) before submitting PRs.
+
+### Dev Setup
+To set up your local development environment:
+```bash
+# 1. Install all development dependencies
+uv pip install --system -e .[dev]
+
+# 2. Run the tests (Ensure 90%+ coverage)
+python -m pytest tests --cov=src --cov-report=term-missing
+
+# 3. Format and lint the codebase
+uv run ruff check --fix src/ tests/
+uv run black src/ tests/
+```
+
+---
+
+## 🗺️ Roadmap
+
+- **Plugin Architecture:** Allow custom plugins to manage other CLI authentication tokens.
+- **Enhanced Cloud Coverage:** Add direct first-class integrations for Google Cloud Storage and Azure Blob.
+- **Automated Rotation Daemon:** A background worker to automatically rotate accounts when token limits are reached in real-time.

{codex_manager-3.0.0 → codex_manager-4.0.0}/pyproject.toml

@@ -1,8 +1,8 @@
 [project]
 name = "codex-manager"
-version = "3.0.0"
+version = "4.0.0"
 description = "Codex account snapshot manager"
-readme = "CODEX_MANAGER_SPEC.md"
+readme = "README.md"
 requires-python = ">=3.10"
 license = { text = "MIT" }
 authors = [

{codex_manager-3.0.0 → codex_manager-4.0.0}/src/codex_manager/__init__.py

@@ -1,3 +1,3 @@
 __all__ = ["__version__"]
 
-__version__ = "3.0.0"
+__version__ = "4.0.0"

{codex_manager-3.0.0 → codex_manager-4.0.0}/src/codex_manager/account_status.py

@@ -9,9 +9,8 @@ from typing import Any
 
 from .backup import read_status_text_from_args
 from .cloud import get_cloud_provider
-from .list_backups import list_backups, list_cloud_backups
+from .list_backups import list_cloud_backups
 from .registry import sync_registry_with_cloud, update_registry_entry
-from .status import parse_live_status_text
 from .ui import console
 from .utils import build_archive_name
 
@@ -24,6 +23,7 @@ def patch_metadata(
     args: Any = None,
     session_start_at: Any | None = None,
     is_expired: bool = False,
+    dry_run: bool = False,
 ) -> None:
     backup_dir = Path(args.backup_dir).expanduser() if args and hasattr(args, "backup_dir") else Path("~/.codex-manager/backups").expanduser()
     
@@ -78,10 +78,13 @@ def patch_metadata(
                     data["quota_percent_left"] = quota_percent_left
                 data["is_expired"] = is_expired
                 data["updated_at"] = datetime.now().astimezone().isoformat()
-                metadata_path.write_text(json.dumps(data, indent=2), encoding="utf-8")
-                console.print(
-                    f"Updated local metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]"
-                )
+                if not dry_run:
+                    metadata_path.write_text(json.dumps(data, indent=2), encoding="utf-8")
+                    console.print(
+                        f"Updated local metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]"
+                    )
+                else:
+                    console.print(f"Would update local metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]")
             except Exception as exc:
                 console.print(f"[yellow]Warning:[/] Failed to patch local metadata: {exc}")
         else:
@@ -113,10 +116,13 @@ def patch_metadata(
                 "metadata_only": True,
             }
             try:
-                metadata_path.write_text(json.dumps(data, indent=2), encoding="utf-8")
-                console.print(
-                    f"Created cooldown-only metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]"
-                )
+                if not dry_run:
+                    metadata_path.write_text(json.dumps(data, indent=2), encoding="utf-8")
+                    console.print(
+                        f"Created cooldown-only metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]"
+                    )
+                else:
+                    console.print(f"Would create cooldown-only metadata for [cyan]{email}[/]: [dim]{metadata_path.name}[/]")
             except Exception as exc:
                 console.print(f"[yellow]Warning:[/] Failed to create local metadata: {exc}")
 
@@ -127,12 +133,13 @@ def patch_metadata(
         quota_text=quota_text,
         quota_percent_left=quota_percent_left,
         session_start_at=final_session_start_at,
+        dry_run=dry_run,
     )
 
     if args and getattr(args, "cloud", False):
         cp = get_cloud_provider(args)
         if cp:
-            sync_registry_with_cloud(cp)
+            sync_registry_with_cloud(cp, dry_run=dry_run)
             entries = list_cloud_backups(cp, email=email, latest_per_email=True)
             if entries:
                 selected = entries[0]
@@ -161,11 +168,14 @@ def patch_metadata(
                         data["updated_at"] = datetime.now().astimezone().isoformat()
                         local_metadata.write_text(json.dumps(data, indent=2), encoding="utf-8")
 
-                        console.print(
-                            f"Uploading updated metadata to Cloud: [dim]{metadata_name}[/] ..."
-                        )
-                        cp.upload_file(local_metadata, metadata_name)
-                        console.print("Cloud metadata update complete.")
+                        if not dry_run:
+                            console.print(
+                                f"Uploading updated metadata to Cloud: [dim]{metadata_name}[/] ..."
+                            )
+                            cp.upload_file(local_metadata, metadata_name)
+                            console.print("Cloud metadata update complete.")
+                        else:
+                            console.print(f"Would upload updated metadata to Cloud: [dim]{metadata_name}[/]")
                     except Exception as exc:
                         console.print(f"[yellow]Warning:[/] Failed to patch cloud metadata: {exc}")
         else:
@@ -215,6 +225,7 @@ def sync_current_account_status(args: Any) -> None:
             quota_percent_left=None,
             args=args,
             session_start_at=session_start_at,
+            dry_run=getattr(args, "dry_run", False),
         )
         return
 
@@ -241,6 +252,7 @@ def sync_current_account_status(args: Any) -> None:
                     args=args,
                     session_start_at=None,
                     is_expired=True,
+                    dry_run=getattr(args, "dry_run", False),
                 )
             except Exception:
                 # If we can't even get the email, we use the one from auth.json
@@ -251,6 +263,7 @@ def sync_current_account_status(args: Any) -> None:
                     quota_percent_left=None,
                     args=args,
                     is_expired=True,
+                    dry_run=getattr(args, "dry_run", False),
                 )
             sys.exit(1)
         except Exception as exc:
@@ -268,6 +281,7 @@ def sync_current_account_status(args: Any) -> None:
 
     if text:
         try:
+            from .status import parse_live_status_text
             status = parse_live_status_text(
                 text,
                 reference_year=getattr(args, "reference_year", None),
@@ -280,6 +294,7 @@ def sync_current_account_status(args: Any) -> None:
                 args=args,
                 session_start_at=status.session_start_at,
                 is_expired=status.is_expired,
+                dry_run=getattr(args, "dry_run", False),
             )
         except Exception as exc:
             console.print(

{codex_manager-3.0.0 → codex_manager-4.0.0}/src/codex_manager/args.py

@@ -565,6 +565,11 @@ def get_parser() -> argparse.ArgumentParser:
         "file",
         help="Path to the profile archive (.tar.gz) to export to or import from.",
     )
+    profile_parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show what would happen without modifying metadata.",
+    )
 
     doctor_parser = subparsers.add_parser(
         "doctor",
@@ -689,6 +694,11 @@ def get_parser() -> argparse.ArgumentParser:
     use_parser.add_argument("--bucket", help="B2 Bucket Name")
     use_parser.add_argument("--b2-id", help="B2 Key ID")
     use_parser.add_argument("--b2-key", help="B2 App Key")
+    status_parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show what would happen without modifying metadata.",
+    )
 
     sync_parser = subparsers.add_parser(
         "sync",
@@ -727,4 +737,59 @@ def get_parser() -> argparse.ArgumentParser:
         help="Show what would happen without actually syncing.",
     )
 
+    purge_parser = subparsers.add_parser(
+        "purge",
+        help="Total Wipeout: Completely delete the Codex home directory (Auth, Identity, and all State).",
+    )
+    purge_parser.add_argument(
+        "--source-dir",
+        default=str(_get_default("codex_home", str(DEFAULT_CODEX_HOME))),
+        help="Codex home directory to purge.",
+    )
+    purge_parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show what would be removed without deleting anything.",
+    )
+    purge_parser.add_argument(
+        "--yes",
+        "-y",
+        action="store_true",
+        help="Confirm the purge without prompting.",
+    )
+
+    remove_parser = subparsers.add_parser(
+        "remove",
+        help="Account Cleanup: Delete all local (and optionally cloud) backups and registry entries for a specific email.",
+    )
+    remove_parser.add_argument(
+        "--email",
+        required=True,
+        help="The email address of the account to remove.",
+    )
+    remove_parser.add_argument(
+        "--backup-dir",
+        default=str(_get_default("backup_dir", str(DEFAULT_BACKUP_DIR))),
+        help="Directory containing backup archives and metadata.",
+    )
+    remove_parser.add_argument(
+        "--cloud",
+        action="store_true",
+        help="Also remove backups and registry entries from Cloud (B2).",
+    )
+    remove_parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show what would be removed without deleting anything.",
+    )
+    remove_parser.add_argument(
+        "--yes",
+        "-y",
+        action="store_true",
+        help="Confirm the removal without prompting.",
+    )
+    remove_parser.add_argument("--bucket", help="B2 Bucket Name")
+    remove_parser.add_argument("--b2-id", help="B2 Key ID")
+    remove_parser.add_argument("--b2-key", help="B2 App Key")
+
     return parser