77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
|
import requests
|
||
|
|
import time
|
||
|
|
import threading
|
||
|
|
|
||
|
|
import cherrypy
|
||
|
|
|
||
|
|
class TwitchClient:
|
||
|
|
def __init__(self, token, freq=2):
|
||
|
|
self.token = token
|
||
|
|
self.lock = threading.Lock()
|
||
|
|
|
||
|
|
self.header_v5 = {'Client-ID': self.token, 'Accept': 'application/vnd.twitchtv.v5+json'}
|
||
|
|
self.header_v6 = {'Client-ID': self.token}
|
||
|
|
self.urlbase_v5 = 'https://api.twitch.tv/kraken'
|
||
|
|
self.urlbase_v6 = 'https://api.twitch.tv/helix'
|
||
|
|
|
||
|
|
self.last_q = time.time()
|
||
|
|
self.delay = 1/freq
|
||
|
|
|
||
|
|
|
||
|
|
def do_q(self, base, header):
|
||
|
|
self.lock.acquire()
|
||
|
|
try:
|
||
|
|
cherrypy.log('Request: %s' % base)
|
||
|
|
delta = time.time() - self.last_q
|
||
|
|
if delta < self.delay:
|
||
|
|
time.sleep(delta)
|
||
|
|
r = requests.get(base, headers=header).json()
|
||
|
|
self.last_q = time.time()
|
||
|
|
cherrypy.log('Request: OK')
|
||
|
|
except:
|
||
|
|
cherrypy.log('Request: FAIL')
|
||
|
|
r = None
|
||
|
|
finally:
|
||
|
|
self.lock.release()
|
||
|
|
return r
|
||
|
|
|
||
|
|
def get_base(self, ver):
|
||
|
|
if ver == 'v5': return (self.header_v5, self.urlbase_v5)
|
||
|
|
elif ver == 'v6': return (self.header_v6, self.urlbase_v6)
|
||
|
|
else: return None
|
||
|
|
|
||
|
|
# - # - #
|
||
|
|
|
||
|
|
def raw_query_v6(self, q):
|
||
|
|
header, base = self.get_base('v6')
|
||
|
|
return self.do_q(base+q, header)
|
||
|
|
|
||
|
|
def raw_query_v5(self, q):
|
||
|
|
header, base = self.get_base('v5')
|
||
|
|
return self.do_q(base+q, header)
|
||
|
|
|
||
|
|
# Returns (ID, GAMENAME) or None
|
||
|
|
def get_game_id(self, name):
|
||
|
|
header, base = self.get_base('v5')
|
||
|
|
r = self.do_q('%s/search/games?query=%s' % (base, name), header)
|
||
|
|
if r and r.get('games'): return (r['games'][0]['_id'], r['games'][0]['name'])
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_live_streams(self, name, lang):
|
||
|
|
header, base = self.get_base('v5')
|
||
|
|
data = self.do_q('%s/streams/?game=%s&language=%s&limit=%s&stream_type=live' % (base, name, lang, 100), header)
|
||
|
|
if data is None: return None
|
||
|
|
total = data['_total']; streams = len(data['streams'])
|
||
|
|
|
||
|
|
while total > streams:
|
||
|
|
r = self.do_q('%s/streams/?game=%s&language=%s&limit=%s&stream_type=live&offset=%s' % (base, name, lang, 100, streams), header)
|
||
|
|
if r is None: return None
|
||
|
|
data['streams'].extend(r['streams'])
|
||
|
|
total = r['_total']; streams = len(data['streams'])
|
||
|
|
|
||
|
|
# Tweak for getting only live sterams
|
||
|
|
data['streams'] = [x for x in data['streams'] if x['stream_type'] == 'live']
|
||
|
|
data['_total'] = len(data['streams'])
|
||
|
|
return data
|
||
|
|
|