| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897 |
- from sqlalchemy import event
- from sqlalchemy.exc import OperationalError
- from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
- from sqlalchemy.orm import DeclarativeBase
- from backend.app.core.config import settings
- def _set_sqlite_pragmas(dbapi_conn, connection_record):
- """Set SQLite pragmas on each new connection for concurrency and performance."""
- cursor = dbapi_conn.cursor()
- # WAL mode allows concurrent readers + one writer (vs default DELETE mode which locks entirely)
- cursor.execute("PRAGMA journal_mode = WAL")
- # Wait up to 5 seconds when the database is locked instead of failing immediately
- cursor.execute("PRAGMA busy_timeout = 15000")
- cursor.execute("PRAGMA synchronous = NORMAL")
- cursor.close()
- engine = create_async_engine(
- settings.database_url,
- echo=settings.debug,
- pool_size=20,
- max_overflow=200,
- )
- # Register the pragma listener on the underlying sync engine
- event.listen(engine.sync_engine, "connect", _set_sqlite_pragmas)
- async_session = async_sessionmaker(
- engine,
- class_=AsyncSession,
- expire_on_commit=False,
- )
- async def close_all_connections():
- """Close all database connections for backup/restore operations."""
- global engine
- await engine.dispose()
- async def reinitialize_database():
- """Reinitialize database connection after restore."""
- global engine, async_session
- engine = create_async_engine(
- settings.database_url,
- echo=settings.debug,
- pool_size=20,
- max_overflow=200,
- )
- event.listen(engine.sync_engine, "connect", _set_sqlite_pragmas)
- async_session = async_sessionmaker(
- engine,
- class_=AsyncSession,
- expire_on_commit=False,
- )
- class Base(DeclarativeBase):
- pass
- async def get_db() -> AsyncSession:
- async with async_session() as session:
- try:
- yield session
- await session.commit()
- except Exception:
- await session.rollback()
- raise
- finally:
- await session.close()
- async def init_db():
- # Import models to register them with SQLAlchemy
- from backend.app.models import ( # noqa: F401
- active_print_spoolman,
- ams_history,
- ams_label,
- api_key,
- archive,
- bug_report,
- color_catalog,
- external_link,
- filament,
- github_backup,
- group,
- kprofile_note,
- library,
- local_preset,
- maintenance,
- notification,
- notification_template,
- orca_base_cache,
- pending_upload,
- print_batch,
- print_log,
- print_queue,
- printer,
- project,
- project_bom,
- settings,
- slot_preset,
- smart_plug,
- spool,
- spool_assignment,
- spool_catalog,
- spool_k_profile,
- spool_usage_history,
- spoolbuddy_device,
- user,
- user_email_pref,
- virtual_printer,
- )
- async with engine.begin() as conn:
- await conn.run_sync(Base.metadata.create_all)
- # Run migrations for new columns (SQLite doesn't auto-add columns)
- await run_migrations(conn)
- # Seed default notification templates
- await seed_notification_templates()
- # Seed default groups and migrate existing users
- await seed_default_groups()
- # Seed default catalog entries
- await seed_spool_catalog()
- await seed_color_catalog()
- async def run_migrations(conn):
- """Add new columns to existing tables if they don't exist."""
- from sqlalchemy import text
- # Migration: Add is_favorite column to print_archives
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN is_favorite BOOLEAN DEFAULT 0"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add content_hash column to print_archives for duplicate detection
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN content_hash VARCHAR(64)"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add auto_off_executed column to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN auto_off_executed BOOLEAN DEFAULT 0"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add on_print_stopped column to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_print_stopped BOOLEAN DEFAULT 1"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add source_3mf_path column to print_archives
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN source_3mf_path VARCHAR(500)"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add f3d_path column to print_archives for Fusion 360 design files
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN f3d_path VARCHAR(500)"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add on_maintenance_due column to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_maintenance_due BOOLEAN DEFAULT 0"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add location column to printers for grouping
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN location VARCHAR(100)"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add interval_type column to maintenance_types
- try:
- await conn.execute(text("ALTER TABLE maintenance_types ADD COLUMN interval_type VARCHAR(20) DEFAULT 'hours'"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add is_deleted column to maintenance_types for soft-deletes
- try:
- await conn.execute(text("ALTER TABLE maintenance_types ADD COLUMN is_deleted BOOLEAN DEFAULT 0"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add custom_interval_type column to printer_maintenance
- try:
- await conn.execute(text("ALTER TABLE printer_maintenance ADD COLUMN custom_interval_type VARCHAR(20)"))
- except OperationalError:
- # Column already exists
- pass
- # Migration: Add power alert columns to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN power_alert_enabled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN power_alert_high REAL"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN power_alert_low REAL"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN power_alert_last_triggered DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add schedule columns to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN schedule_enabled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN schedule_on_time VARCHAR(5)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN schedule_off_time VARCHAR(5)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add daily digest columns to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN daily_digest_enabled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN daily_digest_time VARCHAR(5)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add missing-spool-assignment print-start notification toggle
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_print_missing_spool_assignment BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add project_id column to print_archives
- try:
- await conn.execute(
- text("ALTER TABLE print_archives ADD COLUMN project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add project_id column to print_queue
- try:
- await conn.execute(
- text("ALTER TABLE print_queue ADD COLUMN project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Create FTS5 virtual table for archive full-text search
- try:
- await conn.execute(
- text("""
- CREATE VIRTUAL TABLE IF NOT EXISTS archive_fts USING fts5(
- print_name,
- filename,
- tags,
- notes,
- designer,
- filament_type,
- content='print_archives',
- content_rowid='id'
- )
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Create triggers to keep FTS index in sync
- try:
- await conn.execute(
- text("""
- CREATE TRIGGER IF NOT EXISTS archive_fts_insert AFTER INSERT ON print_archives BEGIN
- INSERT INTO archive_fts(rowid, print_name, filename, tags, notes, designer, filament_type)
- VALUES (new.id, new.print_name, new.filename, new.tags, new.notes, new.designer, new.filament_type);
- END
- """)
- )
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(
- text("""
- CREATE TRIGGER IF NOT EXISTS archive_fts_delete AFTER DELETE ON print_archives BEGIN
- INSERT INTO archive_fts(archive_fts, rowid, print_name, filename, tags, notes, designer, filament_type)
- VALUES ('delete', old.id, old.print_name, old.filename, old.tags, old.notes, old.designer, old.filament_type);
- END
- """)
- )
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(
- text("""
- CREATE TRIGGER IF NOT EXISTS archive_fts_update AFTER UPDATE ON print_archives BEGIN
- INSERT INTO archive_fts(archive_fts, rowid, print_name, filename, tags, notes, designer, filament_type)
- VALUES ('delete', old.id, old.print_name, old.filename, old.tags, old.notes, old.designer, old.filament_type);
- INSERT INTO archive_fts(rowid, print_name, filename, tags, notes, designer, filament_type)
- VALUES (new.id, new.print_name, new.filename, new.tags, new.notes, new.designer, new.filament_type);
- END
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add auto_off_pending columns to smart_plugs (for restart recovery)
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN auto_off_pending BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN auto_off_pending_since DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add auto_off_persistent column to smart_plugs (keep auto-off enabled between prints)
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN auto_off_persistent BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add AMS alarm notification columns to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_ams_humidity_high BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_ams_temperature_high BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add AMS-HT alarm notification columns to notification_providers
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_ams_ht_humidity_high BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_ams_ht_temperature_high BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add plate not empty notification column to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_plate_not_empty BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Add notes column to projects (Phase 2)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN notes TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add attachments column to projects (Phase 3)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN attachments JSON"))
- except OperationalError:
- pass # Already applied
- # Migration: Add tags column to projects (Phase 4)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN tags TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add due_date column to projects (Phase 5)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN due_date DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add priority column to projects (Phase 5)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN priority VARCHAR(20) DEFAULT 'normal'"))
- except OperationalError:
- pass # Already applied
- # Migration: Add budget column to projects (Phase 6)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN budget REAL"))
- except OperationalError:
- pass # Already applied
- # Migration: Add is_template column to projects (Phase 8)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN is_template BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add template_source_id column to projects (Phase 8)
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN template_source_id INTEGER"))
- except OperationalError:
- pass # Already applied
- # Migration: Add parent_id column to projects (Phase 10)
- try:
- await conn.execute(
- text("ALTER TABLE projects ADD COLUMN parent_id INTEGER REFERENCES projects(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Rename quantity_printed to quantity_acquired in project_bom_items
- try:
- await conn.execute(text("ALTER TABLE project_bom_items RENAME COLUMN quantity_printed TO quantity_acquired"))
- except OperationalError:
- pass # Already applied
- # Migration: Add unit_price column to project_bom_items
- try:
- await conn.execute(text("ALTER TABLE project_bom_items ADD COLUMN unit_price REAL"))
- except OperationalError:
- pass # Already applied
- # Migration: Add sourcing_url column to project_bom_items
- try:
- await conn.execute(text("ALTER TABLE project_bom_items ADD COLUMN sourcing_url VARCHAR(512)"))
- except OperationalError:
- pass # Already applied
- # Migration: Rename notes to remarks in project_bom_items
- try:
- await conn.execute(text("ALTER TABLE project_bom_items RENAME COLUMN notes TO remarks"))
- except OperationalError:
- pass # Already applied
- # Migration: Add show_in_switchbar column to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN show_in_switchbar BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add runtime tracking columns to printers
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN runtime_seconds INTEGER DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN last_runtime_update DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add quantity column to print_archives for tracking item count
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN quantity INTEGER DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Add manual_start column to print_queue for staged prints
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN manual_start BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add wiki_url column to maintenance_types for documentation links
- try:
- await conn.execute(text("ALTER TABLE maintenance_types ADD COLUMN wiki_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add ams_mapping column to print_queue for storing filament slot assignments
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN ams_mapping TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add target_parts_count column to projects for tracking total parts needed
- try:
- await conn.execute(text("ALTER TABLE projects ADD COLUMN target_parts_count INTEGER"))
- except OperationalError:
- pass # Already applied
- # Migration: Make printer_id nullable in print_queue for unassigned queue items
- # SQLite doesn't support ALTER COLUMN, so we need to recreate the table
- try:
- # Check if printer_id is already nullable by trying to insert NULL
- # This is a safe check that won't affect existing data
- result = await conn.execute(text("SELECT sql FROM sqlite_master WHERE type='table' AND name='print_queue'"))
- row = result.fetchone()
- if row and "printer_id INTEGER NOT NULL" in (row[0] or ""):
- # Need to migrate - printer_id is currently NOT NULL
- await conn.execute(
- text("""
- CREATE TABLE print_queue_new (
- id INTEGER PRIMARY KEY,
- printer_id INTEGER REFERENCES printers(id) ON DELETE CASCADE,
- archive_id INTEGER NOT NULL REFERENCES print_archives(id) ON DELETE CASCADE,
- project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL,
- position INTEGER DEFAULT 0,
- scheduled_time DATETIME,
- manual_start BOOLEAN DEFAULT 0,
- require_previous_success BOOLEAN DEFAULT 0,
- auto_off_after BOOLEAN DEFAULT 0,
- ams_mapping TEXT,
- status VARCHAR(20) DEFAULT 'pending',
- started_at DATETIME,
- completed_at DATETIME,
- error_message TEXT,
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- await conn.execute(
- text("""
- INSERT INTO print_queue_new
- SELECT id, printer_id, archive_id, project_id, position, scheduled_time,
- manual_start, require_previous_success, auto_off_after, ams_mapping,
- status, started_at, completed_at, error_message, created_at
- FROM print_queue
- """)
- )
- await conn.execute(text("DROP TABLE print_queue"))
- await conn.execute(text("ALTER TABLE print_queue_new RENAME TO print_queue"))
- except OperationalError:
- pass # Already applied
- # Migration: Add plug_type column to smart_plugs for HA integration
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN plug_type VARCHAR(20) DEFAULT 'tasmota'"))
- except OperationalError:
- pass # Already applied
- # Migration: Add ha_entity_id column to smart_plugs for HA integration
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN ha_entity_id VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add project_id column to library_folders for linking folders to projects
- try:
- await conn.execute(
- text("ALTER TABLE library_folders ADD COLUMN project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add archive_id column to library_folders for linking folders to archives
- try:
- await conn.execute(
- text(
- "ALTER TABLE library_folders ADD COLUMN archive_id INTEGER REFERENCES print_archives(id) ON DELETE SET NULL"
- )
- )
- except OperationalError:
- pass # Already applied
- # Migration: Make ip_address nullable for HA plugs (SQLite requires table recreation)
- try:
- # Check if ip_address is currently NOT NULL
- result = await conn.execute(text("SELECT sql FROM sqlite_master WHERE type='table' AND name='smart_plugs'"))
- row = result.fetchone()
- if row and "ip_address VARCHAR(45) NOT NULL" in (row[0] or ""):
- # Need to migrate - ip_address is currently NOT NULL
- await conn.execute(
- text("""
- CREATE TABLE smart_plugs_new (
- id INTEGER PRIMARY KEY,
- name VARCHAR(100) NOT NULL,
- ip_address VARCHAR(45),
- plug_type VARCHAR(20) DEFAULT 'tasmota',
- ha_entity_id VARCHAR(100),
- printer_id INTEGER UNIQUE REFERENCES printers(id) ON DELETE SET NULL,
- enabled BOOLEAN NOT NULL DEFAULT 1,
- auto_on BOOLEAN NOT NULL DEFAULT 1,
- auto_off BOOLEAN NOT NULL DEFAULT 1,
- auto_off_persistent BOOLEAN NOT NULL DEFAULT 0,
- off_delay_mode VARCHAR(20) NOT NULL DEFAULT 'time',
- off_delay_minutes INTEGER NOT NULL DEFAULT 5,
- off_temp_threshold INTEGER NOT NULL DEFAULT 70,
- username VARCHAR(50),
- password VARCHAR(100),
- power_alert_enabled BOOLEAN NOT NULL DEFAULT 0,
- power_alert_high FLOAT,
- power_alert_low FLOAT,
- power_alert_last_triggered DATETIME,
- schedule_enabled BOOLEAN NOT NULL DEFAULT 0,
- schedule_on_time VARCHAR(5),
- schedule_off_time VARCHAR(5),
- show_in_switchbar BOOLEAN DEFAULT 0,
- last_state VARCHAR(10),
- last_checked DATETIME,
- auto_off_executed BOOLEAN NOT NULL DEFAULT 0,
- auto_off_pending BOOLEAN DEFAULT 0,
- auto_off_pending_since DATETIME,
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL,
- updated_at DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL
- )
- """)
- )
- await conn.execute(
- text("""
- INSERT INTO smart_plugs_new
- SELECT id, name, ip_address,
- COALESCE(plug_type, 'tasmota'), ha_entity_id, printer_id,
- enabled, auto_on, auto_off, COALESCE(auto_off_persistent, 0),
- off_delay_mode, off_delay_minutes, off_temp_threshold,
- username, password, power_alert_enabled, power_alert_high, power_alert_low,
- power_alert_last_triggered, schedule_enabled, schedule_on_time, schedule_off_time,
- COALESCE(show_in_switchbar, 0), last_state, last_checked, auto_off_executed,
- COALESCE(auto_off_pending, 0), auto_off_pending_since, created_at, updated_at
- FROM smart_plugs
- """)
- )
- await conn.execute(text("DROP TABLE smart_plugs"))
- await conn.execute(text("ALTER TABLE smart_plugs_new RENAME TO smart_plugs"))
- except OperationalError:
- pass # Already applied
- # Migration: Add plate_id column to print_queue for multi-plate 3MF support
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN plate_id INTEGER"))
- except OperationalError:
- pass # Already applied
- # Migration: Add print options columns to print_queue
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN bed_levelling BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN flow_cali BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN vibration_cali BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN layer_inspect BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN timelapse BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN use_ams BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Add library_file_id column to print_queue and make archive_id nullable
- # This allows queue items to reference library files directly (archive created at print start)
- try:
- await conn.execute(
- text(
- "ALTER TABLE print_queue ADD COLUMN library_file_id INTEGER REFERENCES library_files(id) ON DELETE CASCADE"
- )
- )
- except OperationalError:
- pass # Already applied
- # Check if archive_id needs to be made nullable (requires table recreation in SQLite)
- try:
- result = await conn.execute(text("SELECT sql FROM sqlite_master WHERE type='table' AND name='print_queue'"))
- row = result.fetchone()
- if row and "archive_id INTEGER NOT NULL" in (row[0] or ""):
- # Need to migrate - archive_id is currently NOT NULL
- await conn.execute(
- text("""
- CREATE TABLE print_queue_new2 (
- id INTEGER PRIMARY KEY,
- printer_id INTEGER REFERENCES printers(id) ON DELETE CASCADE,
- archive_id INTEGER REFERENCES print_archives(id) ON DELETE CASCADE,
- library_file_id INTEGER REFERENCES library_files(id) ON DELETE CASCADE,
- project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL,
- position INTEGER DEFAULT 0,
- scheduled_time DATETIME,
- manual_start BOOLEAN DEFAULT 0,
- require_previous_success BOOLEAN DEFAULT 0,
- auto_off_after BOOLEAN DEFAULT 0,
- ams_mapping TEXT,
- plate_id INTEGER,
- bed_levelling BOOLEAN DEFAULT 1,
- flow_cali BOOLEAN DEFAULT 0,
- vibration_cali BOOLEAN DEFAULT 1,
- layer_inspect BOOLEAN DEFAULT 0,
- timelapse BOOLEAN DEFAULT 0,
- use_ams BOOLEAN DEFAULT 1,
- status VARCHAR(20) DEFAULT 'pending',
- started_at DATETIME,
- completed_at DATETIME,
- error_message TEXT,
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- await conn.execute(
- text("""
- INSERT INTO print_queue_new2
- SELECT id, printer_id, archive_id, NULL, project_id, position, scheduled_time,
- manual_start, require_previous_success, auto_off_after, ams_mapping, plate_id,
- COALESCE(bed_levelling, 1), COALESCE(flow_cali, 0), COALESCE(vibration_cali, 1),
- COALESCE(layer_inspect, 0), COALESCE(timelapse, 0), COALESCE(use_ams, 1),
- status, started_at, completed_at, error_message, created_at
- FROM print_queue
- """)
- )
- await conn.execute(text("DROP TABLE print_queue"))
- await conn.execute(text("ALTER TABLE print_queue_new2 RENAME TO print_queue"))
- except OperationalError:
- pass # Already applied
- # Migration: Add HA energy sensor entity columns to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN ha_power_entity VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN ha_energy_today_entity VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN ha_energy_total_entity VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- # Migration: Create users table for authentication
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS users (
- id INTEGER PRIMARY KEY,
- username VARCHAR(100) NOT NULL UNIQUE,
- password_hash VARCHAR(255) NOT NULL,
- role VARCHAR(20) NOT NULL DEFAULT 'user',
- is_active BOOLEAN NOT NULL DEFAULT 1,
- created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
- updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- await conn.execute(text("CREATE INDEX IF NOT EXISTS ix_users_username ON users(username)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add external camera columns to printers
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN external_camera_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN external_camera_type VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN external_camera_enabled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add external_url column to print_archives for user-defined links (Printables, etc.)
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN external_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add sliced_for_model column to print_archives for model-based queue assignment
- try:
- await conn.execute(text("ALTER TABLE print_archives ADD COLUMN sliced_for_model VARCHAR(50)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add is_external column to library_files for external cloud files
- try:
- await conn.execute(text("ALTER TABLE library_files ADD COLUMN is_external BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add project_id column to library_files
- try:
- await conn.execute(
- text("ALTER TABLE library_files ADD COLUMN project_id INTEGER REFERENCES projects(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add is_external column to library_folders for external cloud folders
- try:
- await conn.execute(text("ALTER TABLE library_folders ADD COLUMN is_external BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add external folder settings columns to library_folders
- try:
- await conn.execute(text("ALTER TABLE library_folders ADD COLUMN external_readonly BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE library_folders ADD COLUMN external_show_hidden BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE library_folders ADD COLUMN external_path VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add plate_detection_enabled column to printers
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN plate_detection_enabled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add plate detection ROI columns to printers
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN plate_detection_roi_x REAL"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN plate_detection_roi_y REAL"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN plate_detection_roi_w REAL"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN plate_detection_roi_h REAL"))
- except OperationalError:
- pass # Already applied
- # Migration: Remove UNIQUE constraint from smart_plugs.printer_id
- # This allows HA scripts to coexist with regular plugs (scripts are for multi-device control)
- # SQLite requires table recreation to drop constraints
- try:
- # Check if we need to migrate — look for UNIQUE on printer_id in the
- # CREATE TABLE statement OR as a separate UNIQUE index.
- needs_migration = False
- result = await conn.execute(text("SELECT sql FROM sqlite_master WHERE type='table' AND name='smart_plugs'"))
- row = result.fetchone()
- table_sql = (row[0] or "").upper() if row else ""
- if "PRINTER_ID" in table_sql and "UNIQUE" in table_sql:
- # Check if UNIQUE appears near printer_id — inline or table-level constraint.
- # Handle quoted ("PRINTER_ID") and unquoted column names.
- import re
- if re.search(r'"?PRINTER_ID"?\s+\w+\s+UNIQUE', table_sql) or re.search(
- r'UNIQUE\s*\([^)]*"?PRINTER_ID"?', table_sql
- ):
- needs_migration = True
- # Also check for separate UNIQUE indexes on printer_id
- idx_result = await conn.execute(
- text("SELECT sql FROM sqlite_master WHERE type='index' AND tbl_name='smart_plugs' AND sql IS NOT NULL")
- )
- for idx_row in idx_result.fetchall():
- idx_sql = (idx_row[0] or "").upper()
- if "UNIQUE" in idx_sql and "PRINTER_ID" in idx_sql:
- needs_migration = True
- break
- if needs_migration:
- # Create new table without UNIQUE constraint on printer_id
- await conn.execute(
- text("""
- CREATE TABLE smart_plugs_temp (
- id INTEGER PRIMARY KEY,
- name VARCHAR(100) NOT NULL,
- ip_address VARCHAR(45),
- plug_type VARCHAR(20) DEFAULT 'tasmota',
- ha_entity_id VARCHAR(100),
- ha_power_entity VARCHAR(100),
- ha_energy_today_entity VARCHAR(100),
- ha_energy_total_entity VARCHAR(100),
- printer_id INTEGER REFERENCES printers(id) ON DELETE SET NULL,
- enabled BOOLEAN NOT NULL DEFAULT 1,
- auto_on BOOLEAN NOT NULL DEFAULT 1,
- auto_off BOOLEAN NOT NULL DEFAULT 1,
- auto_off_persistent BOOLEAN NOT NULL DEFAULT 0,
- off_delay_mode VARCHAR(20) NOT NULL DEFAULT 'time',
- off_delay_minutes INTEGER NOT NULL DEFAULT 5,
- off_temp_threshold INTEGER NOT NULL DEFAULT 70,
- username VARCHAR(50),
- password VARCHAR(100),
- power_alert_enabled BOOLEAN NOT NULL DEFAULT 0,
- power_alert_high FLOAT,
- power_alert_low FLOAT,
- power_alert_last_triggered DATETIME,
- schedule_enabled BOOLEAN NOT NULL DEFAULT 0,
- schedule_on_time VARCHAR(5),
- schedule_off_time VARCHAR(5),
- show_in_switchbar BOOLEAN DEFAULT 0,
- last_state VARCHAR(10),
- last_checked DATETIME,
- auto_off_executed BOOLEAN NOT NULL DEFAULT 0,
- auto_off_pending BOOLEAN DEFAULT 0,
- auto_off_pending_since DATETIME,
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL,
- updated_at DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL
- )
- """)
- )
- # Copy data
- await conn.execute(
- text("""
- INSERT INTO smart_plugs_temp
- SELECT id, name, ip_address, plug_type, ha_entity_id, ha_power_entity,
- ha_energy_today_entity, ha_energy_total_entity, printer_id, enabled,
- auto_on, auto_off, COALESCE(auto_off_persistent, 0),
- off_delay_mode, off_delay_minutes, off_temp_threshold,
- username, password, power_alert_enabled, power_alert_high, power_alert_low,
- power_alert_last_triggered, schedule_enabled, schedule_on_time, schedule_off_time,
- show_in_switchbar, last_state, last_checked, auto_off_executed,
- auto_off_pending, auto_off_pending_since, created_at, updated_at
- FROM smart_plugs
- """)
- )
- # Drop old table and rename new one
- await conn.execute(text("DROP TABLE smart_plugs"))
- await conn.execute(text("ALTER TABLE smart_plugs_temp RENAME TO smart_plugs"))
- except OperationalError:
- pass # Already applied
- # Migration: Add show_on_printer_card column to smart_plugs
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN show_on_printer_card BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Add MQTT smart plug fields (legacy)
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_topic VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_power_path VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_energy_path VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_state_path VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_multiplier REAL DEFAULT 1.0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add enhanced MQTT smart plug fields (separate topics and multipliers)
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_power_topic VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_power_multiplier REAL DEFAULT 1.0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_energy_topic VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_energy_multiplier REAL DEFAULT 1.0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_state_topic VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN mqtt_state_on_value VARCHAR(50)"))
- except OperationalError:
- pass # Already applied
- # Migration: Copy existing mqtt_topic to mqtt_power_topic for backward compatibility
- try:
- await conn.execute(
- text("""
- UPDATE smart_plugs
- SET mqtt_power_topic = mqtt_topic,
- mqtt_power_multiplier = mqtt_multiplier
- WHERE mqtt_topic IS NOT NULL AND mqtt_power_topic IS NULL
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Create groups table for permission-based access control
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS groups (
- id INTEGER PRIMARY KEY,
- name VARCHAR(100) NOT NULL UNIQUE,
- description VARCHAR(500),
- permissions JSON,
- is_system BOOLEAN NOT NULL DEFAULT 0,
- created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
- updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- await conn.execute(text("CREATE INDEX IF NOT EXISTS ix_groups_name ON groups(name)"))
- except OperationalError:
- pass # Already applied
- # Migration: Create user_groups association table
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS user_groups (
- user_id INTEGER NOT NULL,
- group_id INTEGER NOT NULL,
- PRIMARY KEY (user_id, group_id),
- FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE,
- FOREIGN KEY (group_id) REFERENCES groups(id) ON DELETE CASCADE
- )
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add model-based queue assignment columns to print_queue
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN target_model VARCHAR(50)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN required_filament_types TEXT"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN waiting_reason TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add nozzle_count column to printers (for dual-extruder detection)
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN nozzle_count INTEGER DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Add print_hours_offset column to printers (baseline hours adjustment)
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN print_hours_offset REAL DEFAULT 0.0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add queue notification event columns to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_added BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_assigned BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_started BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_waiting BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_skipped BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_job_failed BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_queue_completed BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add created_by_id column to print_archives for user tracking (Issue #206)
- try:
- await conn.execute(
- text("ALTER TABLE print_archives ADD COLUMN created_by_id INTEGER REFERENCES users(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add created_by_id column to print_queue for user tracking (Issue #206)
- try:
- await conn.execute(
- text("ALTER TABLE print_queue ADD COLUMN created_by_id INTEGER REFERENCES users(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add created_by_id column to library_files for user tracking (Issue #206)
- try:
- await conn.execute(
- text("ALTER TABLE library_files ADD COLUMN created_by_id INTEGER REFERENCES users(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add target_location column to print_queue for location-based filtering (Issue #220)
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN target_location VARCHAR(100)"))
- except OperationalError:
- pass # Already applied
- # Migration: Convert absolute paths to relative paths in library_files table
- # This ensures backup/restore portability across different installations
- try:
- base_dir_str = str(settings.base_dir)
- # Ensure we have a trailing slash for clean replacement
- if not base_dir_str.endswith("/"):
- base_dir_str += "/"
- # Update file_path - remove base_dir prefix from absolute paths
- await conn.execute(
- text("""
- UPDATE library_files
- SET file_path = SUBSTR(file_path, LENGTH(:base_dir) + 1)
- WHERE file_path LIKE :pattern
- """),
- {"base_dir": base_dir_str, "pattern": base_dir_str + "%"},
- )
- # Update thumbnail_path - remove base_dir prefix from absolute paths
- await conn.execute(
- text("""
- UPDATE library_files
- SET thumbnail_path = SUBSTR(thumbnail_path, LENGTH(:base_dir) + 1)
- WHERE thumbnail_path LIKE :pattern
- """),
- {"base_dir": base_dir_str, "pattern": base_dir_str + "%"},
- )
- except OperationalError:
- pass # Already applied
- # Create active_print_spoolman table for Spoolman per-filament tracking
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS active_print_spoolman (
- id INTEGER PRIMARY KEY AUTOINCREMENT,
- printer_id INTEGER NOT NULL REFERENCES printers(id) ON DELETE CASCADE,
- archive_id INTEGER NOT NULL REFERENCES print_archives(id) ON DELETE CASCADE,
- filament_usage TEXT NOT NULL,
- ams_trays TEXT NOT NULL,
- slot_to_tray TEXT,
- layer_usage TEXT,
- filament_properties TEXT,
- UNIQUE(printer_id, archive_id)
- )
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add preset_source column to slot_preset_mappings for local preset support
- try:
- await conn.execute(
- text("ALTER TABLE slot_preset_mappings ADD COLUMN preset_source VARCHAR(20) DEFAULT 'cloud'")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add email column to users for Advanced Auth (PR #322)
- try:
- await conn.execute(text("ALTER TABLE users ADD COLUMN email VARCHAR(255)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add inventory spool tracking columns
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN added_full BOOLEAN"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN last_used DATETIME"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN encode_time DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add RFID tag matching columns to spool
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN tag_uid VARCHAR(16)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN tray_uuid VARCHAR(32)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN data_origin VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN tag_type VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add core_weight_catalog_id to track which catalog entry was used for empty spool weight
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN core_weight_catalog_id INTEGER"))
- except OperationalError:
- pass # Already applied
- # Migration: Create spool_usage_history table for filament consumption tracking
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS spool_usage_history (
- id INTEGER PRIMARY KEY AUTOINCREMENT,
- spool_id INTEGER NOT NULL REFERENCES spool(id) ON DELETE CASCADE,
- printer_id INTEGER REFERENCES printers(id) ON DELETE SET NULL,
- print_name VARCHAR(500),
- weight_used REAL NOT NULL DEFAULT 0,
- percent_used INTEGER NOT NULL DEFAULT 0,
- status VARCHAR(20) NOT NULL DEFAULT 'completed',
- created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add open_in_new_tab column to external_links
- try:
- await conn.execute(text("ALTER TABLE external_links ADD COLUMN open_in_new_tab BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add bed cooled notification column to notification_providers
- try:
- await conn.execute(text("ALTER TABLE notification_providers ADD COLUMN on_bed_cooled BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add first layer complete notification column to notification_providers
- try:
- await conn.execute(
- text("ALTER TABLE notification_providers ADD COLUMN on_first_layer_complete BOOLEAN DEFAULT 0")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Add weight_locked flag to spool table (skip AMS auto-sync for manually-entered weights)
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN weight_locked BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add SpoolBuddy scale weight tracking columns to spool table
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN last_scale_weight INTEGER"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN last_weighed_at DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add cost tracking fields to spool table
- try:
- await conn.execute(text("ALTER TABLE spool ADD COLUMN cost_per_kg REAL"))
- except OperationalError:
- pass # Already applied
- # Migration: Add cost field to spool_usage_history table
- try:
- await conn.execute(text("ALTER TABLE spool_usage_history ADD COLUMN cost REAL"))
- except OperationalError:
- pass # Already applied
- # Migration: Add archive_id field to spool_usage_history table
- try:
- await conn.execute(
- text("ALTER TABLE spool_usage_history ADD COLUMN archive_id INTEGER REFERENCES print_archives(id)")
- )
- except OperationalError:
- pass # Already applied
- # Migration: Migrate single virtual printer key-value settings to virtual_printers table
- try:
- # Check if virtual_printers table has any rows
- result = await conn.execute(text("SELECT COUNT(*) FROM virtual_printers"))
- count = result.scalar() or 0
- if count == 0:
- # Check if old key-value settings exist
- result = await conn.execute(text("SELECT value FROM settings WHERE key = 'virtual_printer_enabled'"))
- row = result.fetchone()
- if row:
- # Old settings exist — migrate to first virtual printer row
- old_enabled = row[0] == "true" if row[0] else False
- result = await conn.execute(
- text("SELECT value FROM settings WHERE key = 'virtual_printer_access_code'")
- )
- row = result.fetchone()
- old_access_code = row[0] if row else None
- result = await conn.execute(text("SELECT value FROM settings WHERE key = 'virtual_printer_mode'"))
- row = result.fetchone()
- old_mode = row[0] if row else "immediate"
- if old_mode == "queue":
- old_mode = "review"
- result = await conn.execute(text("SELECT value FROM settings WHERE key = 'virtual_printer_model'"))
- row = result.fetchone()
- old_model = row[0] if row else "BL-P001"
- result = await conn.execute(
- text("SELECT value FROM settings WHERE key = 'virtual_printer_target_printer_id'")
- )
- row = result.fetchone()
- old_target_id = int(row[0]) if row and row[0] else None
- result = await conn.execute(
- text("SELECT value FROM settings WHERE key = 'virtual_printer_remote_interface_ip'")
- )
- row = result.fetchone()
- old_remote_iface = row[0] if row else None
- await conn.execute(
- text("""
- INSERT INTO virtual_printers
- (name, enabled, mode, model, access_code, target_printer_id,
- bind_ip, remote_interface_ip, serial_suffix, position)
- VALUES
- (:name, :enabled, :mode, :model, :access_code, :target_id,
- NULL, :remote_iface, '391800001', 0)
- """),
- {
- "name": "Bambuddy",
- "enabled": old_enabled,
- "mode": old_mode or "immediate",
- "model": old_model,
- "access_code": old_access_code,
- "target_id": old_target_id,
- "remote_iface": old_remote_iface,
- },
- )
- except OperationalError:
- pass # Table may not exist yet on first run
- # Migration: Add filament_overrides column to print_queue for filament override in model-based assignment
- try:
- await conn.execute(text("ALTER TABLE print_queue ADD COLUMN filament_overrides TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add NFC reader and display control columns to spoolbuddy_devices
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN nfc_reader_type VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN nfc_connection VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN display_brightness INTEGER DEFAULT 100"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN display_blank_timeout INTEGER DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN has_backlight BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN last_calibrated_at DATETIME"))
- except OperationalError:
- pass # Already applied
- # Migration: Add NFC tag write payload column to spoolbuddy_devices
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN pending_write_payload TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add OTA update tracking columns to spoolbuddy_devices
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN update_status VARCHAR(20)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN update_message VARCHAR(255)"))
- except OperationalError:
- pass # Already applied
- # Migration: Persist SpoolBuddy backend URL and queued system payload
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN backend_url VARCHAR(255)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN pending_system_payload TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Add system_stats JSON blob column to spoolbuddy_devices
- try:
- await conn.execute(text("ALTER TABLE spoolbuddy_devices ADD COLUMN system_stats TEXT"))
- except OperationalError:
- pass # Already applied
- # Migration: Convert ams_labels table from (printer_id, ams_id) key to ams_serial_number key
- # Labels are now keyed by AMS serial number so they persist when the AMS is moved to another printer.
- try:
- await conn.execute(text("DROP TABLE IF EXISTS ams_labels_new"))
- result = await conn.execute(text("SELECT sql FROM sqlite_master WHERE type='table' AND name='ams_labels'"))
- row = result.fetchone()
- if row and "printer_id" in (row[0] or ""):
- # Old schema: rebuild the table with ams_serial_number as the unique key.
- # Existing rows get a synthetic serial "p{printer_id}a{ams_id}" so data is preserved.
- await conn.execute(
- text("""
- CREATE TABLE ams_labels_new (
- id INTEGER PRIMARY KEY,
- ams_serial_number VARCHAR(50) NOT NULL,
- ams_id INTEGER,
- label VARCHAR(100) NOT NULL,
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
- updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
- CONSTRAINT uq_ams_label_serial UNIQUE (ams_serial_number)
- )
- """)
- )
- await conn.execute(
- text("""
- INSERT INTO ams_labels_new (id, ams_serial_number, ams_id, label, created_at, updated_at)
- SELECT id,
- 'p' || CAST(printer_id AS TEXT) || 'a' || CAST(ams_id AS TEXT),
- ams_id,
- label,
- created_at,
- updated_at
- FROM ams_labels
- """)
- )
- await conn.execute(text("DROP TABLE ams_labels"))
- await conn.execute(text("ALTER TABLE ams_labels_new RENAME TO ams_labels"))
- except OperationalError:
- pass # Already migrated or table does not exist yet
- # Migration: Add auto_dispatch column to virtual_printers
- try:
- await conn.execute(text("ALTER TABLE virtual_printers ADD COLUMN auto_dispatch BOOLEAN DEFAULT 1"))
- except OperationalError:
- pass # Already applied
- # Migration: Fix VP model codes — convert legacy SSDP codes and display names to correct SSDP codes
- # Legacy codes (from multi-VP refactor) and display names (from proxy auto-inherit)
- vp_model_fixes = {
- "3DPrinter-X1-Carbon": "BL-P001",
- "3DPrinter-X1": "BL-P002",
- "X1C": "BL-P001",
- "X1": "BL-P002",
- "X1E": "C13",
- "P1P": "C11",
- "P1S": "C12",
- "P2S": "N7",
- "A1": "N2S",
- "A1 Mini": "N1",
- "H2D": "O1D",
- "H2C": "O1C",
- "H2S": "O1S",
- }
- for old_val, new_val in vp_model_fixes.items():
- await conn.execute(
- text("UPDATE virtual_printers SET model = :new WHERE model = :old"),
- {"old": old_val, "new": new_val},
- )
- await conn.execute(
- text("UPDATE settings SET value = :new WHERE key = 'virtual_printer_model' AND value = :old"),
- {"old": old_val, "new": new_val},
- )
- # Migration: Add per-user Bambu Cloud credential columns
- try:
- await conn.execute(text("ALTER TABLE users ADD COLUMN cloud_token VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE users ADD COLUMN cloud_email VARCHAR(255)"))
- except OperationalError:
- pass # Already applied
- # Cleanup: Remove obsolete settings keys that are no longer used
- obsolete_keys = ["slicer_binary_path"]
- for key in obsolete_keys:
- await conn.execute(text("DELETE FROM settings WHERE key = :key"), {"key": key})
- # Migration: Create user_email_preferences table for user-specific email notification settings
- try:
- await conn.execute(
- text("""
- CREATE TABLE IF NOT EXISTS user_email_preferences (
- id INTEGER PRIMARY KEY,
- user_id INTEGER NOT NULL UNIQUE REFERENCES users(id) ON DELETE CASCADE,
- notify_print_start BOOLEAN NOT NULL DEFAULT 1,
- notify_print_complete BOOLEAN NOT NULL DEFAULT 1,
- notify_print_failed BOOLEAN NOT NULL DEFAULT 1,
- notify_print_stopped BOOLEAN NOT NULL DEFAULT 1,
- created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
- updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
- )
- """)
- )
- await conn.execute(
- text("CREATE INDEX IF NOT EXISTS ix_user_email_preferences_user_id ON user_email_preferences(user_id)")
- )
- except OperationalError:
- pass # Already applied
- # Legacy migration: Add notify_print_stopped column (for any existing partial tables)
- try:
- await conn.execute(
- text("ALTER TABLE user_email_preferences ADD COLUMN notify_print_stopped BOOLEAN NOT NULL DEFAULT 1")
- )
- except OperationalError:
- pass # Column already exists or table created with full schema
- # Migration: Add camera_rotation column to printers
- try:
- await conn.execute(text("ALTER TABLE printers ADD COLUMN camera_rotation INTEGER DEFAULT 0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add REST/Webhook smart plug fields
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_on_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_on_body TEXT"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_off_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_off_body TEXT"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_method VARCHAR(10)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_headers TEXT"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_status_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_status_path VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_status_on_value VARCHAR(50)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_power_path VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_energy_path VARCHAR(200)"))
- except OperationalError:
- pass # Already applied
- # Migration: Add separate REST power/energy URLs and multipliers
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_power_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_power_multiplier REAL DEFAULT 1.0"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_energy_url VARCHAR(500)"))
- except OperationalError:
- pass # Already applied
- try:
- await conn.execute(text("ALTER TABLE smart_plugs ADD COLUMN rest_energy_multiplier REAL DEFAULT 1.0"))
- except OperationalError:
- pass # Already applied
- # Migration: Add batch_id column to print_queue for batch grouping
- try:
- await conn.execute(
- text("ALTER TABLE print_queue ADD COLUMN batch_id INTEGER REFERENCES print_batches(id) ON DELETE SET NULL")
- )
- except OperationalError:
- pass
- # Migration: Add backup_spools and backup_archives columns to github_backup_config
- try:
- await conn.execute(text("ALTER TABLE github_backup_config ADD COLUMN backup_spools BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass
- try:
- await conn.execute(text("ALTER TABLE github_backup_config ADD COLUMN backup_archives BOOLEAN DEFAULT 0"))
- except OperationalError:
- pass
- # Seed default settings keys that must exist on fresh install
- default_settings = [
- ("advanced_auth_enabled", "false"),
- ("smtp_auth_enabled", "true"),
- ]
- for key, value in default_settings:
- try:
- await conn.execute(
- text("INSERT OR IGNORE INTO settings (key, value) VALUES (:key, :value)"),
- {"key": key, "value": value},
- )
- except OperationalError:
- pass
- async def seed_notification_templates():
- """Seed default notification templates if they don't exist."""
- from sqlalchemy import select
- from backend.app.models.notification_template import DEFAULT_TEMPLATES, NotificationTemplate
- async with async_session() as session:
- # Get existing template event types
- result = await session.execute(select(NotificationTemplate.event_type))
- existing_types = {row[0] for row in result.fetchall()}
- if not existing_types:
- # No templates exist - insert all defaults
- for template_data in DEFAULT_TEMPLATES:
- template = NotificationTemplate(
- event_type=template_data["event_type"],
- name=template_data["name"],
- title_template=template_data["title_template"],
- body_template=template_data["body_template"],
- is_default=True,
- )
- session.add(template)
- else:
- # Templates exist - only add missing ones
- for template_data in DEFAULT_TEMPLATES:
- if template_data["event_type"] not in existing_types:
- template = NotificationTemplate(
- event_type=template_data["event_type"],
- name=template_data["name"],
- title_template=template_data["title_template"],
- body_template=template_data["body_template"],
- is_default=True,
- )
- session.add(template)
- await session.commit()
- async def seed_default_groups():
- """Seed default groups and migrate existing users to appropriate groups.
- Creates the default system groups (Administrators, Operators, Viewers) if they
- don't exist, then migrates existing users:
- - Users with role='admin' -> Administrators group
- - Users with role='user' -> Operators group
- Also migrates old permissions to new ownership-based permissions (Issue #205).
- """
- import logging
- from sqlalchemy import select
- from backend.app.core.permissions import DEFAULT_GROUPS
- from backend.app.models.group import Group
- from backend.app.models.user import User
- logger = logging.getLogger(__name__)
- # Map old permissions to new ones for migration
- # Administrators get *_all permissions, Operators get *_own permissions
- PERMISSION_MIGRATION_ALL = {
- "queue:update": "queue:update_all",
- "queue:delete": "queue:delete_all",
- "archives:update": "archives:update_all",
- "archives:delete": "archives:delete_all",
- "archives:reprint": "archives:reprint_all",
- "library:update": "library:update_all",
- "library:delete": "library:delete_all",
- }
- PERMISSION_MIGRATION_OWN = {
- "queue:update": "queue:update_own",
- "queue:delete": "queue:delete_own",
- "archives:update": "archives:update_own",
- "archives:delete": "archives:delete_own",
- "archives:reprint": "archives:reprint_own",
- "library:update": "library:update_own",
- "library:delete": "library:delete_own",
- }
- async with async_session() as session:
- # Get existing groups
- result = await session.execute(select(Group))
- existing_groups = {group.name: group for group in result.scalars().all()}
- # Create default groups if they don't exist
- groups_created = []
- for group_name, group_config in DEFAULT_GROUPS.items():
- if group_name not in existing_groups:
- group = Group(
- name=group_name,
- description=group_config["description"],
- permissions=group_config["permissions"],
- is_system=group_config["is_system"],
- )
- session.add(group)
- groups_created.append(group_name)
- logger.info("Created default group: %s", group_name)
- else:
- # Migrate existing group's permissions from old to new format
- group = existing_groups[group_name]
- if group.permissions:
- updated = False
- new_permissions = list(group.permissions)
- # Determine which migration map to use based on group
- migration_map = (
- PERMISSION_MIGRATION_ALL if group_name == "Administrators" else PERMISSION_MIGRATION_OWN
- )
- for old_perm, new_perm in migration_map.items():
- if old_perm in new_permissions:
- new_permissions.remove(old_perm)
- if new_perm not in new_permissions:
- new_permissions.append(new_perm)
- updated = True
- logger.info(
- "Migrated permission '%s' to '%s' in group '%s'", old_perm, new_perm, group_name
- )
- # For Administrators, also ensure they get *_all permissions if they have any new *_own
- if group_name == "Administrators":
- for _own_perm, all_perm in [
- ("queue:update_own", "queue:update_all"),
- ("queue:delete_own", "queue:delete_all"),
- ("archives:update_own", "archives:update_all"),
- ("archives:delete_own", "archives:delete_all"),
- ("archives:reprint_own", "archives:reprint_all"),
- ("library:update_own", "library:update_all"),
- ("library:delete_own", "library:delete_all"),
- ]:
- # Add *_all if not present
- if all_perm not in new_permissions:
- new_permissions.append(all_perm)
- updated = True
- if updated:
- group.permissions = new_permissions
- await session.commit()
- # Migrate new permissions: grant printers:clear_plate to all groups with printers:control
- result = await session.execute(select(Group))
- all_groups = result.scalars().all()
- for group in all_groups:
- if (
- group.permissions
- and "printers:control" in group.permissions
- and "printers:clear_plate" not in group.permissions
- ):
- group.permissions = [*group.permissions, "printers:clear_plate"]
- logger.info("Added printers:clear_plate to group '%s' (has printers:control)", group.name)
- await session.commit()
- # Migrate existing users to groups if they're not already in any group
- if groups_created:
- # Refresh to get newly created groups
- admin_result = await session.execute(select(Group).where(Group.name == "Administrators"))
- admin_group = admin_result.scalar_one_or_none()
- operators_result = await session.execute(select(Group).where(Group.name == "Operators"))
- operators_group = operators_result.scalar_one_or_none()
- # Get all users
- users_result = await session.execute(select(User))
- users = users_result.scalars().all()
- for user in users:
- # Skip if user already has groups
- if user.groups:
- continue
- if user.role == "admin" and admin_group:
- user.groups.append(admin_group)
- logger.info("Migrated admin user '%s' to Administrators group", user.username)
- elif operators_group:
- user.groups.append(operators_group)
- logger.info("Migrated user '%s' to Operators group", user.username)
- await session.commit()
- async def seed_spool_catalog():
- """Seed the spool catalog with default entries if empty."""
- import logging
- from sqlalchemy import func, select
- from backend.app.core.catalog_defaults import DEFAULT_SPOOL_CATALOG
- from backend.app.models.spool_catalog import SpoolCatalogEntry
- logger = logging.getLogger(__name__)
- async with async_session() as session:
- result = await session.execute(select(func.count()).select_from(SpoolCatalogEntry))
- count = result.scalar() or 0
- if count > 0:
- return # Already seeded
- for name, weight in DEFAULT_SPOOL_CATALOG:
- session.add(SpoolCatalogEntry(name=name, weight=weight, is_default=True))
- await session.commit()
- logger.info("Seeded %d default spool catalog entries", len(DEFAULT_SPOOL_CATALOG))
- async def seed_color_catalog():
- """Seed the color catalog with default entries if empty."""
- import logging
- from sqlalchemy import func, select
- from backend.app.core.catalog_defaults import DEFAULT_COLOR_CATALOG
- from backend.app.models.color_catalog import ColorCatalogEntry
- logger = logging.getLogger(__name__)
- async with async_session() as session:
- result = await session.execute(select(func.count()).select_from(ColorCatalogEntry))
- count = result.scalar() or 0
- if count > 0:
- return # Already seeded
- for manufacturer, color_name, hex_color, material in DEFAULT_COLOR_CATALOG:
- session.add(
- ColorCatalogEntry(
- manufacturer=manufacturer,
- color_name=color_name,
- hex_color=hex_color,
- material=material,
- is_default=True,
- )
- )
- await session.commit()
- logger.info("Seeded %d default color catalog entries", len(DEFAULT_COLOR_CATALOG))
|