azdev/utilities/tools.py

# -----------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# -----------------------------------------------------------------------------

from knack.util import CLIError


def require_virtual_env():
    from azdev.utilities import get_env_path

    env = get_env_path()
    if not env:
        raise CLIError('This command can only be run from an active virtual environment.')


def require_azure_cli():
    try:
        import azure.cli.core  # pylint: disable=unused-import, unused-variable
    except ImportError:
        raise CLIError('CLI is not installed. Run `azdev setup`.')