_space_train simple codegen library functions

codegen.py

import json

def get_sheet(filename, sheetname):
    with open(filename) as f:
        item_database = json.load(f)

    item_db_sheets = item_database['sheets']
    sheets = [sh for sh in item_db_sheets if sh['name'] == sheetname]
    sheet = sheets[0]
    sheet_lines = sheet['lines']
    sheet_types = sheet['columns']
    return sheet_lines, sheet_types

def open_header(filename, database, sys_headers, local_headers):
    output_h = open(filename+'.hpp', 'w')

    output_h.write("#ifndef "+filename.upper()+"_HPP \n")
    output_h.write("#define "+filename.upper()+"_HPP \n")
    output_h.write("// generated by codegen.py from "+database+"\n\n")

    for h in sys_headers:
        output_h.write("#include <"+h+"> \n")
    for h in local_headers:
        output_h.write("#include \""+h+"\" \n")

    return output_h

def open_cpp(filename, database, sys_headers, local_headers):
    output_h = open(filename+'.cpp', 'w')

    output_h.write("// generated by codegen.py from "+database+"\n\n")

    for h in sys_headers:
        output_h.write("#include <"+h+"> \n")
    for h in local_headers:
        output_h.write("#include \""+h+"\" \n")

    return output_h