feat: add video dict (#48)

Author: timerring

bilitool/controller/feed_controller.py

@@ -16,3 +16,6 @@ def print_video_info(self, vid: str):
         video_info = self.bili_video_list.get_video_info(bvid)
         extracted_info = self.bili_video_list.extract_video_info(video_info)
         self.bili_video_list.print_video_info(extracted_info)
+
+    def get_video_dict_info(self, size: int = 20, status_type: str = 'pubed,not_pubed,is_pubing'):
+        return self.bili_video_list.get_video_dict_info(size, status_type)

bilitool/controller/upload_controller.py

@@ -80,20 +80,24 @@ def publish_video(self, file):
         if publish_video_response['code'] == 0:
             bvid = publish_video_response['data']['bvid']
             self.logger.info(f'upload success!\tbvid:{bvid}')
+            return True
         else:
             self.logger.error(publish_video_response['message'])
+            return False
         # reset the video title
         Model().reset_upload_config()
 
-    def append_video_entry(self, video_path, bvid):
+    def append_video_entry(self, video_path, bvid, video_name=None):
         bilibili_filename = self.upload_video(video_path)
-        video_name = Path(video_path).name.strip(".mp4")
+        video_name = video_name if video_name else Path(video_path).name.strip(".mp4")
         video_data = self.bili_uploader.get_video_list_info(bvid)
         response = self.bili_uploader.append_video(bilibili_filename, video_name, video_data)
         if response['code'] == 0:
             self.logger.info(f'append {video_name} to {bvid} success!')
+            return True
         else:
             self.logger.error(response['message'])
+            return False
         # reset the video title
         Model().reset_upload_config()
 
@@ -107,5 +111,5 @@ def upload_video_entry(self, video_path, yaml, copyright, tid, title, desc, tag,
                 desc, tag, source, cover, dynamic
             )
         Model().update_multiple_config('upload', upload_metadata)
-        self.publish_video(video_path)
+        return self.publish_video(video_path)
 

bilitool/feed/bili_video_list.py

@@ -65,6 +65,13 @@ def print_video_list_info(self, size: int = 20, status_type: str = 'pubed,not_pu
                 info += f" | {' | '.join(extra_info)}"
             print(info)
 
+    def get_video_dict_info(self, size: int = 20, status_type: str = 'pubed,not_pubed,is_pubing'):
+        video_data = self.get_bili_video_list(size, status_type)
+        data = dict()
+        for item in video_data['items']:
+            data[item['title']] = item['bvid']
+        return data
+
     def get_video_info(self, bvid: str) -> dict:
         """Get the video info of the bvid"""
         url = f"https://api.bilibili.com/x/web-interface/view?bvid={bvid}"

bilitool/upload/bili_upload.py

@@ -289,8 +289,6 @@ def append_video(self, bilibili_filename, video_name, data):
             'desc': "",
         }
         data['videos'].append(video_to_be_appended)
-        print(data)
-        print(type(data))
         headers = {
             'Connection': 'keep-alive',
             'Content-Type': 'application/json',