def generate_h(struct: Struct):
header = Header()
header.set_header_guard(
"__{uname}_{proj}_DB_ENTITY_STRUCT_{name}_H".format(
name=struct.name.upper(),
proj=proj_name.upper(),
uname=os.getlogin()))
header.add_pragma("once")
header.add_local_include("db/orm/entity.h")
header.add_struct(struct)
return str(header)
def generate_entity_h(structs: List[Struct]):
entity_h = Header()
entity_h.set_header_guard("__{name}_{proj}_DB_ENTITY_H".format(
name=os.getlogin().upper(), proj=proj_name.upper()))
entity_h.add_pragma("once")
entity_h.add_global_include("string.h")
entity_h.add_global_include("stdlib.h")
entity_h.add_global_include("mysql/mysql.h")
entity_h.add_local_include("db/dbc.h")
for struct in structs:
entity_h.add_local_include(f"db/orm/{struct.name}.h")
return str(entity_h)
