Move split_names_list function to lib

pages/1_Check_Card_List.py

@@ -5,16 +5,6 @@ import streamlit_common.lib as lib
 
 mslist_path = "output/middleschool.csv"
 
-
-def split_names_list(row: pd.DataFrame):
-    if not isinstance(row["legalnames"], list):
-        return row
-    row["English"] = row["legalnames"][0]
-    if row["legalnames"][1] is not None:
-        row["日本語"] = row["legalnames"][1]
-    return row
-
-
 st.set_page_config(
     page_title="Middle School | Check Card List",
     page_icon="🃏",
@@ -51,7 +41,7 @@ input_cards["Legal"] = input_cards["cardname"].apply(
 input_cards["legalnames"] = input_cards["cardname"].apply(
     lib.get_legal_cardnames, args=[mslist_df]
 )
-input_cards = input_cards.apply(split_names_list, axis=1)
+input_cards = input_cards.apply(lib.split_names_list, axis=1)
 
 col2.write("##### Middle School legality")
 col2.dataframe(input_cards[["Legal", "English", "日本語"]], use_container_width=True)

streamlit_common/lib.py

@@ -55,3 +55,13 @@ def is_cardname_legal(cardname: str, mslist_df: pd.DataFrame) -> bool:
     if mslist_df[mslist_df["name_ja"] == cardname].shape[0] > 0:
         return True
     return False
+
+
+def split_names_list(row: pd.DataFrame):
+    """Splits the English and Japanese card names in a list into two different columns"""
+    if not isinstance(row["legalnames"], list):
+        return row
+    row["English"] = row["legalnames"][0]
+    if row["legalnames"][1] is not None:
+        row["日本語"] = row["legalnames"][1]
+    return row