Code2pdf 6650bd42005b0
Code2pdf 6650bd42005b0
import sqlite3
def create_tables(cursor):
cursor.execute("""
CREATE TABLE IF NOT EXISTS shipping_data_0 (
origin_warehouse TEXT,
destination_store TEXT,
product TEXT,
on_time TEXT,
product_quantity INTEGER,
driver_identifier TEXT
)
""")
cursor.execute("""
CREATE TABLE IF NOT EXISTS shipping_data_1 (
shipment_identifier TEXT,
product TEXT,
on_time TEXT,
origin_warehouse TEXT,
destination_store TEXT
)
""")
def insert_shipping_data_0(cursor):
with open('data/shipping_data_0.csv', 'r') as file:
csv_reader = csv.reader(file)
next(csv_reader)
for row in csv_reader:
origin_warehouse, destination_store, product, on_time, product_quantity, driver_identifier = row
cursor.execute("INSERT INTO shipping_data_0 (origin_warehouse, destination_store, product, on_time, product_quantity, driver_identifier) VALUES (?, ?, ?, ?, ?, ?)",
(origin_warehouse, destination_store, product, on_time, product_quantity, driver_identifier))
def insert_shipping_data_2(cursor):
with open('data/shipping_data_2.csv', 'r') as file:
csv_reader = csv.reader(file)
next(csv_reader)
shipping_data_2_rows = [row for row in csv_reader]
if __name__ == "__main__":
conn = sqlite3.connect('shipment_database.db')
cursor = conn.cursor()
insert_shipping_data_0(cursor)
insert_shipping_data_2(cursor)
conn.commit()
conn.close()