phantom/submodules/phantom-robomimic/robomimic/scripts/setup_macros.py

"""
This script sets up a private macros file.

The private macros file (macros_private.py) is not tracked by git,
allowing user-specific settings that are not tracked by git.

This script checks if macros_private.py exists.
If applicable, it creates the private macros at robomimic/macros_private.py
"""

import os
import robomimic
import shutil

if __name__ == "__main__":
    base_path = robomimic.__path__[0]
    macros_path = os.path.join(base_path, "macros.py")
    macros_private_path = os.path.join(base_path, "macros_private.py")

    if not os.path.exists(macros_path):
        print("{} does not exist! Aborting...".format(macros_path))

    if os.path.exists(macros_private_path):
        ans = input("{} already exists! \noverwrite? (y/n)\n".format(macros_private_path))

        if ans == "y":
            print("REMOVING")
        else:
            exit()

    shutil.copyfile(macros_path, macros_private_path)
    print("copied {}\nto {}".format(macros_path, macros_private_path))