import os
import pyart
import pathlib
from fire import Fire



def label_all_items(
    data_dir="/data/data_WF/NhaBe/2022",
    out_dir="/data/data_WF/NhaBe/2022"
):
    for subdir in os.listdir(data_dir):
        subdir_path = os.path.join(data_dir, subdir)
        for i in range (1,32):
            if i < 10:
                data_new_dir = subdir_path + '/0{}'.format(i)
                out_new_dir = subdir_path + '/0{}'.format(i)
            else:
                data_new_dir = subdir_path + '/{}'.format(i)
                out_new_dir = subdir_path + '/{}'.format(i)
            pathlist = pathlib.Path(data_new_dir).rglob('*.*')
            for path in pathlist:
                f = os.path.join(path)
                radar = pyart.io.read(f)
                t = radar.time['units'].split(" ")[2][:-4]
                outdir = pathlib.Path(out_new_dir)
                t = outdir/t
                os.rename(f, t)
                # print(t)
            # print(data_new_dir)
            
if __name__ == "__main__":
    Fire(label_all_items)