[code.view]

[top] / python / PyMOTW / sqlite3 / sqlite3_load_csv.py

     #!/usr/bin/env python
     # encoding: utf-8
     #
     # Copyright (c) 2010 Doug Hellmann.  All rights reserved.
     #
     """Load bulk data from a CSV file.
     """
     #end_pymotw_header
     
     import csv
     import sqlite3
     import sys
     
     db_filename = 'todo.db'
     data_filename = sys.argv[1]
     
     SQL = """insert into task (details, priority, status, deadline, project)
              values (:details, :priority, 'active', :deadline, :project)
           """
     
     with open(data_filename, 'rt') as csv_file:
         csv_reader = csv.DictReader(csv_file)
         
         with sqlite3.connect(db_filename) as conn:
             cursor = conn.cursor()
             cursor.executemany(SQL, csv_reader)
     

[top] / python / PyMOTW / sqlite3 / sqlite3_load_csv.py

contact | logmethods.com