From 164d4bec0e9e28839f128e4814dcd209ff37ba09 Mon Sep 17 00:00:00 2001
From: anela <anela89@gmail.com>
Date: Fri, 21 Apr 2023 10:43:43 -0700
Subject: [PATCH] Initial commit

---
 new_mexico.py | 113 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 new_mexico.py

diff --git a/new_mexico.py b/new_mexico.py
new file mode 100644
index 0000000..1313be5
--- /dev/null
+++ b/new_mexico.py
@@ -0,0 +1,113 @@
+import logging
+import os
+import re
+from xmlrpc import client
+import traceback
+
+from bs4 import BeautifulSoup as bs
+import requests
+
+
+from lotto_site_parsers.util import save_image
+from lotto_site_parsers.util import save_game
+
+logger = logging.getLogger(__name__)
+
+DB_REPO_URI = os.environ.get("DB_REPO_URI", "http://localhost:8989")
+BASE_URL = "https://www.nmlottery.com"
+INDEX_URL = "https://www.nmlottery.com/games/scratchers"
+HEADERS = {
+    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:70.0) Gecko/20100101 Firefox/70.0",
+}
+
+
+def get_games(site_url):
+    """
+    Takes the URL from the scratcher site
+    parses page for game ids and game info
+    returns and list of tuples with the id and game info for each game
+    """
+    html = requests.get(site_url, headers=HEADERS).text
+    soup = bs(html, "html.parser")
+
+    games_html = soup.find_all("div", class_="filter-block")
+
+    ids = [
+        re.search("\d+", id.text).group(0)
+        for id in soup.find_all("p", class_="game-number")
+    ]
+
+    game_names = [name.text for name in soup.find_all("h3")]
+
+    return list(zip(ids, game_names, games_html))
+
+
+def process_game(game_info):
+    """
+    function takes game info: [game id, game_name, game_html_data]
+
+    parses info to find specific game data
+    ex name, game_id, price, odds, prizes, how to play, image_url
+
+    returns game object
+    """
+
+    game_html = game_info[2]
+
+    name = game_info[1]
+
+    game_id = game_info[0]
+
+    price = float(game_html.find("p", class_="price").text.replace("$", ""))
+
+    how_to_play = game_html.find("p", class_="how-to-play").find_next("span").text
+
+    prizes = [
+        {
+            "prize": row[0].strip(),
+            "value": price
+            if "prize ticket" in row[0].lower()
+            else float(row[0].replace("$", "").replace(",", "")),
+            "claimed": int(row[2].replace(",", "")) - int(row[3].replace(",", "")),
+            "available": int(row[3].replace(",", "")),
+            "total": int(row[2].replace(",", "")),
+            "odds": float(row[1].replace(",", "")),
+        }
+        for row in [
+            row.text.split("\n")[1:-1] for row in game_html.table.find_all("tr")[1:]
+        ]
+    ]
+
+    num_of_tix = int(prizes[0]["odds"] * prizes[0]["total"])
+
+    image_url = game_html.find("div", class_="scratcher-image").find_next("img")["src"]
+    image_location = save_image("nm", game_id, image_url, headers=HEADERS)
+
+    game = {
+        "name": name,
+        "game_id": game_id,
+        "price": price,
+        "how_to_play": how_to_play,
+        "prizes": prizes,
+        "num_tx_initial": num_of_tix,
+        "state": "nm",
+        "image_urls": '["{}"]'.format(image_location),
+    }
+
+    return game
+
+
+def main():
+    games = get_games(INDEX_URL)
+    for game in games:
+        try:
+            game = process_game(game)
+            save_game(game)
+        except Exception as e:
+            logger.warning(f"Unable to process game: {game[0]}-{game[1]}")
+            logger.warning(e)
+            traceback.print_exception(e)
+
+
+if __name__ == "__main__":
+    main()