import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from celery_tasks.celery import app
from base.utils.scrapers import *

@app.task(queue='find_titles')
def find_title(imdb_id: str, title_name: str, requested_season: int, requested_quality: str):
    res = find_title_from_zarfilm(imdb_id)
    if res['status']:
        res = extract_dl_links_from_zarfilm(res['url'], requested_season, requested_quality)
    if not res['status']:
        res = find_title_from_30nama(imdb_id, title_name)
        if res['status']:
            res = extract_dl_links_from_30nama(res['movie_id'], res['is_series'], requested_season, requested_quality)
    return res