Merge pull request #35 from alecrem/feature/issue-34/notify-exact-match

Middle_School_Card_Search.py

@@ -4,12 +4,34 @@ import urllib.parse
 import streamlit_common.footer
 
 
-def compose_scryfall_url(x):
+def compose_scryfall_url(x: str) -> str:
     return f"https://scryfall.com/search?q=prefer%3Aoldest%20!%22{urllib.parse.quote_plus(x)}%22"
 
 
-def row_to_link(x):
-    st.markdown(f"- [{x['name']} / {x.name_ja}]({x.link})")
+def row_to_link(x: pd.DataFrame) -> None:
+    cardname = x["name"]
+    if x.name_ja is not "":
+        cardname = f"{cardname} / {x.name_ja}"
+    st.markdown(f"- [{cardname}]({x.link})")
+
+
+def is_legal(cardname: str) -> list:
+    english_match = mslist_df[mslist_df["name"].str.lower() == cardname.lower()]
+    cardname_en_list = None
+    if english_match.shape[0] > 0:
+        cardname_en_list = english_match["name"].to_list()
+        cardname_ja_list = english_match["name_ja"].to_list()
+    japanese_match = mslist_df[mslist_df["name_ja"] == cardname]
+    if japanese_match.shape[0] > 0:
+        cardname_en_list = japanese_match["name"].to_list()
+        cardname_ja_list = japanese_match["name_ja"].to_list()
+    if cardname_en_list is not None and len(cardname_en_list) > 0:
+        return [
+            cardname_en_list[0] or None,
+            cardname_ja_list[0] or None,
+        ]
+
+    return None
 
 
 mslist_path = "output/middleschool.csv"
@@ -32,7 +54,8 @@ mslist_df = pd.read_csv(mslist_path)
 mslist_df.fillna("", inplace=True)
 st.write(mslist_df.shape[0], "cards are legal")
 
-name_input = st.text_input(f"Search by card name")
+name_input = st.text_input(f"Search by card name").strip()
+exact_match = is_legal(name_input)
 results_en_df = mslist_df[
     mslist_df["name"].str.contains(name_input.lower(), case=False)
 ]
@@ -41,6 +64,13 @@ results_ja_df = mslist_df[
 ]
 results_df = results_en_df.merge(results_ja_df, how="outer")
 if name_input:
+    if exact_match is not None:
+        cardname = exact_match[0]
+        if exact_match[1] is not None:
+            cardname = f"{cardname} / {exact_match[1]}"
+        st.write(
+            f"✅ [{cardname}]({compose_scryfall_url(exact_match[0])}) is an exact match"
+        )
     st.write(results_df.shape[0], f'cards found by "{name_input}"')
     if results_df.shape[0] > number_shown_results:
         st.write(f"Top {number_shown_results} results:")