setup_cookies.py

"""
Cookie Converter
================
Browser ကနေ copy ထားတဲ့ raw cookie string ကို
cookies/cookies.txt ထဲ ထည့်ပေးတဲ့ helper script
"""

from pathlib import Path

COOKIES_DIR = Path(__file__).parent / "cookies"
COOKIES_DIR.mkdir(exist_ok=True)
COOKIES_TXT = COOKIES_DIR / "cookies.txt"

# ─── tiktok.txt ကနေ ဖတ်ထားတဲ့ raw cookie string ────────────────────────────
# ဒီကနေ paste လုပ်ပါ (tiktok.txt ထဲက content)
RAW_COOKIE = """tt_csrf_token=TEln2Epk-2FV1bBciL0fM3fM2uvqkgc1ElAY; sessionid=b0c0bbd657775f6f526b01c0e745822a; sid_tt=b0c0bbd657775f6f526b01c0e745822a; sessionid_ss=b0c0bbd657775f6f526b01c0e745822a; uid_tt=970a6e61aa747ae52a9646c20a3912cfe441cf821ca5567cb93039837eae83f3; msToken=sB6sITa6k2r4gb_19IwCbi4FmEhw9pjrf9N0Hz7SvT0Oikj1RIhAC99gxR_2Q1lcECXnE1bKpI1AAQfEBiN0JPWUCiqIertUMJYBpDI1-1mVrjXAoUWqUnN1GWBaRtQPvLVlx2KJyjSoIacnkMN2MjCdJA==; ttwid=1%7CwFHvgZ1DucrLr63uposDjX1uN1rToWbL62I6g0WgNxQ%7C1775570136%7Cd7ee936646c42febbd124a94857bd191576ba040a6729f5da578e1347f09de2b; store-idc=alisg; store-country-code=th"""

def save_raw_cookie(raw: str):
    """Raw cookie string ကို cookies.txt မှာ သိမ်း"""
    COOKIES_TXT.write_text(raw.strip(), encoding="utf-8")
    print(f"✅ Saved to {COOKIES_TXT}")

    # sessionid verify
    for part in raw.split(";"):
        kv = part.strip()
        if kv.startswith("sessionid="):
            sid = kv.split("=", 1)[1]
            print(f"✅ sessionid found: {sid[:20]}...")
            return
    print("⚠️  sessionid မတွေ့ဘူး — cookie မှန်မမှန် စစ်ပါ")

if __name__ == "__main__":
    save_raw_cookie(RAW_COOKIE)