Simple CSV to XML Conversion - Python

I tried to generalize robertoia's function convert_row for any header instead of writing it by hand.

import csv  
import pandas as pd
            
f = open('movies2.csv')
csv_f = csv.reader(f)   
data = []

for row in csv_f: 
   data.append(row)
f.close()

df = pd.read_csv('movies2.csv')
header= list(df.columns)

def convert_row(row):
     str_row = """<%s>%s</%s> \n"""*(len(header)-1)
     str_row = """<%s>%s""" +"\n"+ str_row + """</%s>"""
     var_values = [list_of_elments[k] for k in range(1,len(header)) for list_of_elments in [header,row,header]]
     var_values = [header[0],row[0]]+var_values+[header[0]]
     var_values =tuple(var_values)
     return str_row % var_values

text ="""<collection shelf="New Arrivals">"""+"\n"+'\n'.join([convert_row(row) for row in data[1:]])+"\n" +"</collection >"
print(text)
with open('output.xml', 'w') as myfile: 
  myfile.write(text)

Of course with pandas now, it is simpler to just use to_xml() :

df= pd.read_csv('movies2.csv')
with open('outputf.xml', 'w') as myfile: 
  myfile.write(df.to_xml())

I found an easier way to insert variables into a string or block of text:

'''Twas brillig and the slithy {what}
Did gyre and gimble in the {where}
All {how} were the borogoves
And the {who} outgrabe.'''.format(what='toves',
                                  where='wabe',
                                  how='mimsy',
                                  who='momeraths')

Alternatively:

'''Twas brillig and the slithy {0}
Did gyre and gimble in the {1}
All {2} were the borogoves
And the {3} outgrabe.'''.format('toves',
                                'wabe',
                                'mimsy',
                                'momeraths')

(substitute name of incoming data variable for 'toves', 'wabe', 'mimsy', and 'momeraths')

Can be done without any libraries too simple script as below assumes the CSV file as cvs_to_xml_data.csv with single column value.

"""This script is to create xmlfile.txt file from a given csv_to_xml_data.csv file"""
row_list = []
with open('csv_to_xml_data.csv', 'r') as csvfile:
    row_list = csvfile.read().splitlines()

with open('xmlfile.txt', 'w') as txtfile:
    for row in row_list:
        txtfile.write(f"""<Student>
    <studentclass>8thstd</studentclass>
    <studentsubject>math</<studentsubject>
    <stuItem>{row}<stuItem>
    <stuItemSting>I am {row} and I like to be {row} because I am {row} 
</stuItemSting>
</Student>""")    

I would modify, simplify a bit xml ouput structure (if possible), and then you will be able to use this simple script. Just instead of movietitle="Piti-piti pa" you would have:

<movie>
<title>Piti-piti Pa</title>
...
</movie>

Variations of the prompt:

echo -e 'id,name,surname\n1,"Jane","Flame"\n2,"John","Ash"' | ./csv2xml.py > outputfile.xml
echo -e 'id,name,surname\n1,"Jane","Flame"\n2,"John","Ash"' > inputfile.csv
./csv2xml.py < inputfile.csv > outputfile.xml 

Example of outputfile.xml:

<document>
 <record>
  <id>1</id>
  <name>Jane</name>
  <surname>Flame</surname>
 </record>
 <record>
  <id>2</id>
  <name>John</name>
  <surname>Ash</surname>
 </record>
</document>

The program itself:

#!/usr/bin/python3
import csv
import fileinput

csv.register_dialect('custom',
                     delimiter=',',
                     doublequote=True,
                     escapechar=None,
                     quotechar='"',
                     quoting=csv.QUOTE_MINIMAL,
                     skipinitialspace=False)
first_line=True
data = csv.reader(fileinput.input(), dialect='custom')
print("<document>")
for record in data:
    if first_line==True:
        first_line=False
        name_of_fields=record
    else:
        print(" <record>")
        for i, field in enumerate(record):
            print("  <" + name_of_fields[i] + ">" + field + "</" + name_of_fields[i] + ">" )
        print(" </record>")
print("</document>")