Function verify_migration

migrate_to_postgres.py:140–170

Verify the migration by comparing document counts

(logger)

def verify_migration(logger):
    """Verify the migration by comparing document counts"""
    from questions.db_models import Document as NDBDocument
    from questions.db_models import client
    from questions.db_models_postgres import Document as PGDocument
    from questions.db_models_postgres import SessionLocal

    logger.info("Verifying migration...")

    db = SessionLocal()

    try:
        # Count PostgreSQL records
        pg_doc_count = db.query(PGDocument).count()

        # Count NDB records
        with client.context():
            ndb_doc_count = NDBDocument.query().count()

        logger.info("Migration verification:")
        logger.info(f"  Documents: NDB={ndb_doc_count}, PostgreSQL={pg_doc_count}")

        if pg_doc_count >= ndb_doc_count:
            logger.info("✅ Migration verification successful!")
            return True
        else:
            logger.warning("⚠️  Migration may be incomplete - counts don't match")
            return False

    finally:
        db.close()


def main():