Source code for pycif.plugins.datastreams.meteos.chimere_meteo.fetch

import datetime
import os

import pandas as pd

from .....utils import path




def fetch(ref_dir, ref_file, input_interval, target_dir, tracer=None, **kwargs):
    list_period_dates = pd.date_range(
        input_interval[0], input_interval[1], freq=tracer.file_freq
    )
    list_dates = {}
    list_files = {}
    for dd in list_period_dates:
        file = dd.strftime("{}/{}".format(ref_dir, ref_file))
        if os.path.isfile(file):
            file_hours = pd.date_range(
                dd, dd + pd.to_timedelta(tracer.file_freq), freq="1h"
            )
            list_dates[dd] = [
                [hh, hh + datetime.timedelta(hours=1)] for hh in file_hours
            ]
            list_files[dd] = len(file_hours) * [file]

            # Fetching
            target_file = "{}/{}".format(target_dir, os.path.basename(file))
            path.link(file, target_file)

    return list_files, list_dates