import time
from playwright.sync_api import sync_playwright
import os

with sync_playwright() as p:
    browser = p.chromium.launch(headless=False)
    state_file = "snapchat_state.json"
    
    if os.path.exists(state_file):
        context = browser.new_context(
            storage_state=state_file,
            viewport={"width": 1280, "height": 720},
            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
        )
    else:
        context = browser.new_context()

    page = context.new_page()
    page.goto("https://web.snapchat.com/")
    
    print("Wait 60s to check if logged in... Please login if you are not.")
    time.sleep(60)
    
    context.storage_state(path="snapchat_state.json")
    browser.close()