New Version

Src/API/toonitalia.py

@@ -0,0 +1,121 @@
+from fake_headers import Headers
+
+random_headers = Headers()
+
+import  Src.Utilities.config as config
+
+from Src.Utilities.info import get_info_imdb,get_info_tmdb
+from urllib.parse import quote
+from Src.Utilities.convert import get_IMDB_id_from_TMDb_id
+from Src.Utilities.info import get_info_tmdb, is_movie, get_info_imdb
+from Src.API.extractors.voe import voe
+from Src.API.extractors.maxstream import maxstream
+import logging
+from Src.Utilities.config import setup_logging
+level = config.LEVEL
+import re
+
+import random
+logger = setup_logging(level)
+from Src.Utilities.loadenv import load_env  
+env_vars = load_env()
+TI_DOMAIN = config.TI_DOMAIN
+
+TI_PROXY = config.TI_PROXY
+proxies = {}
+if TI_PROXY == "1":
+    PROXY_CREDENTIALS = env_vars.get('PROXY_CREDENTIALS')
+    proxy_list = json.loads(PROXY_CREDENTIALS)
+    proxy = random.choice(proxy_list)
+    if proxy == "":
+        proxies = {}
+    else:
+        proxies = {
+            "http": proxy,
+            "https": proxy
+        }   
+TI_ForwardProxy = config.TI_ForwardProxy
+if TI_ForwardProxy == "1":
+    ForwardProxy = env_vars.get('ForwardProxy')
+else:
+    ForwardProxy = ""
+
+
+
+
+
+async def search(showname,date,season,episode,MFP,MFP_CREDENTIALS,ismovie,client,streams):
+    headers = random_headers.generate()
+    response = await client.get(ForwardProxy + f'{TI_DOMAIN}/wp-json/wp/v2/search?search={quote(showname)}',headers=headers, proxies = proxies)
+    data = response.json()
+    for item in data:
+        response = await client.get(ForwardProxy + item['_links']['self'][0]['href'],headers=headers, proxies = proxies)
+        info = response.json()
+        html = info['content']['rendered']
+        if 'Stagione' in html and ismovie == 1:
+            continue
+        if 'Stagione' not in html and ismovie == 0:
+            continue
+        if ismovie == 0:
+            episode = episode.zfill(2)
+            pattern = rf'{season}&#215;{episode}.*?<a[^>]*href="([^"]*)'
+        if ismovie == 1:
+            pattern = r'Link.*href="([^"]*)(?=.*VOE)'
+        match = re.search(pattern,html)
+        if match:
+            url = match.group(1)
+        if 'maxstream' in url:
+            parts = url.split('.video/')
+            url = 'https://maxstream.video/emvvv/' + parts[1]
+            streams = await maxstream(url,client,streams,'ToonItalia','',proxies,ForwardProxy)
+        else:
+            streams = await voe(url,streams,'ToonItalia',MFP,MFP_CREDENTIALS,proxies,ForwardProxy,client)
+        return streams
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+async def toonitalia(streams,id,client,MFP,MFP_CREDENTIALS):
+    try:
+        general = await is_movie(id)
+        ismovie = general[0]
+        clean_id = general[1]
+        type = "Toonitalia"
+        if ismovie == 0 : 
+            season = str(general[2])
+            episode = str(general[3])
+        elif ismovie == 1:
+            season = None
+            episode = None
+        if "tmdb" in id:
+            showname,date = get_info_tmdb(clean_id,ismovie,type)
+        else:
+            showname,date = await get_info_imdb(clean_id,ismovie,type,client)
+        streams = await search(showname,date, season,episode,MFP,MFP_CREDENTIALS,ismovie,client,streams)
+        return streams
+    except Exception as e:
+        print(e)
+        return streams
+
+
+
+async def test_toonitalia():
+    from curl_cffi.requests import AsyncSession
+    async with AsyncSession() as client:
+        results = await toonitalia({'streams': []},"tt1466074:1:1",client,"0",['test','test'])
+        print(results)
+
+        
+if __name__ == "__main__":
+    import asyncio
+    asyncio.run(test_toonitalia())