-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathload_to_sqlite.py
More file actions
35 lines (29 loc) · 1.27 KB
/
load_to_sqlite.py
File metadata and controls
35 lines (29 loc) · 1.27 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import sqlite3
import csv
conn = sqlite3.connect("ecommerce.db")
cur = conn.cursor()
cur.execute("DROP TABLE IF EXISTS users")
cur.execute("DROP TABLE IF EXISTS products")
cur.execute("DROP TABLE IF EXISTS orders")
cur.execute("DROP TABLE IF EXISTS order_items")
cur.execute("DROP TABLE IF EXISTS reviews")
cur.execute("CREATE TABLE users (id INT, name TEXT, email TEXT, created_at TEXT)")
cur.execute("CREATE TABLE products (id INT, name TEXT, category TEXT, price REAL)")
cur.execute("CREATE TABLE orders (id INT, user_id INT, order_date TEXT, total_amount REAL)")
cur.execute("CREATE TABLE order_items (id INT, order_id INT, product_id INT, quantity INT, item_price REAL)")
cur.execute("CREATE TABLE reviews (id INT, user_id INT, product_id INT, rating INT, review_text TEXT, created_at TEXT)")
def load_csv(file, table):
with open(file) as f:
reader = csv.reader(f)
next(reader)
for row in reader:
placeholders = ",".join("?"*len(row))
cur.execute(f"INSERT INTO {table} VALUES ({placeholders})", row)
load_csv("users.csv","users")
load_csv("products.csv","products")
load_csv("orders.csv","orders")
load_csv("order_items.csv","order_items")
load_csv("reviews.csv","reviews")
conn.commit()
conn.close()
print("SQLite DB created successfully!")