104 lines
14 KiB
Plaintext
104 lines
14 KiB
Plaintext
|
Ë
|
|||
|
|
®Œj“) ã óp — d dl Z d dlmZ d dlmZmZmZmZ d dlm Z d dl
|
|||
|
|
m Z d dl m
|
|||
|
|
Z
|
|||
|
|
mZmZ G d„ d« Zy) é N)ÚPath)ÚOptionalÚListÚDictÚAny)Úcontextmanager)Úlogger)ÚPRAGMA_FOREIGN_KEYSÚCREATE_TABLES_SQLÚCREATE_INDEXES_SQLc
|
|||
|
|
ó — e Zd Zd defd„Zed„ « Zdefd„Zd dede e
|
|||
|
|
de e
|
|||
|
|
fd „Z dedefd
|
|||
|
|
„Z d e
|
|||
|
|
d ed
|
|||
|
|
ede
|
|||
|
|
de
|
|||
|
|
de e
|
|||
|
|
f d„Zd
|
|||
|
|
edefd„Zde
|
|||
|
|
deeef defd„Zd dededeeeef fd„Zde
|
|||
|
|
de eeef fd„Zde
|
|||
|
|
dedede
|
|||
|
|
def
|
|||
|
|
d„Zy) Ú DBManagerÚdb_pathc ó2 — || _ | j « y ©N)r Ú_initialize_db)Úselfr s ú:C:\Users\dimir\proects\ComfyGallery\database\db_manager.pyÚ__init__zDBManager.__init__ s € Ø ˆŒ Ø ×ÑÕ ó c # óª K — d } t j | j « }|j t « t j
|
|||
|
|
|_ |–— |r|j « y y # t j $ r=}t j d| j › d|› �« |r|j « ‚ d }~ww xY w# |r|j « w w xY ww)Nu* Ошибка Ñ�оединениÑ� Ñ� БД ú: ) Úsqlite3Úconnectr Úexecuter
|
|||
|
|
ÚRowÚ row_factoryÚErrorr ÚerrorÚrollbackÚclose)r ÚconnÚes r Ú_get_connectionzDBManager._get_connection s¬ è ø€ àˆð Ü—?‘? 4§<¡<Ó0ˆDØ �L‰LÔ,Ô -Ü &Ÿ{™{ˆDÔ Ø‹Jñ Ø—
|
|||
|
|
‘
|
|||
|
|
• ð øô
|
|||
|
|
�}‰}ò Ü �L‰LÐEÀdÇlÁlÀ^ÐSUÐVWÐUXÐYÔ ZÙØ—
|
|||
|
|
‘
|
|||
|
|
”Ø ûð ûñ Ø—
|
|||
|
|
‘
|
|||
|
|
• ð üs5 ‚C†A
|
|||
|
|
A( ÁCÁ(B8Á;8B3Â3B8Â8B; Â;CÃCÚreturnc óV — | j « 5 }|j « }|j t « |j t « |j « d d d « y# 1 sw Y y xY w# t j $ r"}t j d|› �« Y d }~yd }~ww xY w)NTuE КритичеÑ�каÑ� ошибка инициализации БД: F)
|
|||
|
|
r$ ÚcursorÚ
|
|||
|
|
executescriptr r Úcommitr r r Úcritical)r r" r' r# s r r zDBManager._initialize_db s„ € ð Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø×$Ñ$Ô%6Ô7Ø×$Ñ$Ô%7Ô8Ø— ‘ ”
|
|||
|
|
Ø÷ (×'Ñ'ûô �}‰}ò Ü �O‰OÐ cÐdeÐcfÐ gÔ hÜûð ús5 ‚A3 ’A A'Á A3 Á'A0Á,A3 Á0A3 Á3B(ÂB#Â#B(NÚ folder_pathÚ parent_idc ó — |sy t t |« j « j « « } | j « 5 }|j « }|j
|
|||
|
|
d||f« |j « }|s#|j
|
|||
|
|
d|f« |j « }|j « |r|d nd cd d d « S # 1 sw Y y xY w# t j $ r%}t j d|› d|› �« Y d }~y d }~ww xY w)Nz}INSERT INTO folders (path, parent_id) VALUES (?, ?) ON CONFLICT(path) DO UPDATE SET parent_id=excluded.parent_id RETURNING idz%SELECT id FROM folders WHERE path = ?r u4 Ошибка при добавлении папки r ©
|
|||
|
|
Ústrr ÚresolveÚas_posixr$ r' r Úfetchoner) r r r r )r r+ r, Únormalized_pathr" r' Úresultr# s r Ú
|
|||
|
|
add_folderzDBManager.add_folder, sç € ÙØÜ œd ;Ó /× 7Ñ 7Ó 9× BÑ BÓ DÓEˆð Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø—‘ð`à$ iÐ0ôð
|
|||
|
|
Ÿ ™ Ó*�Ù Ø—N‘NÐ#JÈ_ÐL^Ô_Ø #Ÿ_™_Ó .�FØ— ‘ ”
|
|||
|
|
Ù$*�v˜a’y°÷ (×'Ò'ûô �}‰}ò Ü �L‰LÐOÐP_ÐO`Ð`bÐcdÐbeÐfÔ gÜûð ús6 µC
|
|||
|
|
ÁA2CÂ7 C
|
|||
|
|
ÃC
|
|||
|
|
ÃC
|
|||
|
|
Ã
|
|||
|
|
C
|
|||
|
|
Ã
|
|||
|
|
DÃ D Ä Dc óª — t t |« j « j « « } | j « 5 }|j « }|j
|
|||
|
|
d|f« |j « |j dkD cd d d « S # 1 sw Y y xY w# t j $ r%}t j d|› d|› �« Y d }~yd }~ww xY w)Nz"DELETE FROM folders WHERE path = ?r u0 Ошибка при удалении папки õ из БД: F©
|
|||
|
|
r/ r r0 r1 r$ r' r r) Úrowcountr r r r )r r+ r3 r" r' r# s r Úremove_folder_by_pathz DBManager.remove_folder_by_pathB sª € Ü œd ;Ó /× 7Ñ 7Ó 9× BÑ BÓ DÓEˆð Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø—‘Ð CÀoÐEWÔXØ— ‘ ”
|
|||
|
|
Ø —‘¨Ñ*÷ (×'Ò'ûô
|
|||
|
|
�}‰}ò Ü �L‰LÐKÈOÐK\Ð\hÐijÐhkÐlÔ mÜûð úó6 ²B ÁABÂ B ÂBÂB ÂB ÂCÂ-C
|
|||
|
|
Ã
|
|||
|
|
CÚ folder_idÚfilenameÚfilepathÚsizeÚmtimec óÐ — |r|sy t t |« j « j « « } | j « 5 }|j « }|j
|
|||
|
|
d|||||f« |j « } |j « | r| d nd cd d d « S # 1 sw Y y xY w# t j $ r%}
|
|||
|
|
t j d|› d|
|
|||
|
|
› �« Y d }
|
|||
|
|
~
|
|||
|
|
y d }
|
|||
|
|
~
|
|||
|
|
ww xY w)Na�
|
|||
|
|
INSERT INTO files (folder_id, filename, filepath, size, mtime)
|
|||
|
|
VALUES (?, ?, ?, ?, ?)
|
|||
|
|
ON CONFLICT(filepath) DO UPDATE SET
|
|||
|
|
size = excluded.size,
|
|||
|
|
mtime = excluded.mtime,
|
|||
|
|
folder_id = excluded.folder_id
|
|||
|
|
RETURNING id
|
|||
|
|
r u/ Ошибка региÑ�трации файла r r. ) r r< r= r> r? r@ Únormalized_filepathr" r' r4 r# s r Ú
|
|||
|
|
register_filezDBManager.register_fileN sÐ € Ù™xØÜ !¤$ x£.×"8Ñ"8Ó":×"CÑ"CÓ"EÓ FÐð Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø—‘ðð Ð*=¸tÀUÐKô ð Ÿ ™ Ó*�Ø— ‘ ”
|
|||
|
|
Ù$*�v˜a’y°÷! (×'Ò'ûô" �}‰}ò Ü �L‰LÐJÐK^ÐJ_Ð_aÐbcÐadÐeÔ fÜûð ús6 ·B- ÁAB! B- Â!B*Â&B- Â*B- Â-C%à C à C%c óª — t t |« j « j « « } | j « 5 }|j « }|j
|
|||
|
|
d|f« |j « |j dkD cd d d « S # 1 sw Y y xY w# t j $ r%}t j d|› d|› �« Y d }~yd }~ww xY w)Nz$DELETE FROM files WHERE filepath = ?r u) Ошибка удалениÑ� файла r7 Fr8 )r r> rB r" r' r# s r Úremove_file_by_pathz DBManager.remove_file_by_pathh s¬ € Ü !¤$ x£.×"8Ñ"8Ó":×"CÑ"CÓ"EÓ FÐð Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø—‘Ð EÐH[ÐG]Ô^Ø— ‘ ”
|
|||
|
|
Ø —‘¨Ñ*÷ (×'Ò'ûô
|
|||
|
|
�}‰}ò Ü �L‰LÐDÐEXÐDYÐYeÐfgÐehÐiÔ jÜûð úr; Úfile_idÚ meta_payloadc óN — | j « 5 }|j « }|j d||j d« |j d« |j d« |j d« |j d« |j d« |j d« |j d « |j d
|
|||
|
|
« f
|
|||
|
|
« |j « d d d « y # 1 sw Y y xY w# t
|
|||
|
|
j $ r%}t j d |› d
|
|||
|
|
|› �« Y d }~yd }~ww xY w)NaL
|
|||
|
|
INSERT INTO metadata (
|
|||
|
|
file_id, prompt_json, workflow_json, positive_prompt,
|
|||
|
|
negative_prompt, seed, model_name, sampler, steps, cfg
|
|||
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|||
|
|
ON CONFLICT(file_id) DO UPDATE SET
|
|||
|
|
prompt_json = excluded.prompt_json,
|
|||
|
|
workflow_json = excluded.workflow_json,
|
|||
|
|
positive_prompt = excluded.positive_prompt,
|
|||
|
|
negative_prompt = excluded.negative_prompt,
|
|||
|
|
seed = excluded.seed,
|
|||
|
|
model_name = excluded.model_name,
|
|||
|
|
sampler = excluded.sampler,
|
|||
|
|
steps = excluded.steps,
|
|||
|
|
cfg = excluded.cfg
|
|||
|
|
Ú prompt_jsonÚ
|
|||
|
|
workflow_jsonÚpositive_promptÚnegative_promptÚseedÚ
|
|||
|
|
model_nameÚsamplerÚstepsÚcfgTuL Ошибка при Ñ�охранении метаданных файла ID r F) r$ r' r Úgetr) r r r r )r rF rG r" r' r# s r Ú
|
|||
|
|
save_metadatazDBManager.save_metadatat s € ð% Ø×%Ñ%Ô'¨4Ø Ÿ™› �Ø—‘ðð" Ø$×(Ñ(¨ Ó7Ø$×(Ñ(¨ Ó9Ø$×(Ñ(Ð):Ó;Ø$×(Ñ(Ð):Ó;Ø$×(Ñ(¨Ó0Ø$×(Ñ(¨ Ó6Ø$×(Ñ(¨Ó3Ø$×(Ñ(¨Ó1Ø$×(Ñ(¨Ó/ð ô# ð< — ‘ ”
|
|||
|
|
Ø÷C (×'Ñ'ûôD �}‰}ò Ü �L‰LÐgÐhoÐgpÐprÐstÐruÐvÔ wÜûð ús5 ‚C, ’CC ÃC, à C)Ã%C, Ã)C, Ã,D$Ã?D Ä D$Ú search_queryc ó — t t |« j « j « « } | j « 5 }|j « }|r d|› d�}|j
|
|||
|
|
d||||f« n|j
|
|||
|
|
d|f« |j « D �cg c]
|
|||
|
|
}t |« ‘Œ c}cd d d « S c c}w # 1 sw Y y xY w# t j $ r'}t j d|› d|› �« g cY d }~S d }~ww xY w)NÚ%a<
|
|||
|
|
SELECT f.id, f.filename, f.filepath, f.rating, f.favorite,
|
|||
|
|
CASE WHEN m.workflow_json IS NOT NULL AND m.workflow_json != '' THEN 1 ELSE 0 END as has_workflow
|
|||
|
|
FROM files f
|
|||
|
|
JOIN folders fo ON f.folder_id = fo.id
|
|||
|
|
LEFT JOIN metadata m ON f.id = m.file_id
|
|||
|
|
WHERE fo.path = ? AND (f.filename LIKE ? OR m.positive_prompt LIKE ? OR m.negative_prompt LIKE ?)
|