Update database.py

database.py

@@ -102,113 +102,114 @@ import sqlite3
 
 # Create and initialize the database
 def initialize_database():
-    with sqlite3.connect("college.db") as conn:
-        cursor = conn.cursor()
-
-        # Enable foreign key support
-        cursor.execute("PRAGMA foreign_keys = ON;")
-
-        # Create student table
-        cursor.execute("""
-        CREATE TABLE IF NOT EXISTS student (
-            student_id INTEGER PRIMARY KEY,
-            first_name TEXT,
-            last_name TEXT,
-            date_of_birth DATE,
-            email TEXT,
-            phone_number TEXT,
-            major TEXT,
-            year_of_enrollment INTEGER
-        );
-        """)
-
-        # Create employee table
-        cursor.execute("""
-        CREATE TABLE IF NOT EXISTS employee (
-            employee_id INTEGER PRIMARY KEY,
-            first_name TEXT,
-            last_name TEXT,
-            email TEXT,
-            department TEXT,
-            position TEXT,
-            salary REAL,
-            date_of_joining DATE
-        );
-        """)
-
-        # Create course table with foreign key
-        cursor.execute("""
-        CREATE TABLE IF NOT EXISTS course (
-            course_id INTEGER PRIMARY KEY,
-            course_name TEXT,
-            course_code TEXT,
-            instructor_id INTEGER,
-            department TEXT,
-            credits INTEGER,
-            semester TEXT,
-            FOREIGN KEY (instructor_id) REFERENCES employee(employee_id)
-        );
-        """)
-
-        # Clear existing data
-        cursor.execute("DELETE FROM student;")
-
-
-        # Insert 10 students
-        students = [
-            (1, "Alice", "Johnson", "2001-06-15", "alice@example.com", "1234567890", "Computer Science", 2019),
-            (2, "Bob", "Smith", "2000-04-22", "bob@example.com", "0987654321", "Mechanical Engineering", 2018),
-            (3, "Charlie", "Brown", "2002-01-12", "charlie@example.com", "1111111111", "Electrical Engineering", 2020),
-            (4, "Daisy", "Miller", "2001-11-05", "daisy@example.com", "2222222222", "Electronics", 2020),
-            (5, "Ethan", "Hunt", "1999-09-30", "ethan@example.com", "3333333333", "Mechanical Engineering", 2017),
-            (6, "Fiona", "Gray", "2003-03-10", "fiona@example.com", "4444444444", "Civil Engineering", 2021),
-            (7, "George", "King", "2000-08-20", "george@example.com", "5555555555", "Computer Science", 2018),
-            (8, "Hannah", "Scott", "2001-02-25", "hannah@example.com", "6666666666", "Information Technology", 2019),
-            (9, "Ian", "Clark", "2002-07-07", "ian@example.com", "7777777777", "Electronics", 2020),
-            (10, "Julia", "Anderson", "1998-12-18", "julia@example.com", "8888888888", "Civil Engineering", 2016),
-        ]
-        cursor.executemany("INSERT OR IGNORE INTO student VALUES (?, ?, ?, ?, ?, ?, ?, ?);", students)
-        cursor.execute("DELETE FROM employee;")
-
-        # Insert 10 employees
-        employees = [
-            (1, "Drake", "Miller", "drake@example.com", "CSE", "Professor", 75000.00, "2015-08-01"),
-            (2, "Laura", "White", "laura@example.com", "ECE", "Assistant Professor", 65000.00, "2018-07-10"),
-            (3, "Nina", "Gomes", "nina@example.com", "IT", "Lecturer", 55000.00, "2020-01-15"),
-            (4, "Oscar", "Fernandez", "oscar@example.com", "ME", "Professor", 80000.00, "2014-03-12"),
-            (5, "Priya", "Singh", "priya@example.com", "CSE", "Assistant Professor", 62000.00, "2019-06-01"),
-            (6, "Rahul", "Kumar", "rahul@example.com", "EEE", "Lecturer", 50000.00, "2021-09-20"),
-            (7, "Sana", "Shaikh", "sana@example.com", "CIV", "Lecturer", 52000.00, "2022-02-11"),
-            (8, "Tom", "Andrews", "tom@example.com", "ECE", "Associate Professor", 70000.00, "2016-12-09"),
-            (9, "Uma", "Joshi", "uma@example.com", "IT", "Professor", 78000.00, "2013-05-25"),
-            (10, "Vikram", "Reddy", "vikram@example.com", "ME", "Assistant Professor", 60000.00, "2017-11-03"),
-        ]
-        cursor.executemany("INSERT OR IGNORE INTO employee VALUES (?, ?, ?, ?, ?, ?, ?, ?);", employees)
-        cursor.execute("DELETE FROM course;")
-        # Insert 10 courses (linked to instructors from employee table)
-        courses = [
-            (1, "Database Systems", "CS301", 1, "CSE", 4, "Fall"),
-            (2, "Digital Electronics", "EC202", 2, "ECE", 3, "Spring"),
-            (3, "Operating Systems", "CS302", 1, "CSE", 4, "Spring"),
-            (4, "Thermodynamics", "ME101", 4, "ME", 3, "Fall"),
-            (5, "Signals and Systems", "EC301", 2, "ECE", 3, "Fall"),
-            (6, "Data Structures", "CS201", 5, "CSE", 4, "Spring"),
-            (7, "Electrical Machines", "EE101", 6, "EEE", 4, "Fall"),
-            (8, "Structural Analysis", "CIV301", 7, "CIV", 3, "Spring"),
-            (9, "Web Technologies", "IT401", 3, "IT", 3, "Fall"),
-            (10, "Fluid Mechanics", "ME203", 10, "ME", 4, "Spring"),
-        ]
-        cursor.executemany("INSERT OR IGNORE INTO course VALUES (?, ?, ?, ?, ?, ?, ?);", courses)
+    conn = sqlite3.connect("college.db")
+    cursor = conn.cursor()
+
+    # Create student table
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS student (
+        student_id INTEGER PRIMARY KEY,
+        first_name TEXT,
+        last_name TEXT,
+        date_of_birth DATE,
+        email TEXT,
+        phone_number TEXT,
+        major TEXT,
+        year_of_enrollment INTEGER
+    );
+    """)
+
+    # Create employee table
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS employee (
+        employee_id INTEGER PRIMARY KEY,
+        first_name TEXT,
+        last_name TEXT,
+        email TEXT,
+        department TEXT,
+        position TEXT,
+        salary REAL,
+        date_of_joining DATE
+    );
+    """)
+
+    # Create course table
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS course (
+        course_id INTEGER PRIMARY KEY,
+        course_name TEXT,
+        course_code TEXT,
+        instructor_id INTEGER,
+        department TEXT,
+        credits INTEGER,
+        semester TEXT,
+        FOREIGN KEY (instructor_id) REFERENCES employee(employee_id)
+    );
+    """)
+
+    # Clear old data (order matters to avoid foreign key errors)
+    cursor.execute("DELETE FROM course;")
+    cursor.execute("DELETE FROM student;")
+    cursor.execute("DELETE FROM employee;")
+
+    # Insert sample data into student table
+    students = [
+        (1, "Alice", "Johnson", "2001-06-15", "alice@example.com", "1234567890", "Computer Science", 2019),
+        (2, "Bob", "Smith", "2000-04-22", "bob@example.com", "0987654321", "Mechanical Engineering", 2018),
+        (3, "Charlie", "Brown", "2002-01-10", "charlie@example.com", "1112223333", "ECE", 2020),
+        (4, "Daisy", "Miller", "2001-12-12", "daisy@example.com", "2223334444", "Civil", 2019),
+        (5, "Ethan", "Lee", "2000-05-05", "ethan@example.com", "3334445555", "IT", 2018),
+        (6, "Fiona", "Clark", "2001-07-07", "fiona@example.com", "4445556666", "CSE", 2020),
+        (7, "George", "Adams", "2002-02-02", "george@example.com", "5556667777", "CSE", 2021),
+        (8, "Hannah", "Moore", "2001-11-11", "hannah@example.com", "6667778888", "AI", 2019),
+        (9, "Ivan", "Davis", "1999-09-09", "ivan@example.com", "7778889999", "ECE", 2017),
+        (10, "Julia", "King", "2002-03-03", "julia@example.com", "8889990000", "IT", 2020)
+    ]
+    cursor.executemany("INSERT INTO student VALUES (?, ?, ?, ?, ?, ?, ?, ?);", students)
+
+    # Insert sample data into employee table
+    employees = [
+        (1, "Drake", "Miller", "drake@example.com", "CSE", "Professor", 75000.00, "2015-08-01"),
+        (2, "Laura", "White", "laura@example.com", "ECE", "Assistant Professor", 65000.00, "2018-07-10"),
+        (3, "James", "Hall", "james@example.com", "IT", "Lecturer", 55000.00, "2019-01-15"),
+        (4, "Nancy", "Allen", "nancy@example.com", "Civil", "Professor", 80000.00, "2014-06-12"),
+        (5, "Oscar", "Wright", "oscar@example.com", "Mechanical", "Assistant Professor", 60000.00, "2016-10-01"),
+        (6, "Rachel", "Baker", "rachel@example.com", "AI", "Lecturer", 52000.00, "2020-03-05"),
+        (7, "Steve", "Lopez", "steve@example.com", "CSE", "Professor", 78000.00, "2012-08-20"),
+        (8, "Tina", "Hill", "tina@example.com", "ECE", "Lecturer", 53000.00, "2017-09-18"),
+        (9, "Uma", "Scott", "uma@example.com", "IT", "Professor", 72000.00, "2013-04-07"),
+        (10, "Victor", "Green", "victor@example.com", "CSE", "Assistant Professor", 60000.00, "2018-11-23")
+    ]
+    cursor.executemany("INSERT INTO employee VALUES (?, ?, ?, ?, ?, ?, ?, ?);", employees)
+
+    # Insert sample data into course table
+    courses = [
+        (1, "Database Systems", "CS301", 1, "CSE", 4, "Fall"),
+        (2, "Digital Electronics", "EC202", 2, "ECE", 3, "Spring"),
+        (3, "Operating Systems", "CS302", 1, "CSE", 4, "Fall"),
+        (4, "Data Structures", "CS201", 10, "CSE", 4, "Spring"),
+        (5, "Microprocessors", "EC303", 8, "ECE", 3, "Fall"),
+        (6, "Civil Engineering Basics", "CV101", 4, "Civil", 3, "Spring"),
+        (7, "Thermodynamics", "ME201", 5, "Mechanical", 4, "Fall"),
+        (8, "AI Fundamentals", "AI101", 6, "AI", 3, "Fall"),
+        (9, "Web Technologies", "IT301", 3, "IT", 4, "Spring"),
+        (10, "Software Engineering", "CS303", 7, "CSE", 4, "Fall")
+    ]
+    cursor.executemany("INSERT INTO course VALUES (?, ?, ?, ?, ?, ?, ?);", courses)
+
+    conn.commit()
+    conn.close()
 
 # Function to execute SQL query and return results
 def run_sql_query(sql_query):
     try:
-        with sqlite3.connect("college.db") as conn:
-            cursor = conn.cursor()
-            cursor.execute("PRAGMA foreign_keys = ON;")
-            cursor.execute(sql_query)
-            rows = cursor.fetchall()
-            column_names = [description[0] for description in cursor.description] if cursor.description else []
+        conn = sqlite3.connect("college.db")
+        cursor = conn.cursor()
+        cursor.execute(sql_query)
+        rows = cursor.fetchall()
+        column_names = [description[0] for description in cursor.description] if cursor.description else []
+        conn.commit()
+        conn.close()
 
         if not rows:
             return "No rows returned."
@@ -217,11 +218,12 @@ def run_sql_query(sql_query):
         for row in rows:
             result += "\t".join(str(cell) for cell in row) + "\n"
         return result.strip()
-
     except Exception as e:
         return f"Error executing query: {e}"
 
-# Initialize database when this script runs directly
+# Run only if this script is the main entry point
 if __name__ == "__main__":
     initialize_database()
+   
+