info.py

from loadenv import load_env
from tmdbv3api import TMDb, Movie, TV
from convert_date import convert_US_date, convert_IT_date
import requests
import config
import json
SC_FAST_SEARCH = config.SC_FAST_SEARCH
TF_FAST_SEARCH = config.TF_FAST_SEARCH
env_vars = load_env()
TMDB_KEY = env_vars.get('TMDB_KEY')


def get_info_tmdb(tmbda,ismovie,type):
    tmdb = TMDb()
    tmdb.api_key = f'{TMDB_KEY}'
    tmdb.language = 'it'
    if ismovie == 0:
        tv = TV()
        show= tv.details(tmbda)
        showname = show.name
        if type == "Filmpertutti":
            date= show.first_air_date
            print("Real date",date)
            return showname,date
        elif type == "StreamingCommunity":
            if SC_FAST_SEARCH == "0":
                n_season = show.number_of_seasons
                full_date = show.first_air_date
                date = full_date.split("-")[0]
                print(date)
                return showname,date
            else:
                return showname
        elif type == "Tuttifilm":
            if TF_FAST_SEARCH == "0":
                date = show.first_air_date
                date = date.split("-")[0]
                print("Real date",date)
                return showname,date
            else:
                return showname
        elif type == "Cool":
            return showname
        elif type == "LordChannel":
            date = show.first_air_date
            date = date.split("-")[0]
            print("Real date",date)
            return showname,date
        elif type == "StreamingWatch":
            date = show.first_air_date
            date = date.split("-")[0]
            print("Real date",date)
            return showname,date
    
    elif ismovie == 1:
        movie = Movie()
        show= movie.details(tmbda)
        showname= show.title
        #Get all release dates
        if type == "Filmpertutti":
            date = show.release_dates
            #GET US RELEASE DATE because filmpertutti somewhy uses US release date
            date = convert_US_date(date)
            return showname,date
        elif type == "StreamingCommunity":
            return showname
        elif type == "Tuttifilm":
            if TF_FAST_SEARCH == "0":
                date = show.release_date
                date = date.split("-")[0]
                print("Real date",date)
                return showname,date
            else:
                return showname
        elif type == "Cool":
            return showname
        elif type == "LordChannel":
            date = show.release_date
            date = date.split("-")[0]
            print("Real date",date)
            return showname,date
        elif type == "StreamingWatch":
            date = show.release_date
            date = date.split("-")[0]
            print("Real date",date)
            return showname,date

async def get_info_imdb(imdb_id, ismovie, type,client):

    resp = await client.get(f'https://api.themoviedb.org/3/find/{imdb_id}?api_key={TMDB_KEY}&language=it&external_source=imdb_id')
    data = resp.json()
    if ismovie == 0:     
        showname = data['tv_results'][0]['name']
        if type == "Filmpertutti":
            date= data['tv_results'][0]['first_air_date']
            print("Real date",date)
            return showname, date
        elif type == "StreamingCommunity":
            return showname
        elif type == "Tuttifilm":
            if TF_FAST_SEARCH == "0":
                date = data['tv_results'][0]['first_air_date']
                date = date.split("-")[0]
                return showname,date
            elif TF_FAST_SEARCH == "1":
                return showname
        elif type == "Cool":
            return showname

    elif ismovie == 1:
        showname= data['movie_results'][0]['title']
        if type == "Filmpertutti":
            return
        elif type == "StreamingCommunity":
            return showname
        elif type == "Tuttifilm":
            date = data['movie_results'][0]['release_date']
            date = date.split("-")[0]
            return showname,date
        elif type == "Cool":
            return showname

async def get_info_kitsu(kitsu_id,client):
    api_url = f'https://kitsu.io/api/edge/anime/{kitsu_id}'
    response = await client.get(api_url)
    data = json.loads(response.text)
    showname = data['data']['attributes']['canonicalTitle']
    date = data['data']['attributes']['startDate']
    return showname,date           




def is_movie(imdb_id):
    if "tmdb:" in imdb_id:
        imdb_id = imdb_id.replace("tmdb:","")
    if ":"  in imdb_id:
        season = imdb_id.split(":")[1]
        episode = imdb_id.split(":")[-1]
        ismovie = 0
        imdb_id = imdb_id.split(":")[0]
        return ismovie,imdb_id,season,episode
    else:
        ismovie = 1
        return ismovie,imdb_id