Quellcode

"""csv2sql.py

Transform CSV data into SQL statements.
"""

import os
import sys


def csv2sql(infile, outfile, table):
    """
    Read a file of comma-separated values and write them, wrapped in SQL
    INSERT statements, into a new file.
    """
    inf = open(infile, 'r')
    outf = open(outfile, 'w')
    for line in inf:
        line = line.strip()
        if line == '':
            continue
        fields = line.split(';')
        outf.write('INSERT INTO `%s` VALUES (`%s`);\n' % (table, '`, `'.join(fields)))
    inf.close()
    outf.close()


if __name__ == '__main__':
    if len(sys.argv) != 4:
        print 'Usage: %s <input (CSV) filename> <output (SQL) filename> <table name>' \
            % os.path.basename(sys.argv[0])
        sys.exit(2)
    csv2sql(sys.argv[1], sys.argv[2], sys.argv[3])

Quellcode

import csv

...

def csv2sql(infile, outfile, table):
    """
    Read a file of comma-separated values and write them, wrapped in SQL
    INSERT statements, into a new file.
    """
    outf = open(outfile, 'w')
    for row in csv.reader(open(infile, 'rb'), delimiter=';'):
        outf.write('INSERT INTO `%s` VALUES (`%s`);\n' % (table, '`, `'.join(row)))
    outf.close()

...

Quellcode

1;Peter;Müller
2;Hans;Wurst
3;Birne;Helene

Quellcode

python csv2sql.py test.csv test.sql mytable

Quellcode

INSERT INTO `mytable` VALUES (`1`, `Peter`, `Müller`);
INSERT INTO `mytable` VALUES (`2`, `Hans`, `Wurst`);
INSERT INTO `mytable` VALUES (`3`, `Birne`, `Helene`);