21671d6bee
- Add Modules and UserModules database tables - Create home page with module selection grid - Implement per-user module assignment in user management - Add route guards for module access control - Refactor navigation: login -> home -> modules, admin console via button - Add Font Awesome icons
252 lines
9.4 KiB
Python
252 lines
9.4 KiB
Python
"""
|
||
ScanLook Database Initialization
|
||
Creates all tables and indexes for the inventory management system
|
||
UPDATED: Reflects post-migration schema (CURRENT baseline is now global)
|
||
"""
|
||
|
||
import sqlite3
|
||
import os
|
||
from datetime import datetime
|
||
from werkzeug.security import generate_password_hash
|
||
|
||
DB_PATH = os.path.join(os.path.dirname(__file__), 'scanlook.db')
|
||
|
||
|
||
def init_database():
|
||
"""Initialize the database with all tables and indexes"""
|
||
conn = sqlite3.connect(DB_PATH)
|
||
cursor = conn.cursor()
|
||
|
||
# Users Table
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS Users (
|
||
user_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
username TEXT UNIQUE NOT NULL,
|
||
password TEXT NOT NULL,
|
||
full_name TEXT NOT NULL,
|
||
email TEXT,
|
||
role TEXT NOT NULL CHECK(role IN ('owner', 'admin', 'staff')),
|
||
is_active INTEGER DEFAULT 1,
|
||
branch TEXT DEFAULT 'Main',
|
||
created_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
''')
|
||
|
||
# Modules Table - defines available system modules
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS Modules (
|
||
module_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
module_name TEXT UNIQUE NOT NULL,
|
||
module_key TEXT UNIQUE NOT NULL,
|
||
description TEXT,
|
||
icon TEXT,
|
||
is_active INTEGER DEFAULT 1,
|
||
display_order INTEGER DEFAULT 0
|
||
)
|
||
''')
|
||
|
||
# UserModules Table - which modules each user can access
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS UserModules (
|
||
user_module_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
user_id INTEGER NOT NULL,
|
||
module_id INTEGER NOT NULL,
|
||
granted_by INTEGER,
|
||
granted_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES Users(user_id),
|
||
FOREIGN KEY (module_id) REFERENCES Modules(module_id),
|
||
FOREIGN KEY (granted_by) REFERENCES Users(user_id),
|
||
UNIQUE(user_id, module_id)
|
||
)
|
||
''')
|
||
|
||
|
||
# CountSessions Table
|
||
# NOTE: current_baseline_version removed - CURRENT is now global
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS CountSessions (
|
||
session_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
session_name TEXT NOT NULL,
|
||
session_type TEXT NOT NULL CHECK(session_type IN ('cycle_count', 'full_physical')),
|
||
created_by INTEGER NOT NULL,
|
||
created_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
master_baseline_timestamp DATETIME,
|
||
current_baseline_timestamp DATETIME,
|
||
status TEXT DEFAULT 'active' CHECK(status IN ('active', 'completed', 'archived')),
|
||
branch TEXT DEFAULT 'Main',
|
||
FOREIGN KEY (created_by) REFERENCES Users(user_id)
|
||
)
|
||
''')
|
||
|
||
# BaselineInventory_Master Table (Session-specific, immutable)
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS BaselineInventory_Master (
|
||
baseline_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
session_id INTEGER NOT NULL,
|
||
lot_number TEXT NOT NULL,
|
||
item TEXT NOT NULL,
|
||
description TEXT,
|
||
system_location TEXT NOT NULL,
|
||
system_bin TEXT NOT NULL,
|
||
system_quantity REAL NOT NULL,
|
||
uploaded_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (session_id) REFERENCES CountSessions(session_id)
|
||
)
|
||
''')
|
||
|
||
# BaselineInventory_Current Table (GLOBAL - shared across all sessions)
|
||
# MIGRATION CHANGE: No session_id, no baseline_version, no is_deleted
|
||
# This table is replaced entirely on each upload
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS BaselineInventory_Current (
|
||
current_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
lot_number TEXT NOT NULL,
|
||
item TEXT NOT NULL,
|
||
description TEXT,
|
||
system_location TEXT,
|
||
system_bin TEXT NOT NULL,
|
||
system_quantity REAL NOT NULL,
|
||
upload_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
UNIQUE(lot_number, system_bin)
|
||
)
|
||
''')
|
||
|
||
# LocationCounts Table
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS LocationCounts (
|
||
location_count_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
session_id INTEGER NOT NULL,
|
||
location_name TEXT NOT NULL,
|
||
counted_by INTEGER NOT NULL,
|
||
start_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
end_timestamp DATETIME,
|
||
status TEXT DEFAULT 'not_started' CHECK(status IN ('not_started', 'in_progress', 'completed')),
|
||
expected_lots_master INTEGER DEFAULT 0,
|
||
lots_found INTEGER DEFAULT 0,
|
||
lots_missing INTEGER DEFAULT 0,
|
||
FOREIGN KEY (session_id) REFERENCES CountSessions(session_id),
|
||
FOREIGN KEY (counted_by) REFERENCES Users(user_id)
|
||
)
|
||
''')
|
||
|
||
# ScanEntries Table
|
||
# MIGRATION CHANGE: Removed current_* columns - now fetched via JOIN
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS ScanEntries (
|
||
entry_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
session_id INTEGER NOT NULL,
|
||
location_count_id INTEGER NOT NULL,
|
||
lot_number TEXT NOT NULL,
|
||
item TEXT,
|
||
description TEXT,
|
||
scanned_location TEXT NOT NULL,
|
||
actual_weight REAL NOT NULL,
|
||
scanned_by INTEGER NOT NULL,
|
||
scan_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
|
||
-- MASTER baseline comparison (immutable, set at scan time)
|
||
master_status TEXT CHECK(master_status IN ('match', 'wrong_location', 'ghost_lot', 'missing')),
|
||
master_expected_location TEXT,
|
||
master_expected_weight REAL,
|
||
master_variance_lbs REAL,
|
||
master_variance_pct REAL,
|
||
|
||
-- Duplicate detection
|
||
duplicate_status TEXT DEFAULT '00' CHECK(duplicate_status IN ('00', '01', '03', '04')),
|
||
duplicate_info TEXT,
|
||
|
||
-- CURRENT baseline comparison removed - now via JOIN in queries
|
||
-- Removed: current_status, current_system_location, current_system_weight,
|
||
-- current_variance_lbs, current_variance_pct, current_baseline_version
|
||
|
||
-- Metadata
|
||
comment TEXT,
|
||
is_deleted INTEGER DEFAULT 0,
|
||
deleted_by INTEGER,
|
||
deleted_timestamp DATETIME,
|
||
modified_timestamp DATETIME,
|
||
|
||
FOREIGN KEY (session_id) REFERENCES CountSessions(session_id),
|
||
FOREIGN KEY (location_count_id) REFERENCES LocationCounts(location_count_id),
|
||
FOREIGN KEY (scanned_by) REFERENCES Users(user_id),
|
||
FOREIGN KEY (deleted_by) REFERENCES Users(user_id)
|
||
)
|
||
''')
|
||
|
||
# MissingLots Table
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS MissingLots (
|
||
missing_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
session_id INTEGER NOT NULL,
|
||
location_count_id INTEGER,
|
||
lot_number TEXT NOT NULL,
|
||
item TEXT,
|
||
master_expected_location TEXT NOT NULL,
|
||
master_expected_quantity REAL NOT NULL,
|
||
current_system_location TEXT,
|
||
current_system_quantity REAL,
|
||
marked_by INTEGER NOT NULL,
|
||
marked_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
found_later TEXT DEFAULT 'N' CHECK(found_later IN ('Y', 'N')),
|
||
found_location TEXT,
|
||
FOREIGN KEY (session_id) REFERENCES CountSessions(session_id),
|
||
FOREIGN KEY (location_count_id) REFERENCES LocationCounts(location_count_id),
|
||
FOREIGN KEY (marked_by) REFERENCES Users(user_id)
|
||
)
|
||
''')
|
||
|
||
# Create Indexes
|
||
# MASTER baseline indexes
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_baseline_master_lot ON BaselineInventory_Master(session_id, lot_number)')
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_baseline_master_loc ON BaselineInventory_Master(session_id, system_location)')
|
||
|
||
# ScanEntries indexes
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_scanentries_session ON ScanEntries(session_id)')
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_scanentries_location ON ScanEntries(location_count_id)')
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_scanentries_lot ON ScanEntries(lot_number)')
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_scanentries_deleted ON ScanEntries(is_deleted)')
|
||
|
||
# LocationCounts indexes
|
||
cursor.execute('CREATE INDEX IF NOT EXISTS idx_location_counts ON LocationCounts(session_id, status)')
|
||
|
||
# Note: No indexes on BaselineInventory_Current needed - UNIQUE constraint handles lookups
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
print(f"✅ Database initialized at: {DB_PATH}")
|
||
print("📝 Schema version: Post-migration (CURRENT baseline is global)")
|
||
|
||
|
||
def create_default_users():
|
||
"""Create default users for testing"""
|
||
conn = sqlite3.connect(DB_PATH)
|
||
cursor = conn.cursor()
|
||
|
||
default_users = [
|
||
('owner', generate_password_hash('owner123'), 'System Owner', 'owner'),
|
||
('admin', generate_password_hash('admin123'), 'Admin User', 'admin'),
|
||
('staff1', generate_password_hash('staff123'), 'John Doe', 'staff'),
|
||
('staff2', generate_password_hash('staff123'), 'Jane Smith', 'staff'),
|
||
]
|
||
|
||
try:
|
||
cursor.executemany('''
|
||
INSERT INTO Users (username, password, full_name, role)
|
||
VALUES (?, ?, ?, ?)
|
||
''', default_users)
|
||
conn.commit()
|
||
print("✅ Default users created:")
|
||
print(" Owner: owner / owner123")
|
||
print(" Admin: admin / admin123")
|
||
print(" Staff: staff1 / staff123")
|
||
print(" Staff: staff2 / staff123")
|
||
except sqlite3.IntegrityError:
|
||
print("ℹ️ Default users already exist")
|
||
|
||
conn.close()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
init_database()
|
||
create_default_users()
|