From 75a386788b17748baa785e04ec23cbfa854a8309 Mon Sep 17 00:00:00 2001
From: lkugler <lukas.kugler@gmail.com>
Date: Wed, 15 Mar 2023 10:37:05 +0100
Subject: [PATCH] imports
---
dartwrf/obsseq_to_netcdf.py | 2 +-
dartwrf/prepare_namelist.py | 22 +++++++++++++---------
dartwrf/prepare_wrfrundir.py | 10 +++++++++-
3 files changed, 23 insertions(+), 11 deletions(-)
diff --git a/dartwrf/obsseq_to_netcdf.py b/dartwrf/obsseq_to_netcdf.py
index 4076595..2b5ba45 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 85bf245..d223aa3 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 d38ceda..b1199cc 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__':
--
GitLab