import yaml

inventory = {}
for spacecraft in ['a', 'b', 'c', 'd', 'e']:
    for dataset in ['efi', 'esa', 'esd', 'fbk', 'fft', 'fgm', 'fit', 'gmom', 'mom', 'scm', 'sst']:
        inventory[f'th{spacecraft}_{dataset}'] = {
            'url_pattern': f'http://cdpp.irap.omp.eu/themisdata/th{spacecraft}/l2/{dataset}/{{Y}}/th{spacecraft}_l2_{dataset}_{{Y}}{{M:02d}}{{D:02d}}_v\\d+.cdf',
            'use_file_list': True,
            'master_cdf': f"http://cdpp.irap.omp.eu/themisdata/th{spacecraft}/l2/{dataset}/0000/th{spacecraft}_l2_{dataset}_00000000_v01.cdf",
            'inventory_path': f'cdpp/THEMIS/TH{spacecraft.upper()}/L2',
            'split_rule': "regular",
            'split_frequency': "daily"
        }

with open('../speasy/data/archive/themis_cdpp.yaml', 'w') as inv_f:
    yaml.dump(inventory, inv_f)

inventory = {}

inventory['erg_lepe_l3_pa'] = {
            'url_pattern': 'https://cdaweb.gsfc.nasa.gov/pub/data/arase/lepe/l3/pa/{Y}/erg_lepe_l3_pa_{Y}{M:02d}{D:02d}_v\\d+_\\d+.cdf',
            'use_file_list': True,
            'master_cdf': "https://cdaweb.gsfc.nasa.gov/pub/software/cdawlib/0MASTERS/erg_lepe_l3_pa_00000000_v01.cdf",
            'inventory_path': 'cda/Arase_ERG/LEPE',
            'split_rule': "regular"
        }

inventory['erg_pwe_hfa_l3_1min'] = {
            'url_pattern': 'https://cdaweb.gsfc.nasa.gov/pub/data/arase/pwe/hfa/l3_1min/{Y}/erg_pwe_hfa_l3_1min_{Y}{M:02d}{D:02d}_v\\d+_\\d+.cdf',
            'use_file_list': True,
            'master_cdf': "https://cdaweb.gsfc.nasa.gov/pub/software/cdawlib/0MASTERS/erg_pwe_hfa_l3_1min_00000000_v01.cdf",
            'inventory_path': 'cda/Arase_ERG/PWE/HFA',
            'split_rule': "regular"
        }

for spacecraft in range(1,5):
    for mode,MODE in (('fast','FAST'), ('brst','BURST')):
        inventory[f'mms{spacecraft}_fpi_{mode}_l2_des_moms'] = {
            'url_pattern': f'https://cdaweb.gsfc.nasa.gov/pub/data/mms/mms{spacecraft}/fpi/{mode}/l2/des-moms/{{Y}}/{{M:02d}}/mms{spacecraft}_fpi_{mode}_l2_des-moms_{{Y}}{{M:02d}}\\d+_v\\d+.\\d+.\\d+.cdf',
            'use_file_list': True,
            'master_cdf': f"https://cdaweb.gsfc.nasa.gov/pub/software/cdawlib/0MASTERS/mms{spacecraft}_fpi_{mode}_l2_des-moms_00000000_v01.cdf",
            'inventory_path': f'cda/MMS/MMS{spacecraft}/FPI/{MODE}/MOMS',
            'split_rule': "random",
            'split_frequency': "monthly",
            'fname_regex': f'mms{spacecraft}_fpi_{mode}_l2_des-moms_(?P<start>\\d+)_v(?P<version>[\\d\\.]+)\\.cdf'
        }

    inventory[f'mms{spacecraft}_fgm_srvy_l2'] = {
        'url_pattern': f'https://cdaweb.gsfc.nasa.gov/pub/data/mms/mms{spacecraft}/fgm/srvy/l2/{{Y}}/{{M:02d}}/mms{spacecraft}_fgm_srvy_l2_{{Y}}{{M:02d}}{{D:02d}}_v\\d+.\\d+.\\d+.cdf',
        'use_file_list': True,
        'master_cdf': f"https://cdaweb.gsfc.nasa.gov/pub/software/cdawlib/0MASTERS/mms{spacecraft}_fgm_srvy_l2_00000000_v01.cdf",
        'inventory_path': f'cda/MMS/MMS{spacecraft}/FGM/SRVY',
        'split_rule': "regular"
    }


with open('../speasy/data/archive/cda.yaml', 'w') as inv_f:
    yaml.dump(inventory, inv_f)