def __init__(self): super().__init__() self._store = PersistentStore() # request cancer levels self.context = zmq.Context() self.poller = zmq.Poller() self._connect()
def __init__(self): super().__init__() self._store = PersistentStore() # subscribe to summaries from the publisher socket self.context = zmq.Context() self.socket = self.context.socket(zmq.SUB) self.socket.setsockopt(zmq.SUBSCRIBE, b'summary') read_socket = Config.get( 'monitor.socket.read.cancer_summary') or Config.get( 'monitor.socket.cancer_summary') self.socket.connect(read_socket) logger.info("connected summary socket to %s", read_socket)
""" Add Twitch channel id to leaderboards """ import sys from twitchcancer.storage.persistentstore import PersistentStore from twitchcancer.utils.twitchapi import TwitchApi if __name__ == '__main__': store = PersistentStore() leaderboards = store.db.leaderboard.find({'channel': { '$regex': '#.*' }}, limit=100) names = [ leaderboard['channel'].replace('#', '') for leaderboard in leaderboards ] if not names: print('Nothing to migrate') sys.exit() users = TwitchApi.request('https://api.twitch.tv/kraken/users?login='******','.join(names))["users"] for user in users: channel_name = user['name'] channel_id = user['_id'] query = {'channel': '#' + channel_name} update = {'$set': {'channel': channel_name, 'channel_id': channel_id}} print('Migrating {} to {}'.format(channel_name, channel_id)) store.db.daily_leaderboard.update_one(query, update)
def setUp(self, init): self.store = PersistentStore() self.store.db = MagicMock()