sitelist-convert.py

import yaml
import rich

sitelist = {}
sitelist['sites'] = []


with open('sitelists/sources/cohort-8-final.txt') as input:
    for line in input:
        line = line.strip()
        if len(line) == 0:
            continue
        line = line.split(',')
        site = {}
        site['src'] = line[0]
        site['src'] = site['src'][0:51]
        site['path'] = line[1][13:]
        site['dst'] = 'ucbprod-' + line[0][4:]
        site['dst'] = site['dst'][0:51]
        site['training'] = 'ucbtraining-' + line[0][4:]
        site['training'] = site['training'][0:51]
        # rich.print(site)
        found = False
        for s in sitelist['sites']:
            if s['src'] == site['src']:
                found = True
        if found != True:
            sitelist['sites'].append(site)

# rich.print(sitelist)
print(yaml.dump(sitelist))