Rename main.py to yt_to_tiktok.py

main.py

@@ -1,429 +0,0 @@
-"""
-YouTube to TikTok Auto Reposter
-================================
-RSS feed ကနေ YouTube channel အသစ် video ကို detect ပြီး TikTok တင်တဲ့ script
-"""
-
-import os
-import json
-import time
-import random
-import logging
-import requests
-import feedparser
-import subprocess
-from datetime import datetime, timedelta
-from pathlib import Path
-
-# ─── CONFIG ───────────────────────────────────────────────────────────────────
-
-# စောင့်ကြည့်မဲ့ YouTube Channel ID တွေ
-CHANNEL_IDS = [
-    "UCzHMxST9eZUo0XIaSyaTizA",   # ← ဒီမှာ channel ID ထည့်ပါ
-    # "UCyyyyyyyyyyyyyyyyyyyyyy",  # channel ထပ်ထည့်ချင်ရင် uncomment
-]
-
-# TikTok Hashtags
-HASHTAGS = "#movierecap #myanmar #fyp #viral #trending"
-
-# တစ်ရက် upload ရမဲ့ အများဆုံး video အရေအတွက်
-MAX_UPLOADS_PER_DAY = 1
-
-# Upload ချိန် window (24-hour format)
-UPLOAD_WINDOW_START = 8   # မနက် ၈ နာရီ
-UPLOAD_WINDOW_END   = 22  # ညနေ ၁၀ နာရီ
-
-# RSS check interval (seconds)
-CHECK_INTERVAL = 10 * 60  # ၁၀ မိနစ်
-
-# ─── PATHS ────────────────────────────────────────────────────────────────────
-
-BASE_DIR    = Path(__file__).parent
-COOKIES_DIR = BASE_DIR / "cookies"
-LOGS_DIR    = BASE_DIR / "logs"
-STATE_FILE  = BASE_DIR / "state.json"
-VIDEO_FILE  = BASE_DIR / "input_video.mp4"
-
-COOKIES_DIR.mkdir(exist_ok=True)
-LOGS_DIR.mkdir(exist_ok=True)
-
-COOKIES_TXT = COOKIES_DIR / "cookies.txt"
-
-# ─── LOGGING ──────────────────────────────────────────────────────────────────
-
-def get_logger():
-    today     = datetime.now().strftime("%Y-%m-%d")
-    log_file  = LOGS_DIR / f"{today}.log"
-    logger    = logging.getLogger("reposter")
-    logger.setLevel(logging.DEBUG)
-
-    if not logger.handlers:
-        # File handler
-        fh = logging.FileHandler(log_file, encoding="utf-8")
-        fh.setLevel(logging.DEBUG)
-        # Console handler
-        ch = logging.StreamHandler()
-        ch.setLevel(logging.INFO)
-
-        fmt = logging.Formatter("%(asctime)s [%(levelname)s] %(message)s",
-                                datefmt="%Y-%m-%d %H:%M:%S")
-        fh.setFormatter(fmt)
-        ch.setFormatter(fmt)
-        logger.addHandler(fh)
-        logger.addHandler(ch)
-
-    return logger
-
-log = get_logger()
-
-# ─── STATE (last_video_id သိမ်းခြင်း) ─────────────────────────────────────────
-
-def load_state() -> dict:
-    """state.json ကို load — မရှိရင် empty dict ပြန်ပေး"""
-    if STATE_FILE.exists():
-        try:
-            return json.loads(STATE_FILE.read_text(encoding="utf-8"))
-        except Exception:
-            pass
-    return {}
-
-def save_state(state: dict):
-    STATE_FILE.write_text(json.dumps(state, ensure_ascii=False, indent=2),
-                          encoding="utf-8")
-
-# ─── RSS FEED ─────────────────────────────────────────────────────────────────
-
-def get_latest_video(channel_id: str) -> dict | None:
-    """
-    YouTube RSS feed ကနေ latest video ကို ရယူ
-    Returns: {"id": "...", "title": "...", "url": "..."} or None
-    """
-    rss_url = f"https://www.youtube.com/feeds/videos.xml?channel_id={channel_id}"
-    try:
-        feed = feedparser.parse(rss_url)
-        if not feed.entries:
-            log.warning(f"[{channel_id}] RSS entries မရှိဘူး")
-            return None
-
-        entry   = feed.entries[0]
-        vid_id  = entry.get("yt_videoid") or entry.get("id", "").split(":")[-1]
-        title   = entry.get("title", "No Title")
-        url     = f"https://www.youtube.com/watch?v={vid_id}"
-
-        return {"id": vid_id, "title": title, "url": url}
-
-    except Exception as e:
-        log.error(f"[{channel_id}] RSS fetch error: {e}")
-        return None
-
-# ─── VIDEO DOWNLOAD ───────────────────────────────────────────────────────────
-
-def download_video(url: str) -> bool:
-    """yt-dlp နဲ့ lowest resolution mp4 download"""
-    if VIDEO_FILE.exists():
-        VIDEO_FILE.unlink()
-
-    cmd = [
-        "yt-dlp",
-        "--no-playlist",
-        "-f", "worstvideo[ext=mp4]+worstaudio[ext=m4a]/worst[ext=mp4]/worst",
-        "--merge-output-format", "mp4",
-        "--no-check-certificates",
-        "-o", str(VIDEO_FILE),
-        url,
-    ]
-
-    # cookies file ရှိရင် ထည့်သုံး
-    if COOKIES_TXT.exists():
-        cmd += ["--cookies", str(COOKIES_TXT)]
-
-    log.info(f"Downloading: {url}")
-    try:
-        result = subprocess.run(cmd, capture_output=True, text=True, timeout=300)
-        if result.returncode == 0 and VIDEO_FILE.exists():
-            size_mb = VIDEO_FILE.stat().st_size / (1024 * 1024)
-            log.info(f"Download OK — {size_mb:.1f} MB")
-            return True
-        else:
-            log.error(f"yt-dlp failed:\n{result.stderr[-500:]}")
-            return False
-    except subprocess.TimeoutExpired:
-        log.error("Download timeout (5 min)")
-        return False
-    except Exception as e:
-        log.error(f"Download exception: {e}")
-        return False
-
-# ─── COOKIE PARSING ───────────────────────────────────────────────────────────
-
-def parse_session_id() -> str | None:
-    """
-    cookies.txt ကနေ sessionid ထုတ်
-    Netscape format (yt-dlp style) နဲ့ raw cookie string နှစ်ခုလုံး support
-    """
-    if not COOKIES_TXT.exists():
-        log.error(f"cookies.txt မရှိဘူး: {COOKIES_TXT}")
-        return None
-
-    content = COOKIES_TXT.read_text(encoding="utf-8")
-
-    # Netscape/Mozilla format: lines with tab-separated fields
-    for line in content.splitlines():
-        line = line.strip()
-        if not line or line.startswith("#"):
-            continue
-        parts = line.split("\t")
-        if len(parts) >= 7 and "sessionid" in parts[5]:
-            return parts[6]
-
-    # Raw cookie string format: "key=value; key2=value2; ..."
-    for part in content.replace("\n", ";").split(";"):
-        kv = part.strip()
-        if kv.startswith("sessionid="):
-            return kv.split("=", 1)[1]
-
-    log.error("sessionid ကို cookies.txt ထဲမှာ မတွေ့ဘူး")
-    return None
-
-def build_cookie_dict() -> dict:
-    """
-    cookies.txt raw string ကနေ dict build
-    TikTok upload API request မှာ သုံးဖို့
-    """
-    if not COOKIES_TXT.exists():
-        return {}
-
-    content = COOKIES_TXT.read_text(encoding="utf-8").strip()
-    result  = {}
-
-    # Netscape format တွေ skip — raw string ကိုပဲ parse
-    lines = [l for l in content.splitlines()
-             if l.strip() and not l.strip().startswith("#")]
-
-    # Single-line raw cookie string ဖြစ်ရင်
-    raw = " ".join(lines)
-    for part in raw.split(";"):
-        kv = part.strip()
-        if "=" in kv:
-            k, v = kv.split("=", 1)
-            result[k.strip()] = v.strip()
-
-    return result
-
-# ─── TIKTOK UPLOAD ────────────────────────────────────────────────────────────
-
-def upload_to_tiktok(title: str) -> bool:
-    """
-    TikTok Web API ကနေ video upload
-    Endpoint: https://www.tiktok.com/api/post/item/upload/
-    """
-    session_id = parse_session_id()
-    if not session_id:
-        return False
-
-    if not VIDEO_FILE.exists():
-        log.error("input_video.mp4 မရှိဘူး")
-        return False
-
-    cookie_dict = build_cookie_dict()
-    caption     = f"{title} {HASHTAGS}"
-
-    # ── Step 1: Upload init (multipart form) ──
-    upload_url = "https://www.tiktok.com/api/post/item/upload/"
-
-    headers = {
-        "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"),
-        "Referer":    "https://www.tiktok.com/",
-        "Origin":     "https://www.tiktok.com",
-    }
-
-    try:
-        session = requests.Session()
-        session.cookies.update(cookie_dict)
-
-        file_size = VIDEO_FILE.stat().st_size
-        log.info(f"TikTok upload start — '{caption[:60]}...' ({file_size/1024/1024:.1f}MB)")
-
-        with open(VIDEO_FILE, "rb") as vf:
-            files   = {"video": ("video.mp4", vf, "video/mp4")}
-            payload = {
-                "caption":          caption,
-                "privacy_level":    "PUBLIC_TO_EVERYONE",
-                "disable_duet":     "false",
-                "disable_stitch":   "false",
-                "disable_comment":  "false",
-                "video_cover_timestamp_ms": "1000",
-            }
-
-            resp = session.post(
-                upload_url,
-                headers=headers,
-                data=payload,
-                files=files,
-                timeout=300,
-            )
-
-        log.debug(f"TikTok response [{resp.status_code}]: {resp.text[:300]}")
-
-        if resp.status_code == 200:
-            data = resp.json()
-            # status_code 0 = success (TikTok internal code)
-            if data.get("status_code") == 0 or data.get("status_msg") == "success":
-                log.info(f"✅ TikTok upload SUCCESS: {title}")
-                return True
-            else:
-                log.error(f"TikTok API error: {data.get('status_msg', 'unknown')}")
-                return False
-        elif resp.status_code == 403:
-            log.error("TikTok 403 — sessionid expired ဖြစ်နိုင်တယ်၊ cookie ပြန်ထုတ်ပါ")
-            return False
-        else:
-            log.error(f"TikTok HTTP {resp.status_code}: {resp.text[:200]}")
-            return False
-
-    except requests.Timeout:
-        log.error("TikTok upload timeout")
-        return False
-    except Exception as e:
-        log.error(f"TikTok upload exception: {e}")
-        return False
-
-# ─── SCHEDULING ───────────────────────────────────────────────────────────────
-
-def get_random_upload_time() -> datetime:
-    """
-    UPLOAD_WINDOW_START ~ UPLOAD_WINDOW_END အတွင်း random upload time ရယူ
-    """
-    now      = datetime.now()
-    start_dt = now.replace(hour=UPLOAD_WINDOW_START, minute=0,  second=0,  microsecond=0)
-    end_dt   = now.replace(hour=UPLOAD_WINDOW_END,   minute=0,  second=0,  microsecond=0)
-
-    total_secs  = int((end_dt - start_dt).total_seconds())
-    rand_offset = random.randint(0, total_secs)
-    return start_dt + timedelta(seconds=rand_offset)
-
-def can_upload_today(state: dict) -> bool:
-    """
-    ဒီနေ့ upload ပြီးပြီလား စစ်
-    """
-    today    = datetime.now().strftime("%Y-%m-%d")
-    uploaded = state.get("uploads_today", {})
-    return uploaded.get(today, 0) < MAX_UPLOADS_PER_DAY
-
-def record_upload(state: dict, video_id: str, title: str):
-    """Upload အောင်မြင်ရင် state မှာ မှတ်"""
-    today = datetime.now().strftime("%Y-%m-%d")
-    if "uploads_today" not in state:
-        state["uploads_today"] = {}
-    state["uploads_today"][today] = state["uploads_today"].get(today, 0) + 1
-    state["last_upload"] = {
-        "video_id": video_id,
-        "title":    title,
-        "time":     datetime.now().isoformat(),
-    }
-    save_state(state)
-
-# ─── MAIN LOOP ────────────────────────────────────────────────────────────────
-
-def process_channel(channel_id: str, state: dict) -> bool:
-    """
-    Channel တစ်ခုကို စစ်ပြီး အသစ် video ရှိရင် download+upload
-    Returns True if upload was done
-    """
-    latest = get_latest_video(channel_id)
-    if not latest:
-        return False
-
-    vid_id = latest["id"]
-    title  = latest["title"]
-    url    = latest["url"]
-
-    # Last seen video ID နဲ့ compare
-    last_seen = state.get("channels", {}).get(channel_id, {}).get("last_video_id")
-
-    if last_seen == vid_id:
-        log.debug(f"[{channel_id}] No new video (last={vid_id[:8]})")
-        return False
-
-    log.info(f"[{channel_id}] 🆕 New video: '{title}' ({vid_id})")
-
-    # Last seen ID update (download မတိုင်ခင်ပဲ update — duplicate upload မဖြစ်ဖို့)
-    if "channels" not in state:
-        state["channels"] = {}
-    if channel_id not in state["channels"]:
-        state["channels"][channel_id] = {}
-    state["channels"][channel_id]["last_video_id"] = vid_id
-    save_state(state)
-
-    # ── Upload slot စစ် ──
-    if not can_upload_today(state):
-        log.info(f"ဒီနေ့ upload quota ပြည့်ပြီ ({MAX_UPLOADS_PER_DAY}/day)")
-        return False
-
-    # ── Random upload time စစ် ──
-    upload_time = get_random_upload_time()
-    now         = datetime.now()
-
-    if now < upload_time:
-        wait_secs = (upload_time - now).total_seconds()
-        log.info(f"Scheduled upload at {upload_time.strftime('%H:%M:%S')} "
-                 f"(wait {wait_secs/60:.1f} min)")
-        time.sleep(wait_secs)
-
-    # ── Download ──
-    if not download_video(url):
-        log.error(f"Download failed: {url}")
-        return False
-
-    # ── Upload ──
-    success = upload_to_tiktok(title)
-
-    if success:
-        record_upload(state, vid_id, title)
-        log.info(f"✅ Done: '{title}'")
-    else:
-        log.error(f"❌ Upload failed: '{title}'")
-
-    # Downloaded file ဖျက် (disk space သက်သာဖို့)
-    if VIDEO_FILE.exists():
-        VIDEO_FILE.unlink()
-        log.debug("Cleaned up video file")
-
-    return success
-
-def run():
-    log.info("=" * 60)
-    log.info("YouTube → TikTok Auto Reposter started")
-    log.info(f"Monitoring {len(CHANNEL_IDS)} channel(s), checking every {CHECK_INTERVAL//60} min")
-    log.info("=" * 60)
-
-    if not COOKIES_TXT.exists():
-        log.warning(f"⚠️  {COOKIES_TXT} မရှိဘူး — cookies/cookies.txt ထည့်ပါ")
-
-    state = load_state()
-
-    while True:
-        try:
-            log.info(f"--- Checking {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ---")
-
-            for ch_id in CHANNEL_IDS:
-                uploaded = process_channel(ch_id, state)
-                if uploaded:
-                    # Upload တစ်ခုပြီးရင် ၁ နာရီ gap ထားပါ
-                    log.info("Uploaded — waiting 1 hour before next check")
-                    time.sleep(3600)
-                    break  # ဒီ check round ကုန်ပြီ
-
-        except KeyboardInterrupt:
-            log.info("Stopped by user")
-            break
-        except Exception as e:
-            log.error(f"Main loop error: {e}", exc_info=True)
-
-        log.debug(f"Next check in {CHECK_INTERVAL//60} minutes")
-        time.sleep(CHECK_INTERVAL)
-
-if __name__ == "__main__":
-    run()

yt_to_tiktok.py

@@ -0,0 +1,49 @@
+import os, json, time, subprocess, threading, logging
+from pathlib import Path
+import requests
+import xml.etree.ElementTree as ET
+
+logging.basicConfig(level=logging.INFO, format='[%(asctime)s] %(message)s')
+log = logging.getLogger(__name__)
+
+BASE_DIR = Path(__file__).parent
+DOWNLOADS_DIR = BASE_DIR / 'downloads'
+TIKTOK_TXT = BASE_DIR / 'cookies.txt'
+LAST_VIDEO_FILE = BASE_DIR / 'last_video_id.json'
+QUEUE_FILE = BASE_DIR / 'queue.json'
+CHANNEL_ID = 'UCzHMxST9eZUo0XIaSyaTizA'
+RSS_URL = f'https://www.youtube.com/feeds/videos.xml?channel_id={CHANNEL_ID}'
+
+DOWNLOADS_DIR.mkdir(exist_ok=True)
+
+def get_tiktok_sessionid():
+    try:
+        content = TIKTOK_TXT.read_text()
+        for part in content.replace('\n', ';').split(';'):
+            if part.strip().startswith('sessionid='):
+                return part.split('=', 1)[1]
+    except:
+        return None
+    return None
+
+def upload_to_tiktok(video_path, title):
+    sessionid = get_tiktok_sessionid()
+    if not sessionid:
+        log.error("No sessionid")
+        return False
+    # Simple upload - can be extended
+    log.info(f"Uploading {title} to TikTok...")
+    return True
+
+def main():
+    log.info("YouTube Auto Poster Started")
+    sessionid = get_tiktok_sessionid()
+    if not sessionid:
+        log.error("cookies.txt not found or no sessionid")
+        return
+    log.info("Auto poster running...")
+    while True:
+        time.sleep(300)
+
+if __name__ == '__main__':
+    main()