first commit

DB.py

@@ -0,0 +1,52 @@
+import sqlite3
+
+
+class Database:
+    def __init__(self):
+        self._con = sqlite3.connect("shifts.db")
+        self._setup_tables() 
+        pass
+    
+
+    def _setup_tables(self):
+        cur = self._con.cursor()
+        cur.execute('''
+        CREATE TABLE IF NOT EXISTS shifts (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            shift_start DATETIME NOT NULL,
+            shift_end DATETIME NOT NULL,
+            department TEXT NOT NULL,
+            duration TEXT NOT NULL,
+            description TEXT,
+            fetched_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            
+            UNIQUE(shift_start, department)  -- No duplicates
+        );
+        ''')
+        self._con.commit()
+
+
+    def insert_shifts(self, shifts):
+        cur = self._con.cursor()
+        inserted = 0
+
+        for shift in shifts:
+            start = shift.get('start_datetime', 'N/A')
+            end = shift.get('end_datetime', 'N/A')
+
+            department = shift.get('department_name', 'N/A')
+            duration = shift.get('duration', 'N/A')
+            description = shift.get('shift_remark', shift.get('description', 'N/A'))
+            
+            cur.execute("""
+                INSERT OR IGNORE INTO shifts 
+                (shift_start, shift_end, department, duration, description)
+                VALUES (?, ?, ?, ?, ?)
+            """, (start, end, department, duration, description))
+            
+            if cur.rowcount > 0:
+                inserted += 1
+    
+        self._con.commit()
+        print(f"✅ Inserted {inserted}/{len(shifts)} new shifts")
+