mime_type_gen.py

mime_type_gen.py

import json

def load_from_url():
    with open('db.json') as f:
        return json.load(f)

def mime_type_to_var(mime_type):
    return mime_type.upper().replace("+", "_").replace("/","_").replace("-","_").replace(".","_")

def print_lines(file, lines):
    for l in lines:
        file.write(l)
        file.write("\n")

def write_file(file, lines):
    with open(file, 'w') as f:
        for l in lines:
            f.write(l)
            f.write("\n")

def main():
    content_type_struct = []
    content_type_def = []
    ext_to_content_type_map = []

    mime_dict = load_from_url()
    for mime_type in mime_dict:
        var_mime_type = mime_type_to_var(mime_type)
        content_type_struct.append('const static std::string {};'.format(var_mime_type, mime_type))
        content_type_def.append('const std::string ContentType::{} = "{}";'.format(var_mime_type, mime_type))

        if "extensions" in mime_dict[mime_type]:
            for ext in mime_dict[mime_type]["extensions"]:
                ext_to_content_type_map.append('{"%s",  ContentType::%s},' % (ext,var_mime_type))
    
    write_file("content_type_struct.cpp", content_type_struct)
    write_file("content_type_def.cpp", content_type_def)
    write_file("ext_to_content_type_map.cpp", ext_to_content_type_map)


if __name__ == "__main__":
    main()