Update database.py

database.py

@@ -1,69 +1,16 @@
-import sqlite3
+import sqlalchemy
+from sqlalchemy import create_engine
 import pandas as pd
 
-def create_connection(db_file):
-    """Create a database connection to the SQLite database."""
-    conn = sqlite3.connect(db_file)
-    return conn
-def initialize_database(conn):
-    """Initialize the database with tables and constraints."""
-    with conn:
-        conn.execute('PRAGMA foreign_keys = ON;')
-        
-        conn.execute('''CREATE TABLE IF NOT EXISTS vendor_data (
-                            vendor_type TEXT NOT NULL,
-                            vendor_name TEXT PRIMARY KEY,
-                            GST_NO INTEGER NOT NULL,
-                            Contact_person_name TEXT NOT NULL,
-                            address TEXT NOT NULL,
-                            city TEXT NOT NULL,
-                            pan_no TEXT NOT NULL CHECK(length(pan_no) <=10)
-                        )''')
-        
-        conn.execute('''CREATE TABLE IF NOT EXISTS material_data (
-                            Part_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            vendor_name TEXT,
-                            part_no INTEGER NOT NULL,
-                            scf TEXT NOT NULL,
-                            process_type TEXT NOT NULL,
-                            part_od REAL NOT NULL,
-                            part_width INTEGER NOT NULL,
-                            part_inner_dimension REAL NOT NULL,
-                            material_specification TEXT NOT NULL,
-                            finish_wt REAL NOT NULL,
-                            green_drg_no INTEGER NOT NULL,
-                            FOREIGN KEY (vendor_name) REFERENCES vendor_data(vendor_name)
-                        )''')
-        
-        conn.execute('''CREATE TABLE IF NOT EXISTS rm_cost_data (
-                            Part_id INTEGER,
-                            vendor_name TEXT,
-                            grade_type TEXT NOT NULL,
-                            usd_cif REAL NOT NULL,
-                            rate REAL NOT NULL,
-                            final_landed_cost REAL,
-                            FOREIGN KEY (Part_id) REFERENCES material_data(Part_id),
-                            FOREIGN KEY (vendor_name) REFERENCES vendor_data(vendor_name)
-                        )''')
-        
-        conn.execute('''CREATE TABLE IF NOT EXISTS supplier_data (
-                            Part_id INTEGER,
-                            vendor_name TEXT,
-                            input_weight REAL,
-                            process_code TEXT,
-                            machining_time REAL,
-                            inspection_time REAL,
-                            process_cost REAL,
-                            machining_cost REAL,
-                            inspection_cost REAL,
-                            FOREIGN KEY (Part_id) REFERENCES material_data(Part_id),
-                            FOREIGN KEY (vendor_name) REFERENCES vendor_data(vendor_name)
-                        )''')
+def create_connection(db_name):
+    """Create a database connection."""
+    engine = create_engine(f'sqlite:///{db_name}.db')
+    return engine
 
-def insert_data(conn, table, data):
-    """Insert data into SQLite database."""
-    data.to_sql(table, conn, if_exists='append', index=False)
+def insert_data(engine, table_name, data_frame):
+    """Insert data into a specified table."""
+    data_frame.to_sql(table_name, con=engine, if_exists='append', index=False)
 
-def fetch_data(query, conn):
-    """Fetch data from SQLite database."""
-    return pd.read_sql_query(query, conn)
+def fetch_data(engine, query):
+    """Fetch data using a SQL query."""
+    return pd.read_sql(query, con=engine)