Refactor scraping project

Author: gerhynes

web_scraping/scraping_project.py

@@ -5,50 +5,61 @@
 from random import choice
 from csv import writer
 
-all_quotes = []
-base_url = "http://quotes.toscrape.com/"
-url = "/page/1"
+BASE_URL = "http://quotes.toscrape.com/"
 
-while url:
-    res = requests.get(f"{base_url}{url}")
-    print(f"Now scraping {base_url}{url}")
-    soup = BeautifulSoup(res.text, "html.parser")
-    quotes = soup.find_all(class_="quote")
-
-    for quote in quotes:
-        all_quotes.append({
-            "text": quote.find(class_="text").get_text(),
-            "author": quote.find(class_="author").get_text(),
-            "bio-link": quote.find("a")["href"]
-        })
-    next_btn = soup.find(class_="next")
-    url = next_btn.find("a")["href"] if next_btn else None
-    # sleep(3) # wait between scraping pages
-
-quote = choice(all_quotes)
+def scrape_quotes():
+    all_quotes = []
+    url = "/page/1"
+    while url:
+        res = requests.get(f"{BASE_URL}{url}")
+        print(f"Now scraping {BASE_URL}{url}")
+        soup = BeautifulSoup(res.text, "html.parser")
+        quotes = soup.find_all(class_="quote")
 
-guess = ""
-remaining_guesses = 4
+        for quote in quotes:
+            all_quotes.append({
+                "text": quote.find(class_="text").get_text(),
+                "author": quote.find(class_="author").get_text(),
+                "bio-link": quote.find("a")["href"]
+            })
+        next_btn = soup.find(class_="next")
+        url = next_btn.find("a")["href"] if next_btn else None
+        # sleep(3) # wait between scraping pages
+        return all_quotes
 
-print("Here's the quote: ")
-print(quote["text"])
+def start_game(quotes):
+    quote = choice(quotes)
+    guess = ""
+    remaining_guesses = 4
+    print("Here's the quote: ")
+    print(quote["text"])
+    while guess.lower() != quote["author"].lower() and remaining_guesses > 0:
+        guess = input(f"Who said this quote? Guesses remaining: {remaining_guesses}\n")
+        if guess.lower() == quote["author"].lower():
+            print("YOU GOT IT RIGHT!")
+            break
+        remaining_guesses -= 1
+        if remaining_guesses == 3:
+            res = requests.get(f"{BASE_URL}{quote['bio-link']}")
+            soup = BeautifulSoup(res.text, "html.parser")
+            birth_date = soup.find(class_="author-born-date").get_text()
+            birth_place = soup.find(class_="author-born-location").get_text()
+            print(f"Here's a hint: The author was born on {birth_date} {birth_place}")
+        elif remaining_guesses == 2:
+            print(f"Here's a hint: The author's first name starts with: {quote['author'][0]}")
+        elif remaining_guesses == 1:
+            last_initial = quote['author'].split(" ")[1][0]
+            print(f"Here's a hint: The author's last name starts with: {last_initial}")
+        else:
+            print(f"Sorry, you ran out of guesses. The answer was {quote['author']}")
 
-while guess.lower() != quote["author"].lower() and remaining_guesses > 0:
-    guess = input(f"Who said this quote? Guesses remaining: {remaining_guesses}\n")
-    if guess.lower() == quote["author"].lower():
-        print("YOU GOT IT RIGHT!")
-        break
-    remaining_guesses -= 1
-    if remaining_guesses == 3:
-        res = requests.get(f"{base_url}{quote['bio-link']}")
-        soup = BeautifulSoup(res.text, "html.parser")
-        birth_date = soup.find(class_="author-born-date").get_text()
-        birth_place = soup.find(class_="author-born-location").get_text()
-        print(f"Here's a hint: The author was born on {birth_date} {birth_place}")
-    elif remaining_guesses == 2:
-        print(f"Here's a hint: The author's first name starts with: {quote['author'][0]}")
-    elif remaining_guesses == 1:
-        last_initial = quote['author'].split(" ")[1][0]
-        print(f"Here's a hint: The author's last name starts with: {last_initial}")
+    again = ""
+    while again.lower() not in ("y", "yes", "n", "no"):
+        again = input("Would you like to lay again (y/n)?")
+    if again.lower() in ("y", "yes"):
+        return start_game(quotes)
     else:
-        print(f"Sorry, you ran out of guesses. The answer was {quote['author']}")
+        print("OK, GOODBYE")
+
+quotes = scrape_quotes()
+start_game(quotes)