ensure_config.py

"""Script to ensure a configuration file exists."""
import argparse
import asyncio
import os

import homeassistant.config as config_util
from homeassistant.core import HomeAssistant

# mypy: allow-untyped-calls, allow-untyped-defs


def run(args):
    """Handle ensure config commandline script."""
    parser = argparse.ArgumentParser(
        description=("Ensure a Home Assistant config exists, creates one if necessary.")
    )
    parser.add_argument(
        "-c",
        "--config",
        metavar="path_to_config_dir",
        default=config_util.get_default_config_dir(),
        help="Directory that contains the Home Assistant configuration",
    )
    parser.add_argument("--script", choices=["ensure_config"])

    args = parser.parse_args()

    config_dir = os.path.join(os.getcwd(), args.config)

    # Test if configuration directory exists
    if not os.path.isdir(config_dir):
        print("Creating directory", config_dir)
        os.makedirs(config_dir, exist_ok=True)

    config_path = asyncio.run(async_run(config_dir))
    print("Configuration file:", config_path)
    return 0


async def async_run(config_dir):
    """Make sure config exists."""
    hass = HomeAssistant()
    hass.config.config_dir = config_dir
    path = await config_util.async_ensure_config_exists(hass)
    await hass.async_stop(force=True)
    return path