Scraping prescription drug prices from Rx site using the prescription drug name and zipcode

web_programming/fetch_well_rx_price.py

@@ -0,0 +1,112 @@
+"""
+
+Scrape the price and pharmacy name for a prescription drug from rx site
+after providing the drug name and zipcode. 
+
+"""
+
+import lxml
+
+from requests import Response, get
+from bs4 import BeautifulSoup
+
+
+BASE_URL: str = "https://www.wellrx.com/prescriptions/{0}/{1}/?freshSearch=true"
+
+
+def format_price(price: str) -> float:
+    """[summary]
+
+    Remove the dollar from the string and convert it to float.
+
+    >>> format_price("$14")
+    14.0
+
+    >>> format_price("$15.67")
+    15.67
+
+    >>> format_price("$0.00")
+    0.0
+
+    Args:
+        price (str): [price of drug in string format]
+
+    Returns:
+        float: [formatted price of drug in float]
+    """
+    dollar_removed: str = price.replace("$", "")
+    formatted_price: str = float(dollar_removed)
+    return formatted_price
+
+
+def fetch_pharmacy_and_price_list(drug_name: str, zip_code: str) -> list:
+
+    """[summary]
+
+    This function will take input of drug name and zipcode, then request to the BASE_URL site,
+    Get the page data and scrape it to the generate the list of lowest prices for the prescription drug.
+
+    Args:
+        drug_name (str): [Drug name]
+        zip_code(str): [Zip code]
+
+    Returns:
+        list: [List of pharmacy name and price]
+    """
+
+    try:
+
+        # has user provided both inputs?
+        if not drug_name or not zip_code:
+            return []
+
+        request_url: str = BASE_URL.format(drug_name, zip_code)
+        response: Response = get(request_url)
+
+        # Is the status code ok?
+        if response.status_code == 200:
+
+            # Scrape the data using bs4
+            soup: BeautifulSoup = BeautifulSoup(response.text, "lxml")
+
+            # This list will store the name and price.
+            pharmacy_price_list: list = []
+
+            # Fetch all the grids that contains the items.
+            grid_list: list = soup.find_all("div", { "class": "grid-x pharmCard" })
+            if grid_list and len(grid_list) > 0:
+                for grid in grid_list:
+
+                    # Get the pharmacy price.
+                    pharmacy_name: str = grid.find("p", { "class": "list-title" }).text
+
+                    # Get price of the drug.
+                    price: str = grid.find("span", { "p", "price price-large" }).text
+                    formatted_price: float = format_price(price)
+
+                    pharmacy_price_list.append({
+                        "pharmacy_name": pharmacy_name,
+                        "price": formatted_price,
+                    })
+
+            # Print the pharmacy name and price for the drug.
+
+            return pharmacy_price_list
+
+        else:
+            return []
+
+    except Exception as e:
+        return []
+
+
+if __name__ == "__main__":
+
+    # Enter a drug name and a zip code
+    drug_name: str = input("Enter drug Name:\n")
+    zip_code: str = input("Enter zip code:\n")
+    pharmacy_price_list: list = fetch_pharmacy_and_price_list(drug_name, zip_code)
+
+    print("Search results for {0} at location {1}\n".format(drug_name, zip_code))
+    for pharmacy_price in pharmacy_price_list:
+        print("Pharmacy: {0} Price: {1}".format(pharmacy_price["pharmacy_name"], pharmacy_price["price"]))