BMS_MLVDS_register_map_to_schedule.py

Go to the documentation of this file.

# BMS_MLVDS_register_map_to_token.py
# First row of the csv file must be header!
import sys, getopt
import csv

def main(argv):
    csvFile = 'BMS_MLVDS_register_map_oct28.csv'
    xmlFile = 'BmsSchedule.xml'
    try:
        opts, args = getopt.getopt(argv, "hi:o:",["input=", "output="])
    except getopt.GetoptError:
        print("BMS_MLVDS_register_map_to_schedule.py -i <inputfile> -o <outputfile>")
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print 'BMS_MLVDS_register_map_to_schedule.py -i <inputfile> -o <outputfile>'
            sys.exit()
        elif opt in ("-i", "--input"):
            csvFile = arg
        elif opt in ("-o", "--output"):
            xmlFile = arg

    print("Opening " + csvFile)
    data = csv.reader(open(csvFile))

    print("Writing to " + xmlFile)
    xmlData = open(xmlFile, 'w')

    xmlData.write('<?xml version="1.0" encoding="ISO-8859-1"?>' + "\n")
    xmlData.write('<ScheduleData>' + "\n")
    xmlData.write('    <Properties>' + "\n")
    xmlData.write('        The schedule for the Battery Management System' + "\n")
    xmlData.write('    </Properties>' + "\n")
    xmlData.write('    <Elements>' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            tags = row
            # replace spaces w/ underscores in tag names
            for i in range(len(tags)):
                tags[i] = tags[i].strip().replace(' ', '_')
            if("name" in tags):
                iName = tags.index("name")
                print("found name tag at index: " + str(iName))
            else:
                print("can not find name tag")
                break    
            if("data_type" in tags):
                iDataType = tags.index("data_type")
                print("found datatype tag at index: " + str(iDataType))
            else:
                print("can not find datatype tag")
                break
            if("ROBONET_type" in tags):
                iDirection = tags.index("ROBONET_type")
                print("found ROBONET_type tag at index: " + str(iDirection))
            else:
                print("can not find ROBONET_type tag")
                break    
            if("style" in tags):
                iStyle = tags.index("style")
                print("found style tag at index: " + str(iStyle))
            else:
                print("can not find style tag")
                break    
        else:
            if(row[iName].strip().replace(' ', '_')):
                if("status" in row[iDirection].lower()):
                    dirRW = "Read"
                elif("control" in row[iDirection].lower()):
                    dirRW = "Write"
                else:
                    rowNum +=1
                    continue
                xmlData.write('        <Element {0:30} {1:15} {2:30} {3:20} {4:13} {5:40} /> \n'.format('id="' + str(row[iName].strip().replace(' ', '_')) + '"', 
                                                                                                        'type="' + str(row[iDataType]) + '"', 
                                                                                                        'offset1="' + str(row[iName]) + '"', 
                                                                                                        'direction="' + dirRW + '"', 
                                                                                                        'group="Power"', 
                                                                                                        'style="' + str(row[iStyle]) + '"'))
        rowNum +=1
    xmlData.write('    </Elements>' + "\n")
    xmlData.write('</ScheduleData>')

    xmlData.close()

if __name__ == "__main__":
   main(sys.argv[1:])