some safety and optimizations

movie_list/index.py

@@ -11,7 +11,7 @@ import os
 import re
 import time
 import sys
-import http.client
+from http.client import IncompleteRead
 import threading
 from pathlib import Path
 import progressbar
@@ -63,7 +63,7 @@ class MovieList:
                 try:
                     query = imdb.search_movie(arg)
                     break
-                except http.client.IncompleteRead:
+                except IncompleteRead:
                     pass
         sys.stdout = save_stdout
 
@@ -71,16 +71,16 @@ class MovieList:
         imdb.update(movie)
         movie['ID'] = query[0].movieID
         if 'genres' not in movie.keys():
-            movie['genres'] = 'N/A'
+            movie['genres'] = ['N/A']
         if 'rating' not in movie.keys():
             movie['rating'] = 'N/A'
 
         url = f'http://www.imdb.com/title/tt{movie["ID"]}'
-        title = str(movie['title'])
+        title = movie['title']
-        year = str(movie['year'])
+        year = movie['year']
-        rating = str(movie['rating'])
+        rating = movie['rating']
         genres = ', '.join(movie['genres'])
-        status = str(self.status_list[index])
+        status = self.status_list[index]
         self.html += (
             f'\n\t<tr><td>{index + 1}</td>'
             f'<td><p hidden>{title}</p><a href="{url}" target="_blank">{title}</a></td>'