o i @sbddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z dZ da z:ddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlmZddlZ dZ Wn e!yZ"zede"ded#WYdZ"["ndZ"["wwddZ$ddZ%ddZ&ddZ'da(ddZ)ddZ*dd Z+d!d"Z,d#d$Z-d%d&Z.dS)'N)log_startup_warning) get_missing_requirements_message)FileLockTimeout)argsF)command)Config)MigrationContext)ScriptDirectory) create_engineevent) sessionmaker) StaticPool)BaseTzh ------------------------------------------------------------------------ Error importing dependencies:  z This error is happening because ComfyUI now uses a local sqlite database. ------------------------------------------------------------------------ cCstS)zG Temporary function to check if the dependencies are available ) _DB_AVAILABLErr-/mnt/c/Users/fbmor/ComfyUI/app/database/db.pydependencies_available&srcCstotduS)z Temporary function to check if the database is available to create a session During initial release there may be environmental issues (or missing dependencies) that prevent the database from being created N)rSessionrrrrcan_create_session-srcCshtjtjtd}tjtj|d}tjtj|d}t|}|d||dtj |S)Nz../..z alembic.ini alembic_dbscript_locationzsqlalchemy.url) ospathjoindirname__file__abspathrset_main_optionr database_url) root_path config_path scripts_pathconfigrrrget_alembic_config5s r%cCs.tj}|dr|ddStd|d)Nz sqlite:///z///zUnsupported database URL 'z'.)rr startswithsplit ValueError)urlrrr get_db_pathAs r+cCsB|d}t|az tjddWdSty td|dw)zAcquire an OS-level file lock to prevent multi-process access. Uses filelock for cross-platform support (macOS, Linux, Windows). The OS automatically releases the lock when the process exits, even on crashes. z.lockr)timeoutz$Could not acquire lock on database 'zk'. Another ComfyUI process may already be using it. Use --database-url to specify a separate database file.N)r_db_lockacquirer RuntimeError)db_path lock_pathrrr_acquire_file_lockKs  r2cCs|dvS)zACheck if the database URL refers to an in-memory SQLite database.)zsqlite:///:memory:z sqlite://rdb_urlrrr _is_memory_db^sr5cCs6tj}td|t|rt|dSt|dS)NzDatabase URL: )rr loggingdebugr5_init_memory_db _init_file_dbr3rrrinit_dbcs   r:cCs@t|tddid}t|ddd}tj|t|dadS) u Initialize an in-memory SQLite database using metadata.create_all. Alembic migrations don't work with in-memory SQLite because each connection gets its own separate database — tables created by Alembic's internal connection are lost immediately. check_same_threadF) poolclass connect_argsconnectcS|}|d|dSNzPRAGMA foreign_keys=ONcursorexecuteclosedbapi_connectionconnection_recordrBrrrset_sqlite_pragmaz  z*_init_memory_db..set_sqlite_pragmabindN) r rr listens_forrmetadata create_allr r)r4enginerHrrrr8ms   r8c Cs t}tj|}t}t|}t|ddd}|}t |}| }t |} | } | dur:tdnG|| kr|d} |rKt|| nd} zt|| td|d| Wn ty} z| rut| |t| td | d} ~ ww|t|t|d adS) zBInitialize a file-backed SQLite database using Alembic migrations.r>cSr?r@rArErrrrHrIz(_init_file_db..set_sqlite_pragmaNzNo target revision found.z.bkpzDatabase upgraded from z to zError upgrading database: rJ)r+rrexistsr%r r rLr>r configureget_current_revisionr from_configget_current_headr6warningshutilcopyrupgradeinfo Exceptionremove exceptionrDr2r r) r4r0 db_existsr$rOrHconncontext current_revscript target_rev backup_patherrrr9s@           r9cCstS)N)rrrrrcreate_sessionsre)/r6rrV app.loggerrutils.install_utilrfilelockrrcomfy.cli_argsrrralembicralembic.configralembic.runtime.migrationr alembic.scriptr sqlalchemyr r sqlalchemy.ormr sqlalchemy.poolrapp.database.modelsrapp.assets.database.modelsapp ImportErrorrdstriprrr%r+r-r2r5r:r8r9rerrrrsT              8