Function main

docs/update_switcher.py:11–58

()

def main():
    # Define CLI arguments
    parser = argparse.ArgumentParser(description="Update switcher.json")
    parser.add_argument(
        "--version", "-v", required=True, type=str, help="The new version to add"
    )
    args = parser.parse_args()
    # drop the "v" prefix
    new_version_without_prefix = args.version[1:]

    # Setup path to switcher.json (relative to this script) and load it
    switcher_path = os.path.join(os.path.dirname(__file__), "_static", "switcher.json")
    with open(switcher_path) as f:
        switcher = json.load(f)

    # first we get the version number of the previous version
    for i, version in enumerate(switcher):
        if version.get("name", "").startswith("latest"):
            latest_index = i
            previous_version = version["version"]
            if previous_version == new_version_without_prefix:
                print(
                    f"Version {new_version_without_prefix} already is the latest version. Exiting."
                )
                return

            # now replace the name and version of this one with the new version
            switcher[i]["name"] = f"latest ({new_version_without_prefix})"
            switcher[i]["version"] = new_version_without_prefix
            break
    else:
        raise ValueError("'latest' version not found in switcher.json")

    # Add the previous version to the list of versions (we always insert it after latest)
    if any(version["version"] == previous_version for version in switcher):
        print(
            f"Previous version {previous_version} already exists in switcher.json. Not adding it again."
        )
    else:
        previous_version_entry = {
            "version": previous_version,
            "url": f"https://python-visualization.github.io/folium/v{previous_version}/",
        }
        switcher.insert(latest_index + 1, previous_version_entry)

    # Write the updated switcher.json
    with open(switcher_path, "w") as f:
        json.dump(switcher, f, indent=2)


if __name__ == "__main__":