From 4326fb1ea589c7b24cb297ee6eba4a7b615c12d7 Mon Sep 17 00:00:00 2001 From: vmonkey Date: Wed, 30 Dec 2020 08:44:44 +0100 Subject: [PATCH] Quick fix for db entries for non youtube videos Some db data may not be present in other web video services. Defaulting to empty info rather than to an error. --- youtube_dl_webui/db.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/youtube_dl_webui/db.py b/youtube_dl_webui/db.py index 7edd610..080986c 100644 --- a/youtube_dl_webui/db.py +++ b/youtube_dl_webui/db.py @@ -250,16 +250,16 @@ def update_info(self, tid, info_dict): self.logger.debug('db update_info()') db_data = { 'valid': 1, # info_dict is updated - 'title': info_dict['title'], - 'format': info_dict['format'], - 'ext': info_dict['ext'], - 'thumbnail': info_dict['thumbnail'], - 'duration': info_dict['duration'], - 'view_count': info_dict['view_count'], - 'like_count': info_dict['like_count'], - 'dislike_count': info_dict['dislike_count'], - 'average_rating': info_dict['average_rating'], - 'description': info_dict['description'], + 'title': info_dict.get('title',''), + 'format': info_dict.get('format',''), + 'ext': info_dict.get('ext',''), + 'thumbnail': info_dict.get('thumbnail',''), + 'duration': info_dict.get('duration',''), + 'view_count': info_dict.get('view_count',''), + 'like_count': info_dict.get('like_count',''), + 'dislike_count': info_dict.get('dislike_count',''), + 'average_rating': info_dict.get('average_rating',''), + 'description': info_dict.get('description',''), } self.update(tid, {'task_info': db_data})