Fix potential type error when processing Tudou danmaku.

This commit is contained in:
John Chong 2016-05-09 09:53:28 +08:00
parent da21a8b540
commit 7d623dc857

View File

@ -213,10 +213,10 @@ def ReadCommentsTudou2(f, fontsize):
try: try:
c = str(comment['content']) c = str(comment['content'])
prop = json.loads(str(comment['propertis']) or '{}') prop = json.loads(str(comment['propertis']) or '{}')
size = prop.get('size', 1) size = int(prop.get('size', 1))
assert size in (0, 1, 2) assert size in (0, 1, 2)
size = {0: 0.64, 1: 1, 2: 1.44}[size] * fontsize size = {0: 0.64, 1: 1, 2: 1.44}[size] * fontsize
pos = prop.get('pos', 3) pos = int(prop.get('pos', 3))
assert pos in (0, 3, 4, 6) assert pos in (0, 3, 4, 6)
yield ( yield (
int(comment['playat'] * 0.001), int(comment['createtime'] * 0.001), i, c, int(comment['playat'] * 0.001), int(comment['createtime'] * 0.001), i, c,