guess_the_movie.py

import discord
from discord.ext import commands
from dataclasses import dataclass

from cinevoraces.env_variables import load_env_variables, check_env_variables
from cinevoraces.tmdb_movie import get_movie, get_random_picture_from_movie
from cinevoraces.cinevoraces_movie import get_random_movie_title

env_variables = load_env_variables()

# Create a bot instance
bot = commands.Bot(command_prefix="!", intents=discord.Intents.all()) # Create a bot instance

@bot.event
async def on_ready():
    # Check that all environment variables are defined
    check_env_variables(env_variables)
    
    channel = bot.get_channel(int(env_variables['GAME_CHANNEL_ID']))
    print(f"{bot.user.name} is listenning to {channel.name}")
    print("------")
    await channel.send(f"Bonjour, {bot.user.name} est prêt à lancer une partie.\nPour commencer, entrez !begin_guess_movie.\nPour répondre, entrez !my_guess \"Titre du film\".\nPour abandonner, entrez !give_up")

# Guess the movie from a given picture, from Cinévoraces database

@dataclass
class Game:
    is_active: bool = False
    movie_title: str = ""

game = Game()

@bot.command()
async def begin_guess_movie(ctx):
    # Check if a game is already active
    if game.is_active:
        await ctx.send("Une partie est déjà en cours !")
        return
    
    # Get a random movie from cinévoraces database
    random_cinevoraces_movie_title, error = get_random_movie_title(env_variables)
    if error:
        await ctx.send(error['message'])
    print(random_cinevoraces_movie_title)

    # Search for the movie on TMDB and get its id
    tmdb_movie, error = get_movie(env_variables, query=random_cinevoraces_movie_title)
    if error:
        await ctx.send(error['message'])
    tmdb_movie_id = tmdb_movie['id']
    print(tmdb_movie_id)
    
    # Get a random picture from the movie
    image, error = get_random_picture_from_movie(env_variables, tmdb_movie_id)
    if error:
        await ctx.send(error['message'])
    
    image_url = f"https://www.themoviedb.org/t/p/original{image['file_path']}"
    await ctx.send(f"Devinez le film à partir de cette image !\n\n{image_url}")
    game.is_active = True
    game.movie_title = random_cinevoraces_movie_title

@bot.command()
async def my_guess(ctx, movie_title_guess):
    if game.is_active == False:
        await ctx.send(f"Aucune partie en cours, lancez-vous !")
        return
    # Check if the movie_title is correct (case insensitive)
    # If it is, stop the game and congratulate the player
    if not movie_title_guess.lower() == game.movie_title.lower():
        await ctx.send(f"Il ne s'agit pas de {movie_title_guess} ! Essayez encore !")
        return

    await ctx.send(f"Bravo, vous avez trouvé le film ! Il s'agissait bien de {game.movie_title}")
    game.is_active = False

@bot.command()
async def give_up(ctx):
    if game.is_active == False:
        await ctx.send(f"Aucune partie en cours, lancez-vous !")
        return
    
    # Stops the game and give the answer
    await ctx.send(f"Le film à deviner était {game.movie_title}")
    game.is_active = False
    return

# Run the bot
bot.run(env_variables['BOT_TOKEN']) # Run the bot