qbraid-core 0.1.0.dev3__tar.gz → 0.1.0.dev4__tar.gz

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qbraid-core
-Version: 0.1.0.dev3
+Version: 0.1.0.dev4
 Summary: Python library with core abstractions for software development in the qBraid ecosystem.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -80,6 +80,8 @@ Or generate your `~/.qbraid/qbraidrc` file via the qbraid-core Python interface:
 >>> from qbraid_core import QbraidSession
 >>> session = QbraidSession(api_key='API_KEY')
 >>> session.save_config()
+>>> session.get_available_services()
+['environments', 'quantum']
 ```
 
 Other credential configuration methods can be found [here](https://docs.qbraid.com/projects/cli/en/latest/tree/qbraid_configure.html).

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/README.md

@@ -41,6 +41,8 @@ Or generate your `~/.qbraid/qbraidrc` file via the qbraid-core Python interface:
 >>> from qbraid_core import QbraidSession
 >>> session = QbraidSession(api_key='API_KEY')
 >>> session.save_config()
+>>> session.get_available_services()
+['environments', 'quantum']
 ```
 
 Other credential configuration methods can be found [here](https://docs.qbraid.com/projects/cli/en/latest/tree/qbraid_configure.html).

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "qbraid-core"
-version = "0.1.0.dev3"
+version = "0.1.0.dev4"
 authors = [
     {name = "qBraid Development Team", email = "contact@qbraid.com"},
 ]

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/__init__.py

@@ -24,8 +24,6 @@ Functions
    :toctree: ../stubs/
 
    client
-   register_client
-   discover_services
    setup_default_session
 
 Exceptions
@@ -38,9 +36,10 @@ Exceptions
    AuthError
    ConfigError
    RequestsApiError
+   ResourceNotFoundError
+   UserNotFoundError
 
 """
-from ._warnings import _check_version, _disable_urllib3
 from .client import QbraidClient
 from .exceptions import (
     AuthError,
@@ -50,10 +49,15 @@ from .exceptions import (
     ResourceNotFoundError,
     UserNotFoundError,
 )
-from .registry import discover_services, register_client
 from .retry import PostForcelistRetry
 from .session import QbraidSession
 
+try:
+    # Injected in _version.py during the build process.
+    from ._version import __version__
+except ImportError:
+    __version__ = ""
+
 # Hold the default session in a global variable, but don't initialize it yet.
 _DEFAULT_SESSION = None
 
@@ -104,8 +108,21 @@ def client(*args, **kwargs):
     Returns:
         A service client instance.
     """
-    return _get_default_session().client(*args, **kwargs)
-
-
-_disable_urllib3()
-_check_version("qbraid-core")
+    session = _get_default_session()
+    _ = session.get_available_services()  # Ensure the session is initialized
+    return session.client(*args, **kwargs)
+
+
+__all__ = [
+    "QbraidClient",
+    "QbraidSession",
+    "PostForcelistRetry",
+    "client",
+    "setup_default_session",
+    "QbraidException",
+    "AuthError",
+    "ConfigError",
+    "RequestsApiError",
+    "ResourceNotFoundError",
+    "UserNotFoundError",
+]

qbraid-core-0.1.0.dev3/qbraid_core/_warnings.py → qbraid-core-0.1.0.dev4/qbraid_core/_compat.py

@@ -7,8 +7,6 @@ Module for emitting and disabling warnings at top level.
 """
 import warnings
 
-import urllib3
-
 from .exceptions import QbraidException
 from .system import get_latest_package_version, get_local_package_version
 
@@ -16,13 +14,16 @@ from .system import get_latest_package_version, get_local_package_version
 def _warn_new_version(local: str, latest: str) -> bool:
     """Returns True if you should warn user about updated package version,
     False otherwise."""
+    if not local or not latest:
+        return False
+
     installed_major, installed_minor = map(int, local.split(".")[:2])
     latest_major, latest_minor = map(int, latest.split(".")[:2])
 
     return (installed_major, installed_minor) < (latest_major, latest_minor)
 
 
-def _check_version(package: str) -> None:
+def check_version(package: str) -> None:
     """Emits UserWarning if updated package version exists."""
     try:
         latest_version = get_latest_package_version(package)
@@ -37,9 +38,3 @@ def _check_version(package: str) -> None:
             )
     except QbraidException:
         pass
-
-
-# coverage: ignore
-def _disable_urllib3() -> None:
-    """Disables urllib3 InsecureRequestWarning warning."""
-    urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/_version.py

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

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/client.py

@@ -10,6 +10,7 @@ import os
 import re
 from typing import Optional
 
+from ._compat import check_version
 from .exceptions import AuthError, RequestsApiError, ResourceNotFoundError, UserNotFoundError
 from .session import QbraidSession
 
@@ -19,6 +20,7 @@ class QbraidClient:
 
     def __init__(self, session: Optional[QbraidSession] = None):
         self.session = session or QbraidSession()
+        check_version("qbraid-core")
 
     @property
     def session(self) -> QbraidSession:

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/services/environments/create.py

@@ -15,10 +15,7 @@ from qbraid_core.system import replace_str
 
 from .state import update_install_status
 
-logging.basicConfig(
-    level=logging.ERROR,
-    format="%(asctime)s - %(levelname)s - %(message)s",
-)
+logger = logging.getLogger(__name__)
 
 
 def create_local_venv(slug_path: Union[str, Path], prompt: str) -> None:
@@ -43,5 +40,5 @@ def create_local_venv(slug_path: Union[str, Path], prompt: str) -> None:
 
         update_install_status(slug_path, 1, 1)
     except Exception as err:  # pylint: disable=broad-exception-caught
-        logging.error("Error creating virtual environment: %s", err)
+        logger.error("Error creating virtual environment: %s", err)
         update_install_status(slug_path, 1, 0, message=str(err))

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/services/environments/paths.py

@@ -18,10 +18,7 @@ from qbraid_core.system import is_valid_python
 
 from .validate import is_valid_slug
 
-logging.basicConfig(
-    level=logging.ERROR,
-    format="%(asctime)s - %(levelname)s - %(message)s",
-)
+logger = logging.getLogger(__name__)
 
 
 def get_default_envs_paths() -> List[Path]:
@@ -99,7 +96,7 @@ def which_python(slug: str) -> str:
         if is_valid_python(python_path):
             return str(python_path)
     except Exception as err:  # pylint: disable=broad-exception-caught
-        logging.error("Error determining Python path: %s", err)
+        logger.error("Error determining Python path: %s", err)
 
     return sys.executable
 

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/services/environments/state.py

@@ -13,10 +13,7 @@ from typing import Dict, Optional, Union
 
 from .paths import get_env_path
 
-logging.basicConfig(
-    level=logging.ERROR,
-    format="%(asctime)s - %(levelname)s - %(message)s",
-)
+logger = logging.getLogger(__name__)
 
 
 def install_status_codes(slug: str) -> Dict[str, Union[int, str]]:
@@ -28,7 +25,7 @@ def install_status_codes(slug: str) -> Dict[str, Union[int, str]]:
                 json_data = json.load(file)
                 return json_data.get("install", {})
         except (IOError, json.JSONDecodeError) as err:
-            logging.error("Error reading state.json: %s", err)
+            logger.error("Error reading state.json: %s", err)
             return {}
 
     def read_from_txt(file_path: Path) -> Dict[str, Union[int, str]]:
@@ -43,7 +40,7 @@ def install_status_codes(slug: str) -> Dict[str, Union[int, str]]:
                     elif key == "message":
                         data[key] = value.strip()
         except IOError as err:
-            logging.error("Error reading install_status.txt: %s", err)
+            logger.error("Error reading install_status.txt: %s", err)
         return data
 
     slug_path = get_env_path(slug)
@@ -83,7 +80,7 @@ def update_install_status(
                 file.seek(0)  # Reset file position to the beginning
             except json.JSONDecodeError as err:
                 # Keep default data if JSON is invalid
-                logging.error("Error opening state.json: %s", err)
+                logger.error("Error opening state.json: %s", err)
 
             # Update the data
             data["install"]["complete"] = complete

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/services/quantum/proxy.py

@@ -14,7 +14,6 @@ from qbraid_core.system import get_active_site_packages_path, get_venv_site_pack
 
 from .exceptions import QbraidException
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 SUPPORTED_QJOB_LIBS = ["braket"]

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/session.py

@@ -10,6 +10,7 @@ import logging
 import os
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
+import urllib3
 from requests import RequestException, Response, Session
 from requests.adapters import HTTPAdapter
 
@@ -20,13 +21,14 @@ from .retry import STATUS_FORCELIST, PostForcelistRetry
 if TYPE_CHECKING:
     import qbraid_core
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 DEFAULT_ENDPOINT_URL = "https://api.qbraid.com/api"
 DEFAULT_CONFIG_PATH = os.path.join(os.path.expanduser("~"), ".qbraid", "qbraidrc")
 DEFAULT_CONFIG_SECTION = "default"
 
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
 
 class QbraidSession(Session):  # pylint: disable=too-many-instance-attributes
     """Custom session with handling of request urls and authentication.
@@ -258,8 +260,8 @@ class QbraidSession(Session):  # pylint: disable=too-many-instance-attributes
         """
         client_class = client_registry.get(service_name)
         if not client_class:
-            raise ValueError(f"Service {service_name} not registered")
-        return client_class(sessin=self)
+            raise ValueError(f"Service '{service_name}' not registered")
+        return client_class(session=self)
 
     def _initialize_retry(
         self, retries_total: int, retries_connect: int, backoff_factor: float

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/system/executables.py

@@ -14,7 +14,6 @@ import sys
 from pathlib import Path
 from typing import Optional, Union
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core/system/packages.py

@@ -16,7 +16,6 @@ import requests
 from .exceptions import QbraidSystemError
 from .executables import get_python_version_from_cfg, get_python_version_from_exe
 
-logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qbraid-core
-Version: 0.1.0.dev3
+Version: 0.1.0.dev4
 Summary: Python library with core abstractions for software development in the qBraid ecosystem.
 Author-email: qBraid Development Team <contact@qbraid.com>
 License: Proprietary
@@ -80,6 +80,8 @@ Or generate your `~/.qbraid/qbraidrc` file via the qbraid-core Python interface:
 >>> from qbraid_core import QbraidSession
 >>> session = QbraidSession(api_key='API_KEY')
 >>> session.save_config()
+>>> session.get_available_services()
+['environments', 'quantum']
 ```
 
 Other credential configuration methods can be found [here](https://docs.qbraid.com/projects/cli/en/latest/tree/qbraid_configure.html).

{qbraid-core-0.1.0.dev3 → qbraid-core-0.1.0.dev4}/qbraid_core.egg-info/SOURCES.txt

@@ -27,8 +27,8 @@ docs/api/qbraid_core.rst
 docs/api/qbraid_core.services.rst
 docs/api/qbraid_core.system.rst
 qbraid_core/__init__.py
+qbraid_core/_compat.py
 qbraid_core/_version.py
-qbraid_core/_warnings.py
 qbraid_core/client.py
 qbraid_core/exceptions.py
 qbraid_core/registry.py
@@ -61,10 +61,10 @@ qbraid_core/system/threader.py
 tests/__init__.py
 tests/conftest.py
 tests/test_client.py
+tests/test_compat.py
 tests/test_environments.py
 tests/test_session.py
 tests/test_system.py
-tests/test_warnings.py
 tests/fixtures/__init__.py
 tests/fixtures/environments.py
 tools/verify_headers.py

qbraid-core-0.1.0.dev3/tests/test_warnings.py → qbraid-core-0.1.0.dev4/tests/test_compat.py

@@ -7,7 +7,7 @@ Unit tests related to qBraid core functionality and system configurations.
 """
 import pytest
 
-from qbraid_core._warnings import _warn_new_version
+from qbraid_core._compat import _warn_new_version
 
 check_version_data = [
     # local, API, warn

qbraid-core-0.1.0.dev4/tools/verify_headers.py

@@ -0,0 +1,182 @@
+# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+
+# pylint: skip-file
+
+"""
+Script to verify qBraid copyright file headers
+
+"""
+import os
+import sys
+from typing import List, Optional
+
+# ANSI escape codes
+BLUE = "\033[94m"
+BOLD = "\033[1m"
+RESET = "\033[0m"
+
+DEFAULT_HEADER = """# Copyright (c) 2024, qBraid Development Team
+# All rights reserved.
+"""
+
+DEFAULT_HEADER_GPL = """# Copyright (C) 2024 qBraid
+#
+# This file is part of the qBraid-SDK
+#
+# The qBraid-SDK is free software released under the GNU General Public License v3
+# or later. You can redistribute and/or modify it under the terms of the GPL v3.
+# See the LICENSE file in the project root or <https://www.gnu.org/licenses/gpl-3.0.html>.
+#
+# THERE IS NO WARRANTY for the qBraid-SDK, as per Section 15 of the GPL v3.
+"""
+
+
+def verify_headers(
+    src_paths: List[str],
+    header_type: str = "default",
+    skip_files: Optional[List[str]] = None,
+    fix: bool = False,
+) -> None:
+    if header_type == "default":
+        header = DEFAULT_HEADER
+    elif header_type == "gpl":
+        header = DEFAULT_HEADER_GPL
+    else:
+        raise ValueError("Invalid header type")
+
+    header_2023 = header.replace("2024", "2023")
+
+    skip_files = skip_files or []
+
+    failed_headers = []
+    fixed_headers = []
+
+    def should_skip(file_path: str, content: str) -> bool:
+        if file_path in skip_files:
+            return True
+
+        if os.path.basename(file_path) == "__init__.py":
+            return not content.strip()
+
+        skip_header_tag = "# qbraid: skip-header"
+        line_number = 0
+
+        for line in content.splitlines():
+            line_number += 1
+            if 5 <= line_number <= 30 and skip_header_tag in line:
+                return True
+            if line_number > 30:
+                break
+
+        return False
+
+    def replace_or_add_header(file_path: str, fix: bool = False) -> None:
+        with open(file_path, "r", encoding="ISO-8859-1") as f:
+            content = f.read()
+
+        if (
+            content.startswith(header)
+            or content.startswith(header_2023)
+            or should_skip(file_path, content)
+        ):
+            return
+
+        if not fix:
+            failed_headers.append(file_path)
+            return
+
+        # Add or replace the header
+        lines = content.splitlines()
+        first_non_comment_line_index = next(
+            (i for i, line in enumerate(lines) if not line.strip().startswith("#")), None
+        )
+
+        new_content_lines = [header.rstrip("\r\n")] + (
+            lines[first_non_comment_line_index:] if first_non_comment_line_index is not None else []
+        )
+        new_content = "\n".join(new_content_lines) + "\n"
+
+        with open(file_path, "w", encoding="ISO-8859-1") as f:
+            f.write(new_content)
+
+        fixed_headers.append(file_path)
+        return
+
+    def process_files_in_directory(directory: str, fix: bool = False) -> int:
+        count = 0
+        if not os.path.isdir(directory):
+            return count
+        for root, _, files in os.walk(directory):
+            for file in files:
+                if file.endswith(".py"):
+                    file_path = os.path.join(root, file)
+                    replace_or_add_header(file_path, fix)
+                    count += 1
+        return count
+
+    checked = 0
+    for item in src_paths:
+        if os.path.isdir(item):
+            checked += process_files_in_directory(item, fix)
+        elif os.path.isfile(item) and item.endswith(".py"):
+            replace_or_add_header(item, fix)
+            checked += 1
+        else:
+            failed_headers.append(item)
+            print(f"Directory not found: {item}")
+
+    if not fix:
+        if failed_headers:
+            for file in failed_headers:
+                print(f"failed {os.path.relpath(file)}")
+            num_failed = len(failed_headers)
+            s1, s2 = ("", "s") if num_failed == 1 else ("s", "")
+            sys.stderr.write(f"\n{num_failed} file header{s1} need{s2} updating.\n")
+        else:
+            print("All file header checks passed!")
+
+    else:
+        for file in fixed_headers:
+            print(f"fixed {os.path.relpath(file)}")
+        num_fixed = len(fixed_headers)
+        num_ok = checked - num_fixed
+        s_fixed = "" if num_fixed == 1 else "s"
+        s_ok = "" if num_ok == 1 else "s"
+        print("\nAll done! ✨ 🚀 ✨")
+        print(f"{num_fixed} header{s_fixed} fixed, {num_ok} header{s_ok} left unchanged.")
+
+
+def display_help() -> None:
+    help_message = """
+    Usage: python verify_headers.py SRC [OPTIONS] ...
+
+    This script checks for copyright headers at the specified path.
+    If no flags are passed, it will indicate which files would be
+    modified without actually making any changes.
+    
+    Options:
+    --help      Display this help message and exit.
+    --fix       Adds/modifies file headers as necessary.
+    """
+    print(help_message)
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    if "--help" in sys.argv:
+        display_help()
+
+    # Check if at least two arguments are provided and the first argument is not a flag
+    if len(sys.argv) < 2 or sys.argv[1].startswith("--"):
+        print("Usage: python verify_headers.py SRC [OPTIONS] ...")
+        sys.exit(1)
+
+    skip_files = ["qbraid_core/_version.py"]
+    script_directory = os.path.dirname(os.path.abspath(__file__))
+    project_directory = os.path.abspath(os.path.join(script_directory, ".."))
+    skip_files = [os.path.join(project_directory, file_path) for file_path in skip_files]
+
+    fix = "--fix" in sys.argv
+    files_and_dirs = [arg for arg in sys.argv[1:] if arg != "--fix"]
+    verify_headers(files_and_dirs, fix=fix, skip_files=skip_files)

qbraid-core-0.1.0.dev3/tools/verify_headers.py

@@ -1,173 +0,0 @@
-# Copyright (c) 2024, qBraid Development Team
-# All rights reserved.
-
-# pylint: skip-file
-
-"""
-Script to verify qBraid copyright file headers
-
-"""
-import os
-import sys
-
-# ANSI escape codes
-BLUE = "\033[94m"
-BOLD = "\033[1m"
-RESET = "\033[0m"
-
-header = """# Copyright (c) 2024, qBraid Development Team
-# All rights reserved.
-"""
-
-header_gpl = """# Copyright (C) 2024 qBraid
-#
-# This file is part of the qBraid-SDK
-#
-# The qBraid-SDK is free software released under the GNU General Public License v3
-# or later. You can redistribute and/or modify it under the terms of the GPL v3.
-# See the LICENSE file in the project root or <https://www.gnu.org/licenses/gpl-3.0.html>.
-#
-# THERE IS NO WARRANTY for the qBraid-SDK, as per Section 15 of the GPL v3.
-"""
-
-header_2023 = header.replace("2024", "2023")
-header_gpl_2023 = header_gpl.replace("2024", "2023")
-
-skip_files = ["qbraid_core/_version.py"]
-
-failed_headers = []
-fixed_headers = []
-
-
-def should_skip(file_path: str, content: str) -> bool:
-    if file_path in skip_files:
-        return True
-
-    if os.path.basename(file_path) == "__init__.py":
-        return not content.strip()
-
-    skip_header_tag = "# qbraid: skip-header"
-    line_number = 0
-
-    for line in content.splitlines():
-        line_number += 1
-        if 5 <= line_number <= 30 and skip_header_tag in line:
-            return True
-        if line_number > 30:
-            break
-
-    return False
-
-
-def replace_or_add_header(file_path: str, fix: bool = False) -> None:
-    with open(file_path, "r", encoding="ISO-8859-1") as f:
-        content = f.read()
-
-    if (
-        content.startswith(header)
-        or content.startswith(header_2023)
-        or content.startswith(header_gpl)
-        or content.startswith(header_gpl_2023)
-        or should_skip(file_path, content)
-    ):
-        return
-
-    if not fix:
-        failed_headers.append(file_path)
-        return
-
-    lines = content.splitlines()
-
-    comment_lines = []
-    first_skipped_line = None
-    for index, line in enumerate(lines):
-        if line.startswith("#"):
-            comment_lines.append(line)
-        else:
-            first_skipped_line = index
-            break
-
-    new_content_lines = [header.rstrip("\r\n")] + lines[first_skipped_line:]
-    new_content = "\n".join(new_content_lines) + "\n"
-
-    with open(file_path, "w", encoding="ISO-8859-1") as f:
-        f.write(new_content)
-
-    fixed_headers.append(file_path)
-
-
-def process_files_in_directory(directory: str, fix: bool = False) -> int:
-    count = 0
-    for root, _, files in os.walk(directory):
-        for file in files:
-            if file.endswith(".py"):
-                file_path = os.path.join(root, file)
-                replace_or_add_header(file_path, fix)
-                count += 1
-    return count
-
-
-def display_help() -> None:
-    help_message = """
-    Usage: python verify_headers.py SRC [OPTIONS] ...
-
-    This script checks for copyright headers at the specified path.
-    If no flags are passed, it will indicate which files would be
-    modified without actually making any changes.
-    
-    Options:
-    --help      Display this help message and exit.
-    --fix       Adds/modifies file headers as necessary.
-    """
-    print(help_message)
-    sys.exit(0)
-
-
-if __name__ == "__main__":
-    if "--help" in sys.argv:
-        display_help()
-
-    # Check if at least two arguments are provided and the first argument is not a flag
-    if len(sys.argv) < 2 or sys.argv[1].startswith("--"):
-        print("Usage: python verify_headers.py SRC [OPTIONS] ...")
-        sys.exit(1)
-
-    script_directory = os.path.dirname(os.path.abspath(__file__))
-    project_directory = os.path.abspath(os.path.join(script_directory, ".."))
-    skip_files = [os.path.join(project_directory, file_path) for file_path in skip_files]
-
-    fix = "--fix" in sys.argv
-    files_and_dirs = [arg for arg in sys.argv[1:] if arg != "--fix"]
-
-    checked = 0
-    for item in files_and_dirs:
-        full_path = os.path.join(project_directory, item)
-        if os.path.isdir(full_path):
-            checked += process_files_in_directory(full_path, fix)
-        elif os.path.isfile(full_path) and full_path.endswith(".py"):
-            replace_or_add_header(full_path, fix)
-            checked += 1
-        else:
-            failed_headers.append(full_path)
-            print(f"Directory not found: {full_path}")
-
-    if not fix:
-        if failed_headers:
-            for file in failed_headers:
-                print(f"failed {os.path.relpath(file)}")
-            num_failed = len(failed_headers)
-            s1, s2 = ("", "s") if num_failed == 1 else ("s", "")
-            sys.stderr.write(f"\n{num_failed} file header{s1} need{s2} updating.\n")
-            sys.exit(1)
-        else:
-            print("All file header checks passed!")
-
-    else:
-        for file in fixed_headers:
-            print(f"fixed {os.path.relpath(file)}")
-        num_fixed = len(fixed_headers)
-        num_ok = checked - num_fixed
-        s_fixed = "" if num_fixed == 1 else "s"
-        s_ok = "" if num_ok == 1 else "s"
-        print("\nAll done! ✨ 🚀 ✨")
-        print(f"{num_fixed} header{s_fixed} fixed, {num_ok} header{s_ok} left unchanged.")