1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
import logging
import os
from glob import glob
logger = logging.getLogger(__name__)
def make_rundir(path: str) -> str:
"""Create a numbered run directory under the specified path.
./runinfo <- specified path
|----000
|----001 <- Directories for each run
| ....
|----NNN
Args:
- path (str): String path to root of all rundirs
"""
try:
if not os.path.exists(path):
os.makedirs(path)
prev_rundirs = glob(os.path.join(path, "[0-9]*[0-9]"))
current_rundir = os.path.join(path, '000')
if prev_rundirs:
# Since we globbed on files named as 0-9
x = sorted([int(os.path.basename(x)) for x in prev_rundirs])[-1]
current_rundir = os.path.join(path, '{0:03}'.format(x + 1))
os.makedirs(current_rundir)
logger.debug("Parsl run initializing in rundir: {0}".format(current_rundir))
return os.path.abspath(current_rundir)
except Exception:
logger.exception("Failed to create run directory")
raise
|
