o 1Xxi @s(ddlmZmZmZmZddlmZmZddlm Z m Z ddl m m Z ddlmZddlmZddlmZmZmZddlZddlmZddlZdd lmZmZmZmZmZdd l m!Z!dd l"m#Z#ddl$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z*ddl+m,Z,Gddde j-Z-GdddeeeZ.dS)) ConnectionLockPool PerThreadPoolPool) MigratableDB Migration)SystemSettingsN)SqlEmbeddingsQueue)SqlSysDB)OpenTelemetryClientOpenTelemetryGranularity trace_method)override)SequencecastOptionalTypeAny)Literal) TracebackTypeUUID)local)files) Traversablec @sveZdZUeed<eed<dedefddZede j fdd Z ed e e ed e ed e eded fddZdS) TxWrapper_conn_pool conn_poolstackcCs||_||_||_dSN) _tx_stackconnectrr)selfrr r%V/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/chromadb/db/impl/sqlite.py__init__s  zTxWrapper.__init__returncCs@t|jjdkr|jd|jd|jj||jS)NrPRAGMA case_sensitive_like = ONzBEGIN;)lenr"r rexecuteappendcursorr$r%r%r& __enter__"s    zTxWrapper.__enter__exc_type exc_value tracebackFcCsZ|jjt|jjdkr|dur|jn|j|j|j |jdS)NrF) r"r popr*rcommitrollbackr-closerreturn_to_pool)r$r0r1r2r%r%r&__exit__*s   zTxWrapper.__exit__N)__name__ __module__ __qualname__r__annotations__rrr'rbaseCursorr/rr BaseExceptionrrr8r%r%r%r&rs  rcseZdZUeed<eed<eeed<eed<e ed<e ed<de ffdd Z e d ejed/fd d Ze dejed/fdd Zeed eejfddZeed efddZeed efddZed eefddZed efddZe dejed/fdd Ze dejed/ddZe dejed e fddZe d ejed!ed ee fd"d Z!ed#e"j#d$e d d fd%d&Z$eed'e%e&d e%e'fd(d)Z(eed*e%e'd e%e&fd+d,Z)eed ee*fd-d.Z+Z,S)0SqliteDB _conn_pool _settings_migration_imports_db_filer"_is_persistentsystemcs|j|_tdtdtdg|_|jd|_|t|_|js,d|_t |jdd|_ n"|jdd |_t j |jsHt jt j |jdd t|j|_ t|_t|dS) Nz$chromadb.migrations.embeddings_queuezchromadb.migrations.sysdbzchromadb.migrations.metadb is_persistentzfile::memory:?cache=sharedT)is_uripersist_directoryz/chroma.sqlite3)exist_ok)settingsrBrrCrequirerEr _opentelemetry_clientrDrrAospathexistsmakedirsdirnamerrr"superr')r$rF __class__r%r&r'Ds"  zSqliteDB.__init__zSqliteDB.startr(NcsRt|}|d|dWdn1swY|dS)NzPRAGMA foreign_keys = ONr))rSstarttxr+initialize_migrationsr$currTr%r&rV^s     z SqliteDB.stopcst|jdSr!)rSstoprAr6r.rTr%r&r[gs cCtjSr!)pypikaQueryr%r%r%r& querybuildermzSqliteDB.querybuildercCdS)N?r%r%r%r%r&parameter_formatrzSqliteDB.parameter_formatcCra)Nsqliter%r%r%r%r&migration_scopewrdzSqliteDB.migration_scopecCs|jSr!)rCr.r%r%r&migration_dirs|szSqliteDB.migration_dirscCs$t|jds g|j_t|j|jdS)Nr )r )hasattrr"r rrAr.r%r%r&rWs z SqliteDB.txzSqliteDB.reset_statecs|jds td|}|d|D] }|d|dqWdn1s/wY|j|t dS)N allow_resetzResetting the database is not allowed. Set `allow_reset` to true in the config in tests or other non-production environments where reset should be permitted.zo SELECT name FROM sqlite_master WHERE type='table' zDROP TABLE IF EXISTS r) rBrL ValueErrorrWr+fetchallrAr6rVrS reset_state)r$rZrowrTr%r&rls    zSqliteDB.setup_migrationscCs8|}|dWddS1swYdS)Na~ CREATE TABLE IF NOT EXISTS migrations ( dir TEXT NOT NULL, version INTEGER NOT NULL, filename TEXT NOT NULL, sql TEXT NOT NULL, hash TEXT NOT NULL, PRIMARY KEY (dir, version) ) )rWr+rYr%r%r&setup_migrationss "zSqliteDB.migrations_initializedcCs\| }|d|ddkr WddS WddS1s'wYdS)Nz[SELECT count(*) FROM sqlite_master WHERE type='table' AND name='migrations'rFT)rWr+fetchonerYr%r%r&migrations_initializeds  $zSqliteDB.db_migrationsdirc Cs|M}|d|jfg}|D]4}tt|d}tt|d}tt|d}tt|d}tt|d} |t||||| | dq|WdS1sTwYdS)Nz SELECT dir, version, filename, sql, hash FROM migrations WHERE dir = ? ORDER BY version ASC r)rqversionfilenamesqlhashscope) rWr+namerkrstrintr,rrf) r$rqrZ migrationsrm found_dir found_versionfound_filename found_sql found_hashr%r%r& db_migrationss0   $rZ migrationc Cs<||d|d|d|d|d|d|dfdS)Nrxzw INSERT INTO migrations (dir, version, filename, sql, hash) VALUES (?, ?, ?, ?, ?) rqrvrwry) executescriptr+)r$rZrr%r%r&apply_migrationszSqliteDB.apply_migrationvaluecC|durt|SdSr!r)rr%r%r& uuid_from_dbzSqliteDB.uuid_from_dbuuidcCrr!)r|)rr%r%r& uuid_to_dbrzSqliteDB.uuid_to_dbcCr\r!)sqlite3IntegrityErrorr%r%r%r&unique_constraint_errorr`z SqliteDB.unique_constraint_error)r(N)-r9r:r;rr<r rrr|rboolrr'rr ALLrrVr[ staticmethodrr]r^r_rcrfrgrrWrlrnrprrr=r>rrrrrrr?r __classcell__r%r%rTr&r@<sd        r@)/chromadb.db.impl.sqlite_poolrrrrchromadb.db.migrationsrrchromadb.configrr chromadb.db.basedbr=#chromadb.db.mixins.embeddings_queuer chromadb.db.mixins.sysdbr chromadb.telemetry.opentelemetryr r rr overridesrr]typingrrrrrtyping_extensionsrtypesrrNrr threadingrimportlib_resourcesrimportlib_resources.abcrrr@r%r%r%r&s(         #