Пример #1
0
class BTTVEmoteManager:
    def __init__(self, emote_manager):
        from pajbot.apiwrappers import BTTVApi
        self.emote_manager = emote_manager
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        self.channel_emotes = []

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')
        global_emotes = self.bttv_api.get_global_emotes()
        channel_emotes = self.bttv_api.get_channel_emotes(
            self.emote_manager.streamer)

        self.global_emotes = [emote['code'] for emote in global_emotes]
        self.channel_emotes = [emote['code'] for emote in channel_emotes]

        self.emote_manager.bot.mainthread_queue.add(
            self._add_bttv_emotes, args=[global_emotes + channel_emotes])

    def _add_bttv_emotes(self, emotes):
        for emote in emotes:
            key = 'custom_{}'.format(emote['code'])
            if key in self.emote_manager.data:
                self.emote_manager.data[key].emote_hash = emote['emote_hash']
            else:
                self.emote_manager.add_emote(**emote)
        log.debug('Added {} emotes'.format(len(emotes)))
Пример #2
0
class BTTVEmoteManager:
    def __init__(self, emote_manager):
        from pajbot.apiwrappers import BTTVApi

        self.emote_manager = emote_manager
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        self.channel_emotes = []

    def update_emotes(self):
        log.debug("Updating BTTV Emotes...")
        global_emotes = self.bttv_api.get_global_emotes()
        channel_emotes = self.bttv_api.get_channel_emotes(self.emote_manager.streamer)

        self.global_emotes = [emote["code"] for emote in global_emotes]
        self.channel_emotes = [emote["code"] for emote in channel_emotes]

        self.emote_manager.bot.mainthread_queue.add(self._add_bttv_emotes, args=[global_emotes + channel_emotes])

    def _add_bttv_emotes(self, emotes):
        for emote in emotes:
            key = "custom_{}".format(emote["code"])
            if key in self.emote_manager.data:
                self.emote_manager.data[key].emote_hash = emote["emote_hash"]
            else:
                self.emote_manager.add_emote(**emote)
        log.debug("Added {} emotes".format(len(emotes)))
Пример #3
0
 def __init__(self):
     from pajbot.apiwrappers import BTTVApi
     self.bttv_api = BTTVApi()
     self.global_emotes = []
     streamer = StreamHelper.get_streamer()
     redis = RedisManager.get()
     self.channel_emotes = redis.lrange(
         '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer),
         0, -1)
     self.all_emotes = []
Пример #4
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        _global_emotes = RedisManager.get().hgetall('global:emotes:bttv')
        try:
            _channel_emotes = RedisManager.get().hgetall('{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer))
        except:
            _channel_emotes = {}

        self.channel_emotes = list(_channel_emotes.keys())

        _all_emotes = _global_emotes.copy()
        _all_emotes.update(_channel_emotes)

        self.all_emotes = []
        for emote_code, emote_hash in _all_emotes.items():
            self.all_emotes.append(self.build_emote(emote_code, emote_hash))

    def build_emote(self, emote_code, emote_hash):
        return {
                'code': emote_code,
                'type': 'bttv',
                'emote_hash': emote_hash,
                'regex': re.compile('(?<![^ ]){0}(?![^ ])'.format(re.escape(emote_code))),
                }

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')
        global_emotes = self.api.get_global_emotes()
        channel_emotes = self.api.get_channel_emotes(StreamHelper.get_streamer())

        self.global_emotes = [emote['code'] for emote in global_emotes]
        self.channel_emotes = [emote['code'] for emote in channel_emotes]

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote in channel_emotes:
                pipeline.hset(key, emote['code'], emote['emote_hash'])

        self.all_emotes = []
        with RedisManager.pipeline_context() as pipeline:
            for emote in global_emotes + channel_emotes:
                # Store all possible emotes, with their regex in an easily
                # accessible list.
                self.all_emotes.append(self.build_emote(emote['code'], emote['emote_hash']))

                # Make sure all available emotes are available in redis
                pipeline.hset('global:emotes:bttv', emote['code'], emote['emote_hash'])
Пример #5
0
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()

        # Key = Emote code (i.e. KKonaW)
        # Value = Emote hash (i.e. ghjkfghfhjg23fhjg23fh34)
        self.global_emotes = {}
        self.channel_emotes = {}

        # Proper syntax described in build_emote
        self.valid_emotes = []

        self.update_global_emotes()
        self.load_cached_channel_emotes()
        self.update_valid_emotes()
Пример #6
0
    def __init__(self, emote_manager):
        from pajbot.apiwrappers import BTTVApi

        self.emote_manager = emote_manager
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        self.channel_emotes = []
Пример #7
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        redis = RedisManager.get()
        self.channel_emotes = redis.lrange(
            '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer),
            0, -1)
        self.all_emotes = []

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')
        global_emotes = self.bttv_api.get_global_emotes()
        channel_emotes = self.bttv_api.get_channel_emotes(
            StreamHelper.get_streamer())

        self.global_emotes = [emote['code'] for emote in global_emotes]
        self.channel_emotes = [emote['code'] for emote in channel_emotes]

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote in self.channel_emotes:
                pipeline.rpush(key, emote)

        self.all_emotes = []
        with RedisManager.pipeline_context() as pipeline:
            for emote in global_emotes + channel_emotes:
                # Store all possible emotes, with their regex in an easily
                # accessible list.
                self.all_emotes.append({
                    'code':
                    emote['code'],
                    'emote_hash':
                    emote['emote_hash'],
                    'regex':
                    re.compile('(?<![^ ]){0}(?![^ ])'.format(
                        re.escape(emote['code']))),
                })

                # Make sure all available emotes are available in redis
                pipeline.hset('global:emotes:bttv', emote['code'],
                              emote['emote_hash'])
Пример #8
0
 def __init__(self):
     from pajbot.apiwrappers import BTTVApi
     self.bttv_api = BTTVApi()
     self.global_emotes = []
     streamer = StreamHelper.get_streamer()
     redis = RedisManager.get()
     self.channel_emotes = redis.lrange('{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer), 0, -1)
     self.all_emotes = []
Пример #9
0
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        _global_emotes = RedisManager.get().hgetall('global:emotes:bttv')
        try:
            _channel_emotes = RedisManager.get().hgetall(
                '{streamer}:emotes:bttv_channel_emotes'.format(
                    streamer=streamer))
        except:
            _channel_emotes = {}

        self.channel_emotes = list(_channel_emotes.keys())

        _all_emotes = _global_emotes.copy()
        _all_emotes.update(_channel_emotes)

        self.all_emotes = []
        for emote_code, emote_hash in _all_emotes.items():
            self.all_emotes.append(self.build_emote(emote_code, emote_hash))
Пример #10
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        redis = RedisManager.get()
        self.channel_emotes = redis.lrange('{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer), 0, -1)
        self.all_emotes = []

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')
        global_emotes = self.bttv_api.get_global_emotes()
        channel_emotes = self.bttv_api.get_channel_emotes(StreamHelper.get_streamer())

        self.global_emotes = [emote['code'] for emote in global_emotes]
        self.channel_emotes = [emote['code'] for emote in channel_emotes]

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote in self.channel_emotes:
                pipeline.rpush(key, emote)

        self.all_emotes = []
        with RedisManager.pipeline_context() as pipeline:
            for emote in global_emotes + channel_emotes:
                # Store all possible emotes, with their regex in an easily
                # accessible list.
                self.all_emotes.append({
                    'code': emote['code'],
                    'emote_hash': emote['emote_hash'],
                    'regex': re.compile('(?<![^ ]){0}(?![^ ])'.format(re.escape(emote['code']))),
                    })

                # Make sure all available emotes are available in redis
                pipeline.hset('global:emotes:bttv', emote['code'], emote['emote_hash'])
Пример #11
0
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()

        # Key = Emote code (i.e. KKonaW)
        # Value = Emote hash (i.e. ghjkfghfhjg23fhjg23fh34)
        self.global_emotes = {}
        self.channel_emotes = {}

        # Proper syntax described in build_emote
        self.valid_emotes = []

        self.update_global_emotes()
        self.load_cached_channel_emotes()
        self.update_valid_emotes()
Пример #12
0
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.bttv_api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        _global_emotes = RedisManager.get().hgetall('global:emotes:bttv')
        try:
            _channel_emotes = RedisManager.get().hgetall('{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer))
        except:
            _channel_emotes = {}

        self.channel_emotes = list(_channel_emotes.keys())

        _all_emotes = _global_emotes.copy()
        _all_emotes.update(_channel_emotes)

        self.all_emotes = []
        for emote_code, emote_hash in _all_emotes.items():
            self.all_emotes.append(self.build_emote(emote_code, emote_hash))
Пример #13
0
 def __init__(self, emote_manager):
     from pajbot.apiwrappers import BTTVApi
     self.emote_manager = emote_manager
     self.bttv_api = BTTVApi()
     self.global_emotes = []
     self.channel_emotes = []
Пример #14
0
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi

        self.api = BTTVApi()
        super().__init__()
Пример #15
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()
        self.global_emotes = []
        streamer = StreamHelper.get_streamer()
        _global_emotes = RedisManager.get().hgetall('global:emotes:bttv')
        try:
            _channel_emotes = RedisManager.get().hgetall(
                '{streamer}:emotes:bttv_channel_emotes'.format(
                    streamer=streamer))
        except:
            _channel_emotes = {}

        self.channel_emotes = list(_channel_emotes.keys())

        _all_emotes = _global_emotes.copy()
        _all_emotes.update(_channel_emotes)

        self.all_emotes = []
        for emote_code, emote_hash in _all_emotes.items():
            self.all_emotes.append(self.build_emote(emote_code, emote_hash))

    def build_emote(self, emote_code, emote_hash):
        return {
            'code':
            emote_code,
            'type':
            'bttv',
            'emote_hash':
            emote_hash,
            'regex':
            re.compile('(?<![^ ]){0}(?![^ ])'.format(re.escape(emote_code))),
        }

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')
        global_emotes = self.api.get_global_emotes()
        channel_emotes = self.api.get_channel_emotes(
            StreamHelper.get_streamer())

        self.global_emotes = [emote['code'] for emote in global_emotes]
        self.channel_emotes = [emote['code'] for emote in channel_emotes]

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote in channel_emotes:
                pipeline.hset(key, emote['code'], emote['emote_hash'])

        self.all_emotes = []
        with RedisManager.pipeline_context() as pipeline:
            for emote in global_emotes + channel_emotes:
                # Store all possible emotes, with their regex in an easily
                # accessible list.
                self.all_emotes.append(
                    self.build_emote(emote['code'], emote['emote_hash']))

                # Make sure all available emotes are available in redis
                pipeline.hset('global:emotes:bttv', emote['code'],
                              emote['emote_hash'])
Пример #16
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()

        # Key = Emote code (i.e. KKonaW)
        # Value = Emote hash (i.e. ghjkfghfhjg23fhjg23fh34)
        self.global_emotes = {}
        self.channel_emotes = {}

        # Proper syntax described in build_emote
        self.valid_emotes = []

        self.update_global_emotes()
        self.load_cached_channel_emotes()
        self.update_valid_emotes()

    def load_cached_channel_emotes(self):
        streamer = StreamHelper.get_streamer()

        # Try to get the list of global emotes from redis
        try:
            _channel_emotes = RedisManager.get().hgetall(
                '{streamer}:emotes:bttv_channel_emotes'.format(
                    streamer=streamer))
        except:
            _channel_emotes = {}

        for code in _channel_emotes:
            emote_hash = _channel_emotes[code]
            self.channel_emotes[code] = emote_hash

    def update_global_emotes(self):
        # Try to get cached global emotes from redis
        _global_emotes = RedisManager.get().get('global:emotes:bttv_global')
        if _global_emotes and len(_global_emotes) > 0:
            log.info('Got cached BTTV global emotes!')
            _global_emotes = json.loads(_global_emotes)
        else:
            _global_emotes = self.api.get_global_emotes()
            if _global_emotes and len(_global_emotes) > 0:
                RedisManager.get().setex('global:emotes:bttv_global',
                                         time=3600,
                                         value=json.dumps(_global_emotes,
                                                          separators=(',',
                                                                      ':')))

        if _global_emotes and len(_global_emotes) > 0:
            self.global_emotes = {}
            for emote in _global_emotes:
                self.global_emotes[emote['code']] = emote['emote_hash']

    def update_channel_emotes(self):
        _channel_emotes = self.api.get_channel_emotes(
            StreamHelper.get_streamer())
        self.channel_emotes = {}
        for emote in _channel_emotes:
            self.channel_emotes[emote['code']] = emote['emote_hash']

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote_code, emote_hash in self.channel_emotes.items():
                pipeline.hset(key, emote_code, emote_hash)

    def update_valid_emotes(self):
        self.valid_emotes = []

        for emote_code, emote_hash in self.global_emotes.items():
            self.valid_emotes.append(self.build_emote(emote_code, emote_hash))

        for emote_code, emote_hash in self.channel_emotes.items():
            self.valid_emotes.append(self.build_emote(emote_code, emote_hash))

    def build_emote(self, emote_code, emote_hash):
        return {
            'code':
            emote_code,
            'type':
            'bttv',
            'emote_hash':
            emote_hash,
            'regex':
            re.compile('(?<![^ ]){0}(?![^ ])'.format(re.escape(emote_code))),
        }

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')

        self.update_global_emotes()
        self.update_channel_emotes()
        self.update_valid_emotes()
Пример #17
0
class BTTVEmoteManager:
    def __init__(self):
        from pajbot.apiwrappers import BTTVApi
        self.api = BTTVApi()

        # Key = Emote code (i.e. KKonaW)
        # Value = Emote hash (i.e. ghjkfghfhjg23fhjg23fh34)
        self.global_emotes = {}
        self.channel_emotes = {}

        # Proper syntax described in build_emote
        self.valid_emotes = []

        self.update_global_emotes()
        self.load_cached_channel_emotes()
        self.update_valid_emotes()

    def load_cached_channel_emotes(self):
        streamer = StreamHelper.get_streamer()

        # Try to get the list of global emotes from redis
        try:
            _channel_emotes = RedisManager.get().hgetall('{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer))
        except:
            _channel_emotes = {}

        for code in _channel_emotes:
            emote_hash = _channel_emotes[code]
            self.channel_emotes[code] = emote_hash

    def update_global_emotes(self):
        # Try to get cached global emotes from redis
        _global_emotes = RedisManager.get().get('global:emotes:bttv_global')
        if _global_emotes and len(_global_emotes) > 0:
            log.info('Got cached BTTV global emotes!')
            _global_emotes = json.loads(_global_emotes)
        else:
            _global_emotes = self.api.get_global_emotes()
            if _global_emotes and len(_global_emotes) > 0:
                RedisManager.get().setex('global:emotes:bttv_global', time=3600, value=json.dumps(_global_emotes, separators=(',', ':')))

        if _global_emotes and len(_global_emotes) > 0:
            self.global_emotes = {}
            for emote in _global_emotes:
                self.global_emotes[emote['code']] = emote['emote_hash']

    def update_channel_emotes(self):
        _channel_emotes = self.api.get_channel_emotes(StreamHelper.get_streamer())
        self.channel_emotes = {}
        for emote in _channel_emotes:
            self.channel_emotes[emote['code']] = emote['emote_hash']

        # Store channel emotes in redis
        streamer = StreamHelper.get_streamer()
        key = '{streamer}:emotes:bttv_channel_emotes'.format(streamer=streamer)
        with RedisManager.pipeline_context() as pipeline:
            pipeline.delete(key)
            for emote_code, emote_hash in self.channel_emotes.items():
                pipeline.hset(key, emote_code, emote_hash)

    def update_valid_emotes(self):
        self.valid_emotes = []

        for emote_code, emote_hash in self.global_emotes.items():
            self.valid_emotes.append(self.build_emote(emote_code, emote_hash))

        for emote_code, emote_hash in self.channel_emotes.items():
            self.valid_emotes.append(self.build_emote(emote_code, emote_hash))

    def build_emote(self, emote_code, emote_hash):
        return {
                'code': emote_code,
                'type': 'bttv',
                'emote_hash': emote_hash,
                'regex': re.compile('(?<![^ ]){0}(?![^ ])'.format(re.escape(emote_code))),
                }

    def update_emotes(self):
        log.debug('Updating BTTV Emotes...')

        self.update_global_emotes()
        self.update_channel_emotes()
        self.update_valid_emotes()