Add page title (#20)

* Add notion page title

* refactor for lint

* refactor for lint, test

* refactor for lint, test

Author: gnokoheat

.style.yapf

@@ -3,4 +3,4 @@ based_on_style = yapf
 spaces_before_comment = 4
 indent_width: 4
 split_before_logical_operator = true
-column_limit = 80
+column_limit = 79

tests/youtube2notion/test_youtube2notion.py

@@ -15,12 +15,18 @@ def setUp(self):
         output_dir = './tmp/%s/' % video_id
         notion_token_v2 = ''
         notion_page_url = ''
+        info_title = ''
+        info_author_name = ''
+        info_author_url = ''
 
         self.y2n = Youtube2notion(
             video_id=video_id,
             output_dir=output_dir,
             notion_token_v2=notion_token_v2,
-            notion_page_url=notion_page_url)
+            notion_page_url=notion_page_url,
+            info_title=info_title,
+            info_author_name=info_author_name,
+            info_author_url=info_author_url)
 
     def test_execute(self):
         self.y2n.execute()

tests/youtube2notion/test_youtube_info.py

@@ -0,0 +1,17 @@
+import unittest
+
+from youtube2notion.youtube_info import YoutubeInfo
+
+
+class TestYoutubeInfo(unittest.TestCase):
+
+    def setUp(self):
+        self.video_id = '5oNcoj4G0xc'
+
+    def test_get_information_element_1(self):
+        self.assertIsNotNone(
+            YoutubeInfo.get_information_element(self.video_id))
+
+    def test_get_information_element_2(self):
+        self.assertIsNotNone(
+            YoutubeInfo.get_information_element('Kc_cvAXCs4Y'))

youtube2notion/youtube2notion.py

@@ -2,6 +2,7 @@
 from youtube2notion.ffmpeg import Ffmpeg
 from youtube2notion.youtube_video import YoutubeVideo
 from youtube2notion.youtube_subtitle import SubtitleElement, YoutubeSubtitle
+from youtube2notion.youtube_info import InformationElement, YoutubeInfo
 from pathlib import Path
 from notion.client import NotionClient
 from notion.block import PageBlock
@@ -10,12 +11,17 @@
 
 class Youtube2notion:
 
-    def __init__(self,
-                 video_id: str,
-                 output_dir: str = '',
-                 notion_token_v2: str = '',
-                 notion_page_url: str = '',
-                 subtitle_language: str = 'ko'):
+    def __init__(
+        self,
+        video_id: str,
+        output_dir: str = '',
+        notion_token_v2: str = '',
+        notion_page_url: str = '',
+        subtitle_language: str = 'ko',
+        info_title: str = '',
+        info_author_name: str = '',
+        info_author_url: str = '',
+    ):
         self.video_id = video_id
         self.output_dir = output_dir
         self.images_output_dir = self.output_dir + 'images/'
@@ -25,13 +31,23 @@ def __init__(self,
 
         self.subtitle_language = subtitle_language
 
+        self.info_title = info_title
+        self.info_author_name = info_author_name
+        self.info_author_url = info_author_url
+
     def _download_video(self) -> str:
         return YoutubeVideo.download(self.video_id, self.output_dir)
 
     def _get_subtitle_elements(self) -> list[SubtitleElement]:
         return YoutubeSubtitle.get_subtitle_elements(self.video_id,
                                                      [self.subtitle_language])
 
+    def _get_info_element(self) -> InformationElement:
+        info_element = YoutubeInfo.get_information_element(self.video_id)
+        self.info_title = info_element.title
+        self.info_author_name = info_element.author_name
+        self.info_author_url = info_element.author_url
+
     def _take_screenshots(self, input_filename: str):
         Path(self.images_output_dir).mkdir(parents=True, exist_ok=True)
 
@@ -62,17 +78,18 @@ def _upload_to_notion(self, md_file: str, notion_token_v2: str,
         with open(md_file, 'r', encoding='utf-8') as f:
             new_page = page.children.add_new(
                 PageBlock,
-                title=self.video_id + '(' + self.subtitle_language + ')')
+                title=self.info_title + '(' + self.subtitle_language + ')')
             upload(f, new_page)
 
     def execute(self):
         subtitle_elements = self._get_subtitle_elements()
+        self._get_info_element()
 
         downloaded_video_filename = self._download_video()
         self._take_screenshots(downloaded_video_filename)
 
         md = self._generage_markdown(
-            title=self.video_id + '(' + self.subtitle_language + ')',
+            title=self.info_title + '(' + self.subtitle_language + ')',
             subtitle_elements=subtitle_elements,
             images_dir='./images/')
 

youtube2notion/youtube_info.py

@@ -0,0 +1,49 @@
+import requests
+
+
+class InformationElement:
+
+    def __init__(
+        self,
+        title: str = '',
+        author_name: str = '',
+        author_url: str = '',
+    ):
+        self._title = title
+        self._author_name = author_name
+        self._author_url = author_url
+
+    @property
+    def title(self):
+        return self._title
+
+    @property
+    def author_name(self):
+        return self._author_name
+
+    @property
+    def author_url(self):
+        return self._author_url
+
+
+class YoutubeInfo:
+
+    @classmethod
+    def get_information_element(cls, video_id: str) -> InformationElement:
+
+        params = {
+            "format": "json",
+            "url": "https://www.youtube.com/watch?v=%s" % video_id
+        }
+        res = requests.get("https://www.youtube.com/oembed", params=params)
+
+        if res.status_code == 200:
+            res_json = res.json()
+
+            return InformationElement(
+                title=res_json.get('title'),
+                author_name=res_json.get('author_name'),
+                author_url=res_json.get('author_url'),
+            )
+        else:
+            return InformationElement(title=video_id)