diff --git a/dartwrf/obsseq_to_netcdf.py b/dartwrf/obsseq_to_netcdf.py index 40765952cc46edf231b0ce306fdf00ddacdd36f6..2b5ba45716f391f5e6bbccd78e096fd411702236 100644 --- a/dartwrf/obsseq_to_netcdf.py +++ b/dartwrf/obsseq_to_netcdf.py @@ -2,7 +2,7 @@ import os, sys, glob, warnings from config.cfg import exp from config.cluster import cluster -import run_obs_diag as rod +import dartwrf.run_obs_diag as rod def listdir_dirs(path): return [a for a in os.listdir(path) if os.path.isdir(os.path.join(path, a))] diff --git a/dartwrf/prepare_namelist.py b/dartwrf/prepare_namelist.py index 85bf245c95737f10aeb094f4ffd827bd3f3d5d08..d223aa3a6875d0f4743c1d45846968ed5399cbc2 100755 --- a/dartwrf/prepare_namelist.py +++ b/dartwrf/prepare_namelist.py @@ -1,12 +1,12 @@ """Create namelist.input files Usage: - prepare_namelist.py <begin> <end> <intv> [--radt=<minutes>] [--restart=<flag>] [--restart_interval=<minutes>] +prepare_namelist.py <begin> <end> <intv> [--radt=<minutes>] [--restart=<flag>] [--restart_interval=<minutes>] Options: - --radt=<minutes> Radiation interval [default: 5] - --restart=<flag> Restart flag (.true., .false.) [default: .false.] - --restart_interval=<minutes> Restart frequency [default: 720] +--radt=<minutes> Radiation interval [default: 5] +--restart=<flag> Restart flag (.true., .false.) [default: .false.] +--restart_interval=<minutes> Restart frequency [default: 720] """ import os, sys, shutil, warnings import datetime as dt @@ -18,13 +18,16 @@ from dartwrf.utils import sed_inplace, copy, symlink, mkdir def run(iens, begin, end, hist_interval=5, radt=5, archive=True, restart=False, restart_interval=720): - """Create namelist.input files + """Create a namelist.input file for each ensemble member Args: - archive (bool): if True, write to archivedir of experiment - if False, write to WRF run directory - restart (str): fortran bool whether to use wrfinput or wrfrst - restart_interval (int): output frequency of wrfrst (minutes) + archive (bool): if True, write to archivedir of experiment + if False, write to WRF run directory + restart (str): fortran bool whether to use wrfinput or wrfrst + restart_interval (int): output frequency of wrfrst (minutes) + + Returns + None """ rundir = cluster.wrf_rundir(iens) copy(cluster.namelist, rundir+'/namelist.input') @@ -73,6 +76,7 @@ def run(iens, begin, end, hist_interval=5, radt=5, archive=True, if __name__ == '__main__': + args = docopt(__doc__) begin = dt.datetime.strptime(args['<begin>'], '%Y-%m-%d_%H:%M') end = dt.datetime.strptime(args['<end>'], '%Y-%m-%d_%H:%M') diff --git a/dartwrf/prepare_wrfrundir.py b/dartwrf/prepare_wrfrundir.py index d38cedab19e661032c48f11987faa570c1a68dab..b1199cc7545df256a9831b8fda1f039e3dfbecd7 100755 --- a/dartwrf/prepare_wrfrundir.py +++ b/dartwrf/prepare_wrfrundir.py @@ -1,10 +1,18 @@ +"""Prepare WRF run directories, to use wrf.exe then + +Args: + init_time (str): YYYY-MM-DD_HH:MM + +Returns: + None +""" import os, sys, shutil import datetime as dt from config.cfg import exp from config.cluster import cluster from dartwrf.utils import symlink, copy, link_contents -import prepare_namelist +from dartwrf import prepare_namelist if __name__ == '__main__':