Do rename trackId if it has not already been added

ddxv · Oct 1, 2024 · e54225e · e54225e
1 parent 098bcd4
commit e54225e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/adscrawler/app_stores/apple.py b/adscrawler/app_stores/apple.py
@@ -128,6 +128,8 @@ def scrape_app_ios(store_id: str, country: str) -> dict:
 
 
 def clean_ios_app_df(df: pd.DataFrame) -> pd.DataFrame:
+    if 'store_id' not in df.columns:
+        df = df.rename(columns={"trackId": "store_id"})
     df = df.rename(
         columns={
             # "trackId": "store_id",