install_tech.py

"""Symlink tech to klayout."""

import os
import pathlib
import shutil
import sys


def remove_path_or_dir(dest: pathlib.Path):
    """Remove a path or directory."""
    if dest.is_dir():
        os.unlink(dest)
    else:
        os.remove(dest)


def make_link(src, dest, overwrite: bool = True) -> None:
    """Make a symbolic link from src to dest."""
    dest = pathlib.Path(dest)
    if not src.exists():
        raise FileNotFoundError(f"{src} does not exist")

    if dest.exists() and not overwrite:
        print(f"{dest} already exists")
        return
    if dest.exists() or dest.is_symlink():
        print(f"removing {dest} already installed")
        remove_path_or_dir(dest)
    try:
        os.symlink(src, dest, target_is_directory=True)
    except OSError:
        shutil.copytree(src, dest)
    print("link made:")
    print(f"From: {src}")
    print(f"To:   {dest}")


if __name__ == "__main__":
    klayout_folder = "KLayout" if sys.platform == "win32" else ".klayout"
    cwd = pathlib.Path(__file__).resolve().parent
    home = pathlib.Path.home()
    dest_folder = home / klayout_folder / "tech"
    dest_folder.mkdir(exist_ok=True, parents=True)

    src = cwd / "cspdk" / "si220" / "klayout"
    dest = dest_folder / "cspdk_si220"
    make_link(src=src, dest=dest)

    src = cwd / "cspdk" / "sin300" / "klayout"
    dest = dest_folder / "cspdk_sin300"
    make_link(src=src, dest=dest)