import pickle

from apollo_utils.core.utils.logger.time_logger import time_logger

import config
import constants
from cache.base import init_redis
from logger import logger
from utils import get_new_key, get_or_update_remained_ttl, migrate_data, update_ttl


@time_logger(logger, constants.APP_NAME)
def main():
    redis_client = init_redis()

    redis_pipe = redis_client.pipeline()
    keys = redis_client.keys(config.APP_FROM_REDIS_SEARCH_KEY)

    logger.info(
        f"GOT {len(keys)} user compared tracks keys to migrate"
    )

    for key in keys:
        redis_pipe.get(key)
    get_response = redis_pipe.execute()

    set_response = migrate_data(redis_pipe, redis_client, keys, get_response)

    logger.info(f"Successfully migrated: {set_response.count(True)} \n"
                f"Failed to migrate: {set_response.count(False)} \n")

    update_ttl_response = update_ttl(redis_pipe, redis_client, keys, get_response)

    logger.info(f"Successfully updated ttl: {update_ttl_response.count(True)} \n"
                f"Failed to update ttl: {update_ttl_response.count(False)} \n")

    redis_client.close()


if __name__ == "__main__":
    main()