arcade-mcp/examples/data/make_sqlite_db.py
2024-04-22 14:13:09 -07:00

38 lines
910 B
Python

import csv
import sqlite3
# Path to the CSV file
csv_file_path = './synthetic_people_data.csv'
# Connect to a SQLite database (will be created if it doesn't exist)
conn = sqlite3.connect('people.sqlite')
cur = conn.cursor()
# Create a table
cur.execute('''
CREATE TABLE IF NOT EXISTS people (
id INTEGER PRIMARY KEY,
Name TEXT,
Age INTEGER,
Location TEXT,
Occupation TEXT,
Email TEXT
)
''')
# Read data from the CSV file
with open(csv_file_path, 'r') as csvfile:
csvreader = csv.reader(csvfile)
next(csvreader) # Skip the header row
for row in csvreader:
# Insert each row into the database
cur.execute('''
INSERT INTO people (Name, Age, Location, Occupation, Email)
VALUES (?, ?, ?, ?, ?)
''', row)
# Commit changes and close the connection
conn.commit()
conn.close()
print('Data imported into SQLite database successfully.')