Function save_db_blocks

lib/core.py:711–769

(session_id:str)

        print(error)

def save_db_blocks(session_id:str)->None:
    try:
        session = context.get_session(session_id)
        if not (session and session.get('id', False)):
            return
        data = session['blocks_current']
        if not data:
            return
        db_path = session['blocks_current_db']
        create_db_blocks(db_path)
        with sqlite3.connect(db_path) as conn:
            conn.execute('PRAGMA foreign_keys=ON')
            conn.execute(
                'UPDATE stamp SET page=?, block_resume=?, sentence_resume=?, voice=?, tts_engine=?, fine_tuned=? WHERE id=1',
                (
                    data.get('page'),
                    data.get('block_resume'),
                    data.get('sentence_resume'),
                    data.get('voice'),
                    data.get('tts_engine'),
                    data.get('fine_tuned'),
                )
            )
            new_blocks = data.get('blocks', [])
            new_ids = {b['id'] for b in new_blocks}
            existing_ids = {row[0] for row in conn.execute('SELECT id FROM blocks')}
            removed = existing_ids - new_ids
            if removed:
                conn.executemany('DELETE FROM blocks WHERE id=?', [(rid,) for rid in removed])
            for idx, block in enumerate(new_blocks):
                block_id = block['id']
                conn.execute(
                    'INSERT INTO blocks (id, idx, expand, keep, text, voice, tts_engine, fine_tuned) '
                    'VALUES (?, ?, ?, ?, ?, ?, ?, ?) '
                    'ON CONFLICT(id) DO UPDATE SET '
                    'idx=excluded.idx, expand=excluded.expand, keep=excluded.keep, text=excluded.text, '
                    'voice=excluded.voice, tts_engine=excluded.tts_engine, fine_tuned=excluded.fine_tuned',
                    (
                        block_id,
                        idx,
                        1 if block.get('expand') else 0,
                        1 if block.get('keep') else 0,
                        block.get('text', ''),
                        block.get('voice'),
                        block.get('tts_engine'),
                        block.get('fine_tuned'),
                    )
                )
                conn.execute('DELETE FROM sentences WHERE block_id=?', (block_id,))
                sentences = block.get('sentences', [])
                if sentences:
                    conn.executemany(
                        'INSERT INTO sentences (block_id, idx, text) VALUES (?, ?, ?)',
                        [(block_id, i, s) for i, s in enumerate(sentences)]
                    )
            conn.commit()
    except Exception as e:
        error = f'save_db_blocks() error: {e}'