initialise_site_runs.py

#!/usr/bin/env python

"""
Checkout CABLE repositories, build executables and generate a qsub script to
wrap cable benchmarking scripts when used on raijin.

We can't use the run_comparison script as raijin nodes have no internet
connection.

That's all folks.
"""

__author__ = "Martin De Kauwe"
__version__ = "1.0 (16.06.2020)"
__email__ = "mdekauwe@gmail.com"

import os
import sys
import datetime
import subprocess
from optparse import OptionParser

from user_options import *

sys.path.append("scripts")
from get_cable import GetCable
from build_cable import BuildCable
from generate_qsub_script import create_qsub_script


# i.e. if on NCI
if ("Mac" not in nodename and
    "MacBook" not in nodename and
    "imac" not in nodename and
    "unsw" not in nodename):

    create_qsub_script(qsub_fname, ncpus, mem, wall_time, project,
                       email_address)

parser = OptionParser()
parser.add_option("-s", "--skipbuild", action="store_true", default=False,
                  help="Rebuild src?")
parser.add_option("-g", "--skipget", action="store_true", default=False,
                  help="Get src?")

(options, args) = parser.parse_args()


if options.skipget == False:

    #
    ## Get CABLE ...
    #
    G = GetCable(src_dir=src_dir, user=user)
    G.main(repo_name=repos[0], trunk=trunk) # Default is True

    # Run on a users branch, not integration
    if repos[1] != "integration":
        get_user_branch = True
    else:
        get_user_branch = False

    if share_branch:
        get_user_branch = False

    G.main(repo_name=repos[1], trunk=False, user_branch=get_user_branch,
           share_branch=share_branch) # integration branch

if options.skipbuild == False:

    #
    ## Build CABLE ...
    #
    B = BuildCable(src_dir=src_dir, NCDIR=NCDIR, NCMOD=NCMOD, FC=FC,
                   CFLAGS=CFLAGS, LD=LD, LDFLAGS=LDFLAGS)
    B.main(repo_name=repos[0])

    if share_branch:
        #print(os.path.basename(repos[1]))
        B.main(repo_name=os.path.basename(repos[1]))
    else:
        B.main(repo_name=repos[1])