Movie and TV ratings scraper - fixing lint with flake8 errors

Author: walkhenj

movie_tv_ratings/movie_tv_ratings.py

@@ -1,36 +1,35 @@
 import requests
 import re
-import sys
 from bs4 import BeautifulSoup
 
-##### Gets top 250 movies from IMDB
+# Gets top 250 movies from IMDB
 def scrape_movies():
+
     response = requests.get('http://www.imdb.com/chart/top')
     soup = BeautifulSoup(response.text, 'lxml')
 
     movies = soup.select('td.titleColumn')
-    ratings = [b.attrs.get('data-value')
-           for b in soup.select('td.posterColumn span[name=ir]')]
+    ratings = [b.attrs.get('data-value') for b in soup.select('td.posterColumn span[name=ir]')]
 
     for i in range(len(movies)):
         movie_string = movies[i].get_text()
         movie = (' '.join(movie_string.split()).replace('.', ''))
         movie_title = movie[len(str(i))+1:-7]
-        print("| " + str(i+1) + " | " + movie_title + " | Rating : " + "{:.1f}".format(float(ratings[i])))
-    return 
+        print(("| " + str(i+1)) + (" | " + movie_title) + (" | Rating : " + "{:.1f}".format(float(ratings[i]))))
+    return
 
 
-##### Gets top 250 TV shows from IMDB
+# Gets top 250 TV shows from IMDB
 def scrape_tvshows():
     page = requests.get("https://www.imdb.com/chart/toptv")
     Results = re.findall(r'" alt="(.+?)".*?title="(.*?)".*?strong.*?"(.*?)"', page.text, re.DOTALL)
     for i in range(len(Results)):
         print("| " + str(i+1) + " | " + Results[i][0] + " | Rating : " + Results[i][-1][:3])
-    
-    return 
+
+    return
 
 
-##### USER INTERFACE #####
+# USER INTERFACE
 
 print("Type 'Movies' to get the Top 250 Movies on IMDB\n")
 print("Type 'TV' to get the Top 250 TV Shows on IMDB\n")
@@ -40,17 +39,13 @@ def scrape_tvshows():
 while (val):
     if val == 'Movies':
         globals()['scrape_movies']()
-
         print("\n")
-        
         val = input("Type 'Movies' or 'TV' or 'exit': ")  
-
     elif val == 'TV':
         globals()['scrape_tvshows']()
         print("\n")
         val = input("Type 'Movies' or 'TV' or 'exit': ") 
-
     elif val == 'exit':
         val = ''
     else:
-        val = input("Wrong Input. Try Again: ")
+        val = input("Wrong Input. Try Again: ")