Make it possible to extract video id from url

2020-07-24 14:03:07 +09:00
parent 5ab653a1b2
commit 0abf8dd9f0
8 changed files with 54 additions and 16 deletions
--- a/pytchat/tool/videoinfo.py
+++ b/pytchat/tool/videoinfo.py
@@ -3,6 +3,7 @@ import re
 import requests
 from .. import config
 from ..exceptions import InvalidVideoIdException
+from ..util.extract_video_id import extract_video_id

 headers = config.headers

@@ -78,8 +79,8 @@ class VideoInfo:
    '''

    def __init__(self, video_id):
-        self.video_id = video_id
-        text = self._get_page_text(video_id)
+        self.video_id = extract_video_id(video_id)
+        text = self._get_page_text(self.video_id)
        self._parse(text)

    def _get_page_text(self, video_id):