Now API is used with constant frequency
This commit is contained in:
parent
c7801b81f8
commit
350d2b8c3e
1 changed files with 16 additions and 4 deletions
|
@ -1,8 +1,8 @@
|
||||||
import json
|
|
||||||
import requests
|
import requests
|
||||||
|
import time
|
||||||
|
|
||||||
class TwitchClient:
|
class TwitchClient:
|
||||||
def __init__(self, token):
|
def __init__(self, token, freq=2):
|
||||||
self.token = token
|
self.token = token
|
||||||
|
|
||||||
self.header_v5 = {'Client-ID': self.token, 'Accept': 'application/vnd.twitchtv.v5+json'}
|
self.header_v5 = {'Client-ID': self.token, 'Accept': 'application/vnd.twitchtv.v5+json'}
|
||||||
|
@ -11,19 +11,31 @@ class TwitchClient:
|
||||||
self.urlbase_v5 = 'https://api.twitch.tv/kraken/'
|
self.urlbase_v5 = 'https://api.twitch.tv/kraken/'
|
||||||
self.urlbase_v6 = 'https://api.twitch.tv/helix/'
|
self.urlbase_v6 = 'https://api.twitch.tv/helix/'
|
||||||
|
|
||||||
|
self.last_q = time.time()
|
||||||
|
self.delay = 1/freq
|
||||||
|
|
||||||
|
def do_q(self, base, header):
|
||||||
|
if time.time() - self.last_q < self.delay:
|
||||||
|
time.sleep(self.delay)
|
||||||
|
r = requests.get(base, headers=header).json()
|
||||||
|
self.last_q = time.time()
|
||||||
|
return r
|
||||||
|
|
||||||
def get_base(self, ver):
|
def get_base(self, ver):
|
||||||
if ver == 'v5': return (self.header_v5, self.urlbase_v5)
|
if ver == 'v5': return (self.header_v5, self.urlbase_v5)
|
||||||
elif ver == 'v6': return (self.header_v6, self.urlbase_v6)
|
elif ver == 'v6': return (self.header_v6, self.urlbase_v6)
|
||||||
else: return (None, None)
|
else: return (None, None)
|
||||||
|
|
||||||
|
# - # - #
|
||||||
|
|
||||||
def raw_query_v6(self, q):
|
def raw_query_v6(self, q):
|
||||||
header, base = self.get_base('v6')
|
header, base = self.get_base('v6')
|
||||||
return requests.get(base+q, headers=header).json()
|
return self.do_q(base+q, header)
|
||||||
|
|
||||||
# Returns (ID, GAMENAME) or None
|
# Returns (ID, GAMENAME) or None
|
||||||
def get_game_id(self, name):
|
def get_game_id(self, name):
|
||||||
header, base = self.get_base('v5')
|
header, base = self.get_base('v5')
|
||||||
r = requests.get('%s/search/games?query=%s' % (base, name), headers=header).json()
|
r = self.do_q('%s/search/games?query=%s' % (base, name), header)
|
||||||
if r.get('games', None): return (r['games'][0]['_id'], r['games'][0]['name'])
|
if r.get('games', None): return (r['games'][0]['_id'], r['games'][0]['name'])
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue