Pasted

import openpyxl
 
def main():
    numberOfNewAttributes = int(raw_input("How many new attributes are there? -> "))
    newAttributes = []
    for i in range(numberOfNewAttributes):
        newAttributes.append(raw_input("What is one of the new attributes? -> "))
    
    try:
        wb = openpyxl.load_workbook('survey_results.xlsx')
    except IOError:
        print "The survey_results.xlsx file does not exist yet. This tool should only be used when upgrading an existing file"
        raise SystemExit
        
    ws = wb.worksheets[0]
    startOn = ws.max_column - 3
    oldData = [[0 for i in range(4)] for j in range(ws.max_row)]
    
    for i in range(startOn, ws.max_column + 1):
        for j in range(1, ws.max_row + 1):
            oldData[j - 1][i - startOn] = ws.cell(row = j, column = i).value
    
    for i in range(startOn, startOn + numberOfNewAttributes):
        ws.cell(row = 1, column = startOn+i-startOn).value = newAttributes[i - startOn]
        
    for i in range(startOn, startOn + numberOfNewAttributes):
        for j in range(2, ws.max_row + 2):
            ws.cell(row = j, column = i).value = ""
        
    for i in range(startOn + numberOfNewAttributes, startOn + numberOfNewAttributes + 4):
        for j in range(1, ws.max_row - 2):
            print "i -> " + str(i - (startOn + numberOfNewAttributes))
            print "j -> " + str(j - 1)
            ws.cell(row = j, column = i).value = oldData[j - 1][i - (startOn + numberOfNewAttributes)]
        
    wb.save("survey_results1.xlsx")
    
if __name__ == "__main__":
    main()