stable/freqtrade/commands/db_commands.py

import logging
from typing import Any, Dict

from sqlalchemy import func

from freqtrade.configuration.config_setup import setup_utils_configuration
from freqtrade.enums.runmode import RunMode
from freqtrade.persistence.migrations import set_sequence_ids
from freqtrade.persistence.trade_model import Order


logger = logging.getLogger(__name__)


def start_convert_db(args: Dict[str, Any]) -> None:
    from sqlalchemy.orm import make_transient

    from freqtrade.persistence import Trade, init_db
    from freqtrade.persistence.pairlock import PairLock

    config = setup_utils_configuration(args, RunMode.UTIL_NO_EXCHANGE)

    init_db(config['db_url'], False)
    session_target = Trade._session
    init_db(config['db_url_from'], False)
    logger.info("Starting db migration.")

    # print(f"{id(sessionA)=}, {id(sessionB)=}")
    trade_count = 0
    pairlock_count = 0
    for trade in Trade.get_trades():
        trade_count += 1
        make_transient(trade)
        for o in trade.orders:
            make_transient(o)

        session_target.add(trade)

    session_target.commit()

    for pairlock in PairLock.query:
        pairlock_count += 1
        make_transient(pairlock)
        session_target.add(pairlock)
    session_target.commit()

    # Update sequences
    max_trade_id = session_target.query(func.max(Trade.id)).scalar()
    max_order_id = session_target.query(func.max(Order.id)).scalar()
    max_pairlock_id = session_target.query(func.max(PairLock.id)).scalar()

    set_sequence_ids(session_target.get_bind(),
                     trade_id=max_trade_id,
                     order_id=max_order_id,
                     pairlock_id=max_pairlock_id)

    logger.info(f"Migrated {trade_count} Trades, and {pairlock_count} Pairlocks.")