67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
"""
|
||
pytchat.parser.live
|
||
~~~~~~~~~~~~~~~~~~~
|
||
This module is parser of live chat JSON.
|
||
"""
|
||
|
||
import json
|
||
from .. import config
|
||
from .. import mylogger
|
||
from .. exceptions import (
|
||
ResponseContextError,
|
||
NoContentsException,
|
||
NoContinuationsException )
|
||
|
||
|
||
logger = mylogger.get_logger(__name__,mode=config.LOGGER_MODE)
|
||
|
||
|
||
class Parser:
|
||
def parse(self, jsn):
|
||
"""
|
||
このparse関数はLiveChat._listen() 関数から定期的に呼び出される。
|
||
引数jsnはYoutubeから取得したチャットデータの生JSONであり、
|
||
このparse関数によって与えられたJSONを以下に分割して返す。
|
||
+ timeout (次のチャットデータ取得までのインターバル)
|
||
+ chat data（チャットデータ本体）
|
||
+ continuation （次のチャットデータ取得に必要となるパラメータ）.
|
||
|
||
Parameter
|
||
----------
|
||
+ jsn : dict
|
||
+ Youtubeから取得したチャットデータのJSONオブジェクト。
|
||
（pythonの辞書形式に変換済みの状態で渡される）
|
||
|
||
Returns
|
||
-------
|
||
+ metadata : dict
|
||
+ チャットデータに付随するメタデータ。timeout、 動画ID、continuationパラメータで構成される。
|
||
+ chatdata : list[dict]
|
||
+ チャットデータ本体のリスト。
|
||
"""
|
||
if jsn is None:
|
||
return {'timeoutMs':0,'continuation':None},[]
|
||
if jsn['response']['responseContext'].get('errors'):
|
||
raise ResponseContextError('動画に接続できません。'
|
||
'動画IDが間違っているか、動画が削除／非公開の可能性があります。')
|
||
contents=jsn['response'].get('continuationContents')
|
||
#配信が終了した場合、もしくはチャットデータが取得できない場合
|
||
if contents is None:
|
||
raise NoContentsException('チャットデータを取得できませんでした。')
|
||
|
||
cont = contents['liveChatContinuation']['continuations'][0]
|
||
if cont is None:
|
||
raise NoContinuationsException('Continuationがありません。')
|
||
metadata = (cont.get('invalidationContinuationData') or
|
||
cont.get('timedContinuationData') or
|
||
cont.get('reloadContinuationData')
|
||
)
|
||
if metadata is None:
|
||
unknown = list(cont.keys())[0]
|
||
if unknown:
|
||
logger.error(f"Received unknown continuation type:{unknown}")
|
||
metadata = cont.get(unknown)
|
||
metadata.setdefault('timeoutMs', 10000)
|
||
chatdata = contents['liveChatContinuation'].get('actions')
|
||
return metadata, chatdata
|