csv_to_sql.py

"""
Convert a CSV file to a series of SQL INSERT statements.

ex.
> python csv_to_sql.py data.csv dbo.SomeTable > data.sql
"""

import sys

if len(sys.argv) != 3:
    print 'Provide the CSV file name and table name.'
    print 'ex. python csv_to_sql.py data.csv dbo.SomeTable'
    exit()

csv_file = sys.argv[1]
table_name = sys.argv[2]

with open(csv_file) as f:
    columns = f.readline().rstrip()
    
    print 'DELETE FROM %s;' % table_name
    print 'SET IDENTITY_INSERT %s ON' % table_name
    
    for line in f:
        print 'INSERT INTO %s (%s) VALUES (%s);' % (table_name, columns, line.strip())
    
    print 'SET IDENTITY_INSERT %s OFF' % table_name