Merge pull request #37 from vmonkey/master

Quick fix for db entries for non youtube videos
d0u9 · Jan 27, 2021 · 8d3871b · 8d3871b
2 parents fe67a78 + 4326fb1
commit 8d3871b
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})