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.
d0u9 · Dec 30, 2020 · 4326fb1 · 4326fb1
1 parent fe67a78
commit 4326fb1
Showing 1 changed file with 10 additions and 10 deletions.
diff --git 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})