Src/API/animeworld.py

from bs4 import BeautifulSoup
import datetime
import json
from Src.Utilities.info import get_info_kitsu
import Src.Utilities.config as config
import re
AW_DOMAIN = config.AW_DOMAIN
months = {
        "Gennaio": "January", "Febbraio": "February", "Marzo": "March", 
        "Aprile": "April", "Maggio": "May", "Giugno": "June", 
        "Luglio": "July", "Agosto": "August", "Settembre": "September", 
        "Ottobre": "October", "Novembre": "November", "Dicembre": "December"
    }
showname_replace = {
    "Attack on Titan": "L'attacco dei Giganti",
    "Season": "",
    "  ": " ",
    "Shippuuden": "Shippuden",
    " ": "+",
}

async def get_mp4(anime_url,ismovie,episode,client):
   response = await client.get(anime_url,allow_redirects=True, impersonate = "chrome120")
   soup = BeautifulSoup(response.text,'lxml')
   episode_page = soup.find('a', {'data-episode-num':episode })
   if episode_page is None:
       return None
   episode_page = f'https://animeworld.{AW_DOMAIN}{episode_page["href"]}'
   response = await client.get(episode_page,allow_redirects=True, impersonate = "chrome120")
   soup = BeautifulSoup(response.text,'lxml')
   a_tag  = soup.find('a', {'id': 'alternativeDownloadLink', 'class': 'm-1 btn btn-sm btn-primary'}) 
   url = a_tag['href']
   response = await client.head(url)
   if response.status_code == 404:
        url = None
   return url







async def old_search(showname,date,ismovie,episode,client):
    cookies = {
    'sessionId': 's%3AtGSRfYcsIoaeV0nqFJgN69Zxixb_-uJU.fcNz%2FsJBiiP8v8TwthMN9%2FmynWFciI5gezZuz8CltyQ',
    }

    headers = {
        'authority': f'www.animeworld.{AW_DOMAIN}',
        'accept': 'application/json, text/javascript, */*; q=0.01',
        'accept-language': 'it-IT,it;q=0.9,en-US;q=0.8,en;q=0.7',
         # 'content-length': '0',
         # 'cookie': 'sessionId=s%3AtGSRfYcsIoaeV0nqFJgN69Zxixb_-uJU.fcNz%2FsJBiiP8v8TwthMN9%2FmynWFciI5gezZuz8CltyQ',
         'csrf-token': 'oKFK43s4-BzfqPX27RlAORUd-iyiAfXyfDAo',
         'origin': f'https://www.animeworld.{AW_DOMAIN}',
         'referer': f'https://www.animeworld.{AW_DOMAIN}/',
         'sec-ch-ua': '"Not-A.Brand";v="99", "Chromium";v="124"',
         'sec-ch-ua-mobile': '?0',
         'sec-ch-ua-platform': '"Android"',
         'sec-fetch-dest': 'empty',
         'sec-fetch-mode': 'cors',
         'sec-fetch-site': 'same-origin',
         'user-agent': 'Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
         'x-requested-with': 'XMLHttpRequest',
    }

    params = {
        'keyword': showname,
    }

    response = await client.post(f'https://www.animeworld.{AW_DOMAIN}/api/search/v2', params=params, cookies=cookies, headers=headers,allow_redirects=True, impersonate = "chrome120")

    data = json.loads(response.text)
    final_urls = []
    for anime in data["animes"]:
        release_date = anime["release"]
        for ita, eng in months.items():
            release_date = release_date.replace(ita, eng)
        release_date = datetime.datetime.strptime(release_date, "%d %B %Y")
        release_date = release_date.strftime("%Y-%m-%d")
        if release_date == date:
            identifier = anime["identifier"]
            link = anime["link"]
            anime_url = f'https://animeworld.{AW_DOMAIN}/play/{link}.{identifier}'
            final_url = await get_mp4(anime_url,ismovie,episode,client)
            final_urls.append(final_url)
            break
    showname = showname + " (ITA)"
    params = {
        'keyword': showname,
    }
    response = await client.post(f'https://www.animeworld.{AW_DOMAIN}/api/search/v2', params=params, cookies=cookies, headers=headers, allow_redirects=True, impersonate = "chrome120")
    data = json.loads(response.text)
    for anime in data["animes"]:
        release_date = anime["release"]
        for ita, eng in months.items():
            release_date = release_date.replace(ita, eng)
        release_date = datetime.datetime.strptime(release_date, "%d %B %Y")
        release_date = release_date.strftime("%Y-%m-%d")
        if release_date == date:
            identifier = anime["identifier"]
            link = anime["link"]
            anime_url = f'https://animeworld.{AW_DOMAIN}/play/{link}.{identifier}'
            final_url = await get_mp4(anime_url,ismovie,episode,client)
            final_urls.append(final_url)
            break
    return final_urls

async def search(showname,date,ismovie,episode,client):
    search_year = date[:4] 
    response = await client.get(f'https://www.animeworld.so/filter?year={search_year}&sort=2&keyword={showname}',allow_redirects=True, impersonate = "chrome120")
    soup = BeautifulSoup(response.text,'lxml')
    anime_list = soup.find_all('a', class_=['poster', 'tooltipstered'])
    final_urls = []
    for anime in anime_list:
        anime_info_url = f'https://www.animeworld.{AW_DOMAIN}/{anime["data-tip"]}'
        response = await client.get(anime_info_url,allow_redirects=True, impersonate = "chrome120")
        pattern = r'<label>Data di uscita:</label>\s*<span>\s*(.*?)\s*</span>'
        match = re.search(pattern, response.text, re.S)
        release_date = match.group(1).strip()
        for ita, eng in months.items():
            release_date = release_date.replace(ita, eng)
        release_date = datetime.datetime.strptime(release_date, "%d %B %Y")
        release_date = release_date.strftime("%Y-%m-%d")
        if release_date == date:
            anime_url = f'https://www.animeworld.{AW_DOMAIN}{anime["href"]}'
            final_url = await get_mp4(anime_url,ismovie,episode,client)
            if final_url:
                final_urls.append(final_url)

    return final_urls

async def animeworld(id,client):
    try:
        print(id)
        kitsu_id = id.split(":")[1]
        episode = id.split(":")[2]
        ismovie = 1 if len(id.split(":")) == 2 else 0
        showname,date = await get_info_kitsu(kitsu_id,client)
        for key in showname_replace:
            if key in showname:  # Check if the key is a substring of showname
                showname = showname.replace(key, showname_replace[key])
                if "Naruto:" in showname:
                    showname = showname.replace(":", "")
        final_urls = await search(showname,date,ismovie,episode,client)
        return final_urls
    except:
        print("Animeworld failed")
        return None

#async def test_animeworld():
    async with httpx.AsyncClient() as client:
        # Replace with actual id, for example 'anime_id:episode' format
        test_id = "kitsu:1555:1"  # This is an example ID format
        results = await animeworld(test_id, client)
        print(results)

#if __name__ == "__main__":
    import httpx
    import asyncio
    asyncio.run(test_animeworld())