o :i@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z dZ dZ de ddfd d ZGd d d Zdd dZdS)N)Path)Iterable)modelsz storage.dbmzstorage.sqlitelocationreturnc Cst|t}t|t}|rdS|sdSzUtt|d}tt|}| }| d| D]%}||}t |trE| d}t|}t|}| d|t|fq5||||WdSty} z td| td| d} ~ ww)NcCCREATE TABLE IF NOT EXISTS points (id TEXT PRIMARY KEY, point BLOB)utf-8z INSERT INTO points VALUES (?, ?)z Failed to migrate dbm to sqlite:zKPlease try to use previous version of qdrant-client or re-create collection)rSTORAGE_FILE_NAME_OLDSTORAGE_FILE_NAMEexistsdbmopenstrsqlite3connectcursorexecutekeys isinstanceencodepickleloadsCollectionPersistence encode_keyBinarycommitcloseunlink Exceptionloggingerror) rdbm_pathsql_path dbm_storageconcurkeyvalue sqlite_keyer+l/home/aiteam/pcaa-dev/dashboard-backend/venv/lib/python3.10/site-packages/qdrant_client/local/persistence.pytry_migrate_to_sqlitesF         r-c@seZdZUdZedBed<edejde fddZ dde d efd d Z dd d Z dddZ dejddfddZdejddfddZdeejfddZdS)rNCHECK_SAME_THREADr'rcCstt|dS)Nr )base64 b64encoderdumpsdecode)clsr'r+r+r,r?sz CollectionPersistence.encode_keyFrforce_disable_check_same_threadcCst|t|t|_|jjjddd|jdurA|durAtd}| d d}|dk|j _Wdn1sr+r+r,reszCollectionPersistence.closecCs"|j}|d|jdS)Nr)r<rrr)r>rr+r+r,r=hs  z#CollectionPersistence._ensure_tablepointcCsD||j}t|}|j}|d|t|f|j dS)zi Persist a point in the local storage. Args: point: point to persist z+INSERT OR REPLACE INTO points VALUES (?, ?)N) ridrr1r<rrrrr)r>rBr'r(rr+r+r,persistms   zCollectionPersistence.persistpoint_idcCs0||}|j}|d|f|jdS)zv Delete a point from the local storage. Args: point_id: id of the point to delete zDELETE FROM points WHERE id = ?N)rr<rrr)r>rEr'rr+r+r,deletes  zCollectionPersistence.deleteccs8|j}|d|D] }t|dVqdS)zg Load a point from the local storage. Returns: point: loaded point zSELECT point FROM pointsrN)r<rrfetchallrr)r>rrowr+r+r,loads    zCollectionPersistence.load)FrN)__name__ __module__ __qualname__r.bool__annotations__ classmethodrExtendedPointIdrrrArr= PointStructrDrFrrIr+r+r+r,r<s  " rcCsddl}|S}t|}tjdgdddid}|||D]}||ks*J~t|}|D]}||kss   -`