feat(pipeline): finalize audited multi-platform dataset with clean provenance and entity resolution
This commit is contained in:
1 parent
ef0c0ca258
commit
4aa24d68c7
14 files changed
+2299
-169
No files matched your search
@@ -0,0 +1,65 @@
|
||||
"""
|
||||
pipeline/patch_anomalies.py
|
||||
Patches specific Instagram handle anomalies in multiplatform_checkpoints.csv
|
||||
and regenerates the final dataset in seconds without re-running 500 artists.
|
||||
"""
|
||||
|
||||
import sys
|
||||
import json
|
||||
from pathlib import Path
|
||||
from datetime import datetime, timezone
|
||||
|
||||
PROJECT_ROOT = Path(__file__).resolve().parent.parent
|
||||
if str(PROJECT_ROOT) not in sys.path:
|
||||
sys.path.insert(0, str(PROJECT_ROOT))
|
||||
|
||||
import pandas as pd
|
||||
from collectors.browser_collector import MultiPlatformSignalCollector
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
CHECKPOINT_FILE = PROJECT_ROOT / "data" / "checkpoints" / "multiplatform_checkpoints.csv"
|
||||
OVERRIDES_FILE = PROJECT_ROOT / "data" / "handle_overrides.json"
|
||||
|
||||
if not OVERRIDES_FILE.exists():
|
||||
print("Overrides file not found.")
|
||||
sys.exit(1)
|
||||
|
||||
with open(OVERRIDES_FILE, "r") as f:
|
||||
overrides = json.load(f)
|
||||
|
||||
df_ckpt = pd.read_csv(CHECKPOINT_FILE)
|
||||
collector = MultiPlatformSignalCollector(headless=True)
|
||||
|
||||
print("=" * 85)
|
||||
print(f"PATCHING {len(overrides)} IDENTIFIED INSTAGRAM ANOMALIES")
|
||||
print("=" * 85)
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
page = browser.new_page(
|
||||
viewport={"width": 1920, "height": 1080},
|
||||
user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36"
|
||||
)
|
||||
|
||||
for artist, official_handle in overrides.items():
|
||||
mask = df_ckpt["name"].str.strip().str.lower() == artist.lower()
|
||||
if mask.any():
|
||||
ig_url = f"https://www.instagram.com/{official_handle}/"
|
||||
res = collector.collect_artist({"name": artist, "ig_url": ig_url}, page)
|
||||
followers = res.get("instagram_followers")
|
||||
|
||||
df_ckpt.loc[mask, "instagram_handle"] = official_handle
|
||||
df_ckpt.loc[mask, "instagram_followers"] = followers
|
||||
df_ckpt.loc[mask, "ig_status"] = res.get("ig_status", "SUCCESS")
|
||||
|
||||
f_str = f"{followers:,}" if followers is not None else "Failed"
|
||||
print(f"Updated: {artist} -> @{official_handle} ({f_str} followers)")
|
||||
else:
|
||||
print(f"Artist {artist} not found in checkpoints.")
|
||||
|
||||
browser.close()
|
||||
|
||||
# Save updated checkpoints
|
||||
df_ckpt.to_csv(CHECKPOINT_FILE, index=False)
|
||||
print("\nCheckpoints successfully updated!")
|
||||
print("=" * 85)
|
||||
Reference in new issue
Block a user