cppFileGenerator.py

import os
import re

inFileText = "Replace this text with your variable names and types,\nuse returnType methodName() to create a blank method with a return type, \nand add #include <file> to add an include to the auto-generated header file.\nSemicolons (;) will be removed automatically\nExample:\n\n#include \"exampleInclude.h\"\nfloat x\nfloat y\nfloat z\nvoid doSomething()\nfloat doSomethingElse()\nvoid voidTestMethod(float testInput)\nfloat testMethod(float testInput)\nfloat testMethod(float testInput, int testInput2)\n\nThen run the generator with:\npython3 fileGenerator.py\nYou will be asked to input a class name\n\nPlease note that #include generation and method generation are only available when using the file generator"

def fixCase(key):
    key = re.sub('([a-zA-Z])', lambda x: x.groups()[0].upper(), key, 1)
    return key

#Create empty dictionary
varDict = {}
includeList = []
methodList = []

#Get user input for the class name
className = input("Please enter class name:\n")
#className = "testClass"
print("\n")

#Read input
inFile = open("input.txt","r")
lines = inFile.readlines()

if(os.path.exists((className + ".h"))):
    os.remove((className + ".h")) #Clear output file to get rid of any old output
outFile = open((className + ".h"), "w+")

#Loop to read all lines from the file
try:
    for line in lines:
        line = line.strip()
        print("Read line \"" + line + "\"")

        if(line.startswith("#include")):
            includeList.append(line)
        elif(line.endswith(")")):
            methodList.append(line)
        else:
            #Convert to name and type
            line = line.replace(";","")
            splitLine = line.split(" ")
            varName = splitLine[1]
            varType = splitLine[0]
            
            #Add variable name to dictionary
            varDict[varName] = varType
except:
    #Print error to user
    print("\n\n")
    print("*******************************************************")
    print(" Input file could not be read successfully,\n did you save input.txt before running the generator?")
    print("*******************************************************\n")
    
    #Delete the header file because there was an error
    if(os.path.exists((className + ".h"))):
        os.remove((className + ".h"))

    #Exit program
    exit()

keys = varDict.keys()
#Writing header file

inFile.close()

print("Finished reading variable list, writing to output file...")
print("Writing .h file")

#Will write the .h file first

#Setting up the header file
outFile.write("//Header file auto-generated using CPP-Getter-Setter-Generator\n")
outFile.write("#ifndef " + className.upper() + "_H\n")
outFile.write("#define " + className.upper() + "_H\n\n")
outFile.write("//Includes\n")

for i in includeList:
    outFile.write(i + "\n")

outFile.write("\n\nclass " + className + " {\n\tpublic:\n")

print("Writing Constructor and Destructor")

#Constructor and Destructor
outFile.write("\t\t//Constructor and Destructor\n")
outFile.write("\t\t" + className + "();\n")
outFile.write("\t\t~" + className + "();\n")
outFile.write("\n\t\t//Overloaded constructors\n")

#Setup for writing constructors
keysList = list(keys)
count = 0

#Write constructors
for key in keysList:
    count += 1
    outFile.write("\t\t" + className + "(")
    for i in range(count):
        outFile.write(varDict[keysList[i]] + " _" + keysList[i])
        if(i != count-1):
            outFile.write(", ")
    outFile.write(");\n")
outFile.write("\n")

print("Writing Getters and Setters")

#Getters
outFile.write("\t\t//Getters and Setters\n")
for key in keys:
    outFile.write("\t\t" + varDict[key] + " get" + fixCase(key) + "();\n")

outFile.write("\n")

#Setters
for key in keys:
    outFile.write("\t\tvoid set" + fixCase(key) + "( " + varDict[key] + " _" + key + " );\n")

outFile.write("\n")

#Other methods
outFile.write("\t\t//Other methods\n")
for i in methodList:
    outFile.write("\t\t" + i + ";\n")
print("Writing other methods to header file")    

print("Writing Variable Declarations")

#Private variables
outFile.write("\n\tprivate:\n")
outFile.write("\t\t//Variables\n")
#Variable declarations
for key in keys:
    outFile.write("\t\t" + varDict[key] + " " + key + ";\n")

outFile.write("\n")

#End of header file
outFile.write("};\n#endif")

#Tidy up
outFile.close()

print("Finished writing header file")
print("Writing source file")

#Set up for creating the source file
if(os.path.exists((className + ".cpp"))):
    os.remove((className + ".cpp")) #Clear output file to get rid of any old output
outFile = open((className + ".cpp"), "w+")

#Now to write the source file
outFile.write("//Source file auto-generated using CPP-Getter-Setter-Generator\n\n")
outFile.write("//Includes\n")
outFile.write("#include <" + className + ".h>\n")

print("Writing constructors and destructors")

outFile.write("\n//Constructor and Destructor\n")
outFile.write("" + className + "::" + className + "(){}\n")
outFile.write("" + className + "::~" + className + "(){}\n")

outFile.write("\n//Overloaded constructors\n")

#Setup for writing constructors
keysList = list(keys)
count = 0

#Write constructors
for key in keysList:
    count += 1
    outFile.write("" + className + "::" + className + "(")
    for i in range(count):
        outFile.write(varDict[keysList[i]] + " _" + keysList[i])
        if(i != count-1):
            outFile.write(", ")
    outFile.write("){\n")

    for i in range(count):
        outFile.write("\t" + 
        keysList[i] + " = _" + keysList[i])
        outFile.write(";\n")
    outFile.write("}\n")

print("Writing Getters and setters")

outFile.write("\n//Getters and Setters\n")
#Getters
for key in keys:
    outFile.write(varDict[key] + " " + className + "::get" + fixCase(key) + "(){ return " + key + "; }\n")

outFile.write("\n")

#Setters
for key in keys: 
    outFile.write("void " + className + "::set" + fixCase(key) + "( " + varDict[key] + " _" + key + " ){ " + key + " = _" + key + "; }\n")

#Write other methods
print("Writing other methods to source file")
outFile.write("\n//Other methods\n")
for i in methodList:
    splitStr = i.split(" ")
    outFile.write(splitStr[0] + " " + className + "::" + splitStr[1])
    for i in range(len(splitStr)):
        if(i < 2):
            continue
        outFile.write(" " + splitStr[i])
    outFile.write("{\n\t\n}\n")

#All done writing files, time to clean up
print("Done writing to source file, closing file...")
outFile.close()

print("Reformatting input file...")
os.remove("input.txt")
inFile = open("input.txt", "w+")

inFile.write(inFileText)
inFile.close()

print("Done!")