ChCh/CloudBot
Archived
11
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Update twitch.py

Browse source 
Fix online/offline check (L21)
This commit is contained in:
Steven Smith 2013-06-21 18:49:27 +08:00
parent 79a25f2600
commit cc5c81544d
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
plugins/twitch.py
View file

@ -1,5 +1,6 @@
import re import re
from util import hook, http from util import hook, http
import json
twitch_re = (r'(.*:)//(twitch.tv|www.twitch.tv)(:[0-9]+)?(.*)', re.I) twitch_re = (r'(.*:)//(twitch.tv|www.twitch.tv)(:[0-9]+)?(.*)', re.I)
valid = set('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_/') valid = set('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_/')
@ -17,10 +18,11 @@ def twitch_url(match):
title = soup.findAll('span', {'class': 'real_title js-title'})[0].text title = soup.findAll('span', {'class': 'real_title js-title'})[0].text
except IndexError: except IndexError:
return "That user has no stream or videos." return "That user has no stream or videos."
online = True isplaying = http.get_json("http://api.justin.tv/api/stream/list.json?channel=%s" % location)
try: if isplaying:
isplaying = soup.findAll('a', {'data-content_type': 'live'})[0]; online = True
except IndexError: watchers = isplaying[0]["channel_count"]
else:
online = False online = False
try: try:
name = soup.findAll('a', {'class': 'channel_name'})[0].text name = soup.findAll('a', {'class': 'channel_name'})[0].text
@ -33,9 +35,9 @@ def twitch_url(match):
np = True np = True
if online: if online:
if np: if np:
return u"%s: %s playing %s (\x033\x02Online now!\x02\x0f)" % (title, name, playing) return u"%s: %s playing %s (\x033\x02Online now!\x02\x0f %s viewers)" % (title, name, playing, watchers)
else: else:
return u"%s: %s (\x033\x02Online now!\x02\x0f)" % (title, name) return u"%s: %s (\x033\x02Online now!\x02\x0f %s viewers)" % (title, name, watchers)
else: else:
if np: if np:
return u"%s: %s playing %s (\x034\x02Offline\x02\x0f)" % (title, name, playing) return u"%s: %s playing %s (\x034\x02Offline\x02\x0f)" % (title, name, playing)