ares_install.py

#!/usr/bin/env python3

import os.path
import shutil
import stat

ARES_HOME = os.path.expanduser('~/.ares')
BASH_PROFILE_PATH = os.path.expanduser('~/.bash_profile')
PROFILE_PATH = os.path.expanduser('~/.profile')
BASHRC_PATH = os.path.expanduser('~/.bashrc')
ZSHRC_PATH = os.path.expanduser('~/.zshrc')


def main():
    if not os.path.exists(ARES_HOME):
        os.mkdir(ARES_HOME)
    ares_path = ARES_HOME + '/ares'
    shutil.copy('./ares.py', ares_path)
    st = os.stat(ares_path)
    os.chmod(ares_path, st.st_mode | stat.S_IEXEC)

    if os.path.exists(BASH_PROFILE_PATH):
        with open(BASH_PROFILE_PATH, mode='a') as bash_profile:
            bash_profile.write('\nexport PATH=${PATH}:/Users/$USER/.ares\n')

    if os.path.exists(PROFILE_PATH):
        with open(PROFILE_PATH, mode='a') as profile:
            profile.write('\nexport PATH=${PATH}:/Users/$USER/.ares\n')

    if os.path.exists(BASHRC_PATH):
        with open(BASHRC_PATH, mode='a') as bashrc:
            bashrc.write('\nexport PATH=${PATH}:/Users/$USER/.ares\n')

    if os.path.exists(ZSHRC_PATH):
        with open(ZSHRC_PATH, mode='a') as zshrc:
            zshrc.write('\nexport PATH=${PATH}:/Users/$USER/.ares\n')


if __name__ == '__main__':
    main()