from libs.db_redis import DbRedis


class RdsLiveReplaySet:
    live_replay_wait_rds = 'BrandLiveData.LiveReplayWait'

    def __init__(self):
        self.redis = DbRedis.connect()

    # 获取待下载直播的直播间信息
    def get_live_info(self):
        return self.redis.spop(self.live_replay_wait_rds)

    # 添加直播间信息至待下载直播集合
    def add_live_info(self, data):
        return self.redis.sadd(self.live_replay_wait_rds, data)