parametrizaciones.py

# -*- coding: utf-8 -*-
"""

"""

import os
import datetime
# import shlex
import subprocess
from namelists import editar_namelist_wrf


class ParametrizacionWRF(object):
    """ """

    def __init__(self, nombre):
        self.nombre = nombre
        self.carpeta = f"{os.getenv('TEMP_DIR')}/{nombre}"

    def generar_slurm_sh(self):
        # Se copia el template
        os.system(f"cp -a {os.getenv('TEMPLATES_DIR')}/slurm-wrf.sh "
                  f"{self.carpeta}/slurm-wrf.sh")

        os.system(f"cp -a {os.getenv('TEMPLATES_DIR')}/llama.sh "
                  f"{self.carpeta}/llama.sh")
        os.system(f"chmod +X {self.carpeta}/llama.sh")
        # Se reemplazan las variables en el runner
        fecha = datetime.datetime(int(os.getenv('Y')),
                                  int(os.getenv('M')),
                                  int(os.getenv('D')),
                                  int(os.getenv('H')))
        ruta_fecha_hora = fecha.strftime('%Y_%m/%d_%H')
        with open(self.carpeta + '/slurm-wrf.sh', 'r') as file:
            fldata = file.read()

        fldata = fldata.replace('{{NOMBRE}}', f"WRF-{self.nombre}")
        fldata = fldata.replace('{{WORKDIR}}', os.getenv('WRF_OPERATIVO'))
        fldata = fldata.replace('{{TEMP_DIR}}', os.getenv('TEMP_DIR'))
        fldata = fldata.replace('{{LOGS}}', f"{os.getenv('LOGS_DIR')}/"
                                            f"{ruta_fecha_hora}/"
                                            f"slurm-{self.nombre}-%j.out")
        fldata = fldata.replace('{{LOGS-ERR}}', f"{os.getenv('LOGS_DIR')}/"
                                                f"{ruta_fecha_hora}"
                                                f"/slurm-{self.nombre}-"
                                                f"%j.err")
        fldata = fldata.replace('{{PARAM}}', self.nombre)
        fldata = fldata.replace('{{WRFOUT}}', f"{os.getenv('WRFOUT_DIR')}/"
                                              f"{os.getenv('Y')}_"
                                              f"{os.getenv('M')}"
                                              f"/wrfout_{self.nombre}_d01_"
                                              f"{os.getenv('Y')}-"
                                              f"{os.getenv('M')}-"
                                              f"{os.getenv('D')}_"
                                              f"{os.getenv('H')}:00:00.nc")
        fldata = fldata.replace('{{OUTDIR}}', f"/home/wrf/datos-webwrf/img/"
                                              f"{os.getenv('REGION')}/"
                                              f"{self.nombre}/"
                                              f"{os.getenv('Y')}_"
                                              f"{os.getenv('M')}/"
                                              f"{os.getenv('D')}_"
                                              f"{os.getenv('H')}/")
        fldata = fldata.replace('{{OUTDIR_WM}}', f"/home/wrf/"
                                                 f"datos-webwrf/img/webmet/"
                                                 f"{self.nombre}/")
        with open(self.carpeta + '/slurm-wrf.sh', 'w') as file:
            file.write(fldata)

    def generar_slurm_sh_ingestor(self):
        # Se copia el template
        os.system(f"cp -a {os.getenv('TEMPLATES_DIR')}/slurm-wrf-ingestor.sh "
                  f"{self.carpeta}/slurm-wrf-ingestor.sh")

        # Se reemplazan las variables en el runner
        fecha = datetime.datetime(int(os.getenv('Y')),
                                  int(os.getenv('M')),
                                  int(os.getenv('D')),
                                  int(os.getenv('H')))
        ruta_fecha_hora = fecha.strftime('%Y_%m/%d_%H')
        with open(self.carpeta + '/slurm-wrf-ingestor.sh', 'r') as file:
            fldata = file.read()

        fldata = fldata.replace('{{NOMBRE}}', f"WRF-{self.nombre}")
        fldata = fldata.replace('{{TEMP_DIR}}', os.getenv('TEMP_DIR'))
        fldata = fldata.replace('{{WORKDIR}}', os.getenv('WRF_OPERATIVO'))
        fldata = fldata.replace('{{LOGS}}', f"{os.getenv('LOGS_DIR')}/"
                                            f"{ruta_fecha_hora}/"
                                            f"slurm-{self.nombre}-%j.out")
        fldata = fldata.replace('{{LOGS-ERR}}', f"{os.getenv('LOGS_DIR')}/"
                                                f"{ruta_fecha_hora}"
                                                f"/slurm-{self.nombre}-"
                                                f"%j.err")
        fldata = fldata.replace('{{PARAM}}', self.nombre)
        with open(self.carpeta + '/slurm-wrf-ingestor.sh', 'w') as file:
            file.write(fldata)

    def run_wrf_post(self):
        os.chdir(self.carpeta)

        self.generar_slurm_sh()
        self.generar_slurm_sh_ingestor()

        myoutput = open('output.dat', 'w+')
        subprocess.Popen(['sh', 'llama.sh'], stdout=myoutput,
                         stderr=myoutput, universal_newlines=True)

    def run_wrf(self):
        print('Se linkean los ejecutables de WRF')
        os.system(f"mkdir -p {self.carpeta}/WRF/real-log")
        os.system(f"ln -sf {os.getenv('WRF_BASE')}/WRF-4.2.1/test/em_real/* "
                  f"{self.carpeta}/WRF/")

        print('Se linkean las salidas de WPS')
        os.system(f"ln -sf {os.getenv('TEMP_DIR')}/WPS/met_em.* "
                  f"{self.carpeta}/WRF/")

        print(f"Se edita el namelist para {self.nombre}")
        editar_namelist_wrf(self.nombre)

        os.chdir(f"{self.carpeta}/WRF/")

        print('Se ejecuta el REAL')
        os.system(f"time prun ./real.exe > $LOGS_DIR/$Y'_'$M/$D'_'$H/real_"
                  f"{self.nombre}_`date +%Y%m%d%H%M`.log 2>&1")

        os.system(f"cp rsl* real-log/")
        print('Se ejecuta el WRF')
        os.system(f"time prun ./wrf.exe > $LOGS_DIR/$Y'_'$M/$D'_'$H/wrf_"
                  f"{self.nombre}_`date +%Y%m%d%H%M`.log 2>&1")

    def extraer_variables(self):
        pass

    def tabla_datos(self):
        pass

    def generar_graficos(self):
        pass