#!/usr/bin/env python3
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import json
import os
import sys
from pathlib import Path

env_name = sys.argv[1]
CONDA_DIR = os.environ["CONDA_DIR"]

file = Path.home() / f".local/share/jupyter/kernels/{env_name}/kernel.json"
content = json.loads(file.read_text())
content["env"] = {
    "XML_CATALOG_FILES": "",
    "PATH": f"{CONDA_DIR}/envs/{env_name}/bin:$PATH",
    "CONDA_PREFIX": f"{CONDA_DIR}/envs/{env_name}",
    "CONDA_PROMPT_MODIFIER": f"({env_name}) ",
    "CONDA_SHLVL": "2",
    "CONDA_DEFAULT_ENV": env_name,
    "CONDA_PREFIX_1": CONDA_DIR,
}

file.write_text(json.dumps(content, indent=1))