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()
