openfoam_project/openfoam/run/__main__.py

import sys
import shutil
import argparse
import logging
import configparser
from pprint import pp

from time import time
from datetime import timedelta
from datetime import datetime

from pathlib import Path
import subprocess as sp
import urllib.error

from .grafana import Silencer
from .stl import copy_stl
from .run import run


_t0 = time()
parser = argparse.ArgumentParser(
    description='Run the OpenFoam simulation'
)
parser.add_argument('-c', '--config', default='config.ini', type=Path,
    help='Configuration file (default config.ini)')
parser.add_argument('-l', '--log-level', default='INFO', type=str,
    help='Log level')

args = parser.parse_args()

config = configparser.ConfigParser()
config.read(args.config)

logging.basicConfig(
    filename=config.get('main', 'log_file', fallback=None),
    level=args.log_level
)
log = logging.getLogger('openfoam')

log.info('Starting program')


def step(command, alias=None):
    if alias == None: alias = command
    log.info(f'Running {alias}')
    code = run(command, case_dir, alias)
    if code != 0:
        log.error(f'{alias} failed')
        sys.exit(code)
    log.info(f'{alias} finished successfully')


if config.getboolean('grafana', 'pause', fallback=False):
    log.info('Silencing Grafana alert')
    grafana = Silencer(
        url=config.get('grafana', 'url'),
        apikey=config.get('grafana', 'apikey'),
    )
    grafana.silence_alert()

input_dir = Path(config.get('main', 'input_dir'), config.get('main', 'case'))\
    .expanduser()
work_dir = Path(config.get('main', 'work_dir')).expanduser()

if not work_dir.exists():
    log.error(f'Work directory ({work_dir}) not found')
    sys.exit(1)
    
case_dir = work_dir.joinpath(config.get('main', 'case'))
if case_dir.exists():
    log.info(f'Deleting case ({case_dir})')
    shutil.rmtree(case_dir)

log.info(f'Copying case ({input_dir} -> {case_dir})')
shutil.copytree(input_dir, case_dir)

if config.getboolean('stl', 'copy', fallback=False):
    stl_in = Path(config.get('stl', 'from')).expanduser()
    if not stl_in.exists():
        log.error(f'STL from directory does not exist ({stl_in})')
        sys.exit(1)

    log.info(f'Copying stl directory ({stl_in})')
    copy_stl(stl_in, case_dir)

if config.getboolean('blockMesh', 'enable', fallback=False):
    step(('blockMesh'))

if config.getboolean('snappyHexMesh', 'enable', fallback=False):
    step(('snappyHexMesh', '-overwrite'), 'snappyHexMesh')

log.info('Copying 0.org -> 0')
shutil.copytree(
    case_dir.joinpath('0.org'),
    case_dir.joinpath('0'),
)

if config.getboolean('setFields', 'enable', fallback=False):
    step(('setFields'))

if config.getboolean('olaFlow', 'enable', fallback=False):
    cmd = ('olaFlow')
    if config.getboolean('parallel', 'enable', fallback=False):
        cmd = (
            'mpirun',
            '-np', config.get('parallel', 'threads'),
            'olaFlow', '-parallel'
        )
        step(('decomposePar'))

    step(cmd, 'olaFlow')

    if config.getboolean('parallel', 'enable', fallback=False) \
            and config.getboolean('reconstructPar', 'enable', fallback=False):
        step(('reconstructPar'))

        log.info(f'Deleting processor directories')
        for proc_dir in case_dir.glob(r'processor*'):
            log.info(f'Deleting {proc_dir}')
            shutil.rmtree(proc_dir)

if config.getboolean('grafana', 'pause', fallback=False):
    log.info('Unsilencing Grafana alert')
    grafana.unsilence_alert()

_t1 = time()
log.info(f'Program ended successfully after {timedelta(seconds=_t1-_t0)}')