make_config.py

import os
import sys
import urllib.request
import time

cwd = os.getcwd()
sys.path.append(cwd)
NL = "\n"

def create_makefile():
    f = open(f"{cwd}/makefile", "w")
    f.write("# .PHONY defines parts of the makefile that ")
    f.write("are not dependant on any specific file\n")
    f.write(".PHONY = help setup test run clean environment\n\n")
    f.write("# Defines the default target that `make` will to try to make, ")
    f.write("or in the case of a phony target, execute the specified commands\n")
    f.write("# This target is executed whenever we just type `make`\n")
    f.write(".DEFAULT_GOAL = setup\n")

    f.write("# The @ makes sure that the command itself ")
    f.write("isnt echoed in the terminal\n")
    f.write("setup:\n")
    f.write(f"\t@mkdir {snrdir}/eig_files\n")
    f.write(f"\t@mkdir {snrdir}/data_files\n")
    f.write(f"\t@gfortran {cwd}/read_eigen.f90\n")
    f.write(f"\t@{cwd}/a.out\n")
    f.write(f"\t@rm {cwd}/a.out\n")
    f.write(f"\t@python {cwd}/mince_eig.py\n")
    f.write(f"\t@mv *.dat {snrdir}/data_files/.\n")
    f.write(f"\t@mv {dldir}/w_samarth.dat {datadir}/data/w_s/.\n")
    # f.write("# making necessary directory structure\n")
    # f.write("\t# cloning the GitHub repo to get eig_files and data_files to store in snrnmais\n")
    # f.write(f"\t@git clone https://github.com/srijaniiserprinceton/get-solar-eigs.git {outdir}\n")
    # f.write(f"\t@mv {dldir}/* {outdir}/.\n")
    # f.write(f"\t@mv {outdir}/*.dat {datadir}/data/w_s/.\n")
    # if not (cwd == datadir):
    #     f.write(f"\t@cp {cwd}/data/w_s/write_w.py {datadir}/data/w_s/.\n")
    #     f.write(f"\t@cp {cwd}/w.dat {datadir}/data/w.dat\n")
    # f.write(f"\t@make -C {outdir}/\n")
    # f.write(f"\t@mv {outdir}/eig_files {datadir}/snrnmais_files/.\n")
    # f.write(f"\t@mv {outdir}/data_files {datadir}/snrnmais_files/.\n")
    # f.write(f"\t@rm -rf {outdir}\n")
    # f.write(f"\t@make -C \n")
    f.write("\t# generating the w_s data from data/w_s/w_samarth.dat\n")
    f.write(f"\t@python {datadir}/write_w.py\n")
    f.write(f"\t@rm {datadir}/data/w_s/w_samarth.dat\n")
    f.write(f"\t@rm -rf {datadir}/dlfiles\n")
    f.write(f"\t@rm {snrdir}/data_files/eigU.dat\n")
    f.write(f"\t@rm {snrdir}/data_files/eigV.dat\n")
    f.write("\t@echo '---------------------------------------------------------------------'\n")
    f.write("\t@echo '       Eigenfrequency and Eigenfunciton SUCCESSFULLY GENERATED       '\n")
    f.write("\t@echo '---------------------------------------------------------------------'\n")
    f.write("help:\n")
    f.write("\t@echo '---------------------------------HELP--------------------------------'\n")
    f.write("\t@echo 'TO EXTRACT EIGENFREQUENCIES AND EIGENFUNCTIONS: python make_config.py'\n")
    f.write("\t@echo '-------------------------------------------------------------------'\n\n")
    f.write("clean:\n")
    f.write(f"\t@rm -rf {datadir}/snrnmais_files\n")
    f.write(f"\t@rm -rf {dldir}\n")
    # f.write(f"\t@rm -rf *.egg-info\n")
    return None


def create_dir(fulldirpath):
    if not os.path.isdir(fulldirpath):
        print(f"Creating directory {fulldirpath}")
        os.mkdir(fulldirpath)
    else:
        print(f"Directory {fulldirpath} exists")
    return None

if __name__ == "__main__":
    snrnmais_dir = input(f"Enter location of the SNRNMAIS files:" +
                    f"Default: {cwd} -- ") or f"{cwd}"
    scratch_dir = input(f"Enter location of output files or directory in scratch:" +
                    f"Default: {cwd} -- ") or f"{cwd}"

    # creating miscellaneous directories
    create_dir(f"{cwd}/logs")
    create_dir(f"{scratch_dir}/output_files")
    create_dir(f"{scratch_dir}/output_files/plots")

    # temporary directory to be deleted later
    # this is already created because of git cloning in the make file
    with open(f"{cwd}/.config", "w") as f:
        f.write(f"{cwd}{NL}")
        f.write(f"{scratch_dir}{NL}")
        f.write(f"{snrnmais_dir}")

    # create_makefile()
    # os.system("make")