import os
import json

dir_1x1 = os.path.join("flags", "1x1")
dir_4x3 = os.path.join("flags", "4x3")

files = []
for (dirpath, dirnames, filenames) in os.walk(dir_1x1):
    files.extend(filenames)
    break


def add_ids(directory):
    for f in files:
        if f.find(".svg") == -1:
            continue
        filename = os.path.join(directory, f)
        update = False
        flag_id = "flag-icons-%s" % (f.replace(".svg", ""))
        with open(filename, "r") as flag:
            lines = flag.readlines()
            if lines[0].find("id") == -1 and lines[0].find("viewBox") > 0:
                lines[0] = lines[0].replace("viewBox", 'id="%s" viewBox' % flag_id)
                update = True

        if update:
            print("Adding ID to", filename)
            with open(filename, "w") as flag:
                flag.writelines(lines)


add_ids(dir_1x1)
add_ids(dir_4x3)
