Projet_multi_stream_obs/test/main.py

from tkinter import *
from tkinter import messagebox
import mysql.connector
from obswebsocket import obsws, requests as obsrequests
from twitch.helix.api import TwitchHelix
import webbrowser
import asyncio

class NouvelleFenetre:

    is_live = False

    client_id = '6h883my3p6ndqozotnl6nsbllyyr3h' # id de l'app
    oauth_token ='z11zvdhgxvemn2l8kjl1dvczgiyelt' # token de validation, à changer selon l'utilisateur

    oauth_authorization_url = f"https://id.twitch.tv/oauth2/authorize?client_id={'6h883my3p6ndqozotnl6nsbllyyr3h'}&redirect_uri={'http://localhost'}&response_type=token&scope={'user_read'}"
    
    # Fonction permettant de fermer la deuxième fenetre créee et la première fenetre de connexion de se réafficher tout en se déconnecctant d'OBS
    def fermer_et_reafficher(self):
        self.root.destroy()
        self.ws.disconnect()
        self.root.master.deiconify()

    # Fonction permettant de récupérer la valeur de la clé twitch qui a été renseignée
    def get_key_twitch(self):
        self.twitch = self.entry_twitch.get()
        print(self.twitch)

    # Fonction permettant de récupérer la valeur de la clé youtube qui a été renseignée
    def get_key_youtube(self):
        self.youtube = self.entry_youtube.get()
        print(self.youtube)

    def set_key_twitch(self, username):
        try:
            # Paramètres du serveur
            cnx = mysql.connector.connect(user='root', password='',
                                        host='127.0.0.1',
                                        database='obs_project')
            cur = cnx.cursor()

            # On lance une requête pour vérifier si l'utilisateur correspond à la base de données
            query = "SELECT * FROM utilisateurs WHERE username = %s"
            cur.execute(query, (username,))

            # Renvoie la première ligne de la base de données qui correspond
            user = cur.fetchone()

            if user:
                # Utilisateur trouvé, on met à jour la clé Twitch
                new_twitch_key = self.entry_twitch.get()
                update_query = "UPDATE twitch SET twitch_key = %s WHERE username = %s"
                cur.execute(update_query, (new_twitch_key, username))
                cnx.commit()
                print(f"La clé a été mis à jour pour {username} avec la clé {new_twitch_key}")
            else:
                print(f"Utilisateur {username} non trouvée")

        except mysql.connector.Error as err:
            print(f"Erreur: {err}")
            messagebox.showerror("Erreur", "Une erreur a eu lieu pour mettre à jour votre clé")

        finally:
            # On ferme le tout, la connexion etc...
            cur.close()
            cnx.close()

    def set_key_youtube(self, username):
        try:
            # Paramètres du serveur
            cnx = mysql.connector.connect(user='root', password='',
                                        host='127.0.0.1',
                                        database='obs_project')
            cur = cnx.cursor()

            # On lance une requête pour vérifier si l'utilisateur correspond à la base de données
            query = "SELECT * FROM utilisateurs WHERE username = %s"
            cur.execute(query, (username,))

            # Renvoie la première ligne de la base de données qui correspond
            user = cur.fetchone()

            if user:
                # Utilisateur trouvé, on met à jour la clé Twitch
                new_youtube_key = self.entry_youtube.get()
                update_query = "UPDATE youtube SET youtube_key = %s WHERE username = %s"
                cur.execute(update_query, (new_youtube_key, username))
                cnx.commit()
                print(f"La clé a été mis à jour pour {username} avec la clé {new_youtube_key}")
            else:
                print(f"Utilisateur {username} non trouvée")

        except mysql.connector.Error as err:
            print(f"Erreur: {err}")
            messagebox.showerror("Erreur", "Une erreur a eu lieu pour mettre à jour votre clé")

        finally:
            # On ferme le tout, la connexion etc...
            cur.close()
            cnx.close()

    def get_twitch_key_from_database(self, username):
        try:
            # Paramètres du serveur
            cnx = mysql.connector.connect(user='root', password='',
                                        host='127.0.0.1',
                                        database='obs_project')
            cur = cnx.cursor()

            # On lance une requête pour récupérer la clé Twitch de l'utilisateur
            query = "SELECT twitch_key FROM twitch WHERE username = %s"
            cur.execute(query, (username,))
            twitch_key = cur.fetchone()

            if twitch_key:
                return twitch_key[0]
            else:
                print(f"Clé Twitch non trouvée pour l'utilisateur {username}")
                return None

        except mysql.connector.Error as err:
            print(f"Erreur: {err}")
            messagebox.showerror("Erreur", "Une erreur a eu lieu pour récupérer la clé Twitch")

        finally:
            # On ferme le tout, la connexion etc...
            cur.close()
            cnx.close()

    async def configure_obs_for_twitch(self, twitch_key_from_database):
        try:
            if twitch_key_from_database is not None:
                if not self.is_live:
                    print("Connexion à OBS via Websocket avec les paramètres suivants :")
                    print("Host:", self.obs_host)
                    print("Port:", self.obs_port)
                    print("Password:", self.obs_password)
                    print("Twitch Stream Key:", twitch_key_from_database)

                    self.obs_ws = obsws(self.obs_host, self.obs_port, self.obs_password)
                    self.obs_ws.connect()

                    # Démarre le streaming sur Twitch
                    start_stream_request = obsrequests.StartStreaming(stream_key=twitch_key_from_database)
                    self.obs_ws.call(start_stream_request)

                    self.is_live = True
                else:
                    # Si déjà en live, arrêter le streaming
                    print("Arrêt du streaming Twitch.")
                    self.obs_ws.call(obsrequests.StopStreaming())
                    self.obs_ws.disconnect()
                    self.is_live = False
            else:
                print("La clé Twitch est manquante.")
        except Exception as e:
            print(f"Une erreur s'est produite lors de la configuration d'OBS pour Twitch : {e}")


    async def initialize_twitch(self):
        try:
            twitch = TwitchHelix(self.client_id, self.oauth_token)
            users = twitch.get_users()
            if users:
                print("Utilisateur Twitch trouvé:")
                print("ID:", users[0].id)
                print("Nom d'utilisateur:", users[0].display_name)
            else:
                print("Utilisateur Twitch non trouvé.")
        except Exception as e:
            print(f"Une erreur s'est produite lors de la récupération de l'utilisateur Twitch : {e}")

    async def initialize_window(self, root, username, password):
        self.root = Toplevel(root)
        self.root.title("OBS Extension Multistreaming")
        self.root.geometry("1920x1080")

        self.lancer_auth_twitch()
        await self.initialize_twitch()

    def __init__(self, root, username, password):
        asyncio.run(self.initialize_window(root, username, password))

        self.root.grid_rowconfigure(0, weight=0)
        self.root.grid_rowconfigure(1, weight=0)
        self.root.grid_rowconfigure(2, weight=0)
        self.root.grid_rowconfigure(3, weight=0)
        self.root.grid_rowconfigure(4, weight=0)
        self.root.grid_rowconfigure(5, weight=0)
        self.root.grid_rowconfigure(6, weight=0)
        self.root.grid_rowconfigure(7, weight=0)
        self.root.grid_rowconfigure(8, weight=0)
        self.root.grid_rowconfigure(9, weight=0)
        self.root.grid_rowconfigure(10, weight=0)
        self.root.grid_rowconfigure(11, weight=0)
        self.root.grid_rowconfigure(12, weight=0)
        self.root.grid_rowconfigure(13, weight=0)
        self.root.grid_rowconfigure(14, weight=0)
        self.root.grid_columnconfigure(0, weight=1)
        self.root.grid_columnconfigure(1, weight=0)
        self.root.grid_columnconfigure(2, weight=0)
        self.root.grid_columnconfigure(3, weight=0)

        # Initialisez la connexion OBS WebSocket
        self.obs_host = "localhost"  # Adresse IP de votre ordinateur où OBS est en cours d'exécution
        self.obs_port = 4455  # Port par défaut utilisé par OBSWebsocket
        self.obs_password = "MPcZLZfViZihHqmg"  # Mot de passe OBSWebsocket


        # On récupère la clé Twitch depuis la base de données
        self.twitch_key_from_database = self.get_twitch_key_from_database(username)

        # On affiche bienvenue à la personne s'étant identifiée, en récupérant la variable lastname
        label = Label(self.root, text=f"Bienvenue, {username}!")
        label.grid(row=0, column=0)

        label = Label(self.root, text="Clé Twitch :")
        label.grid(row=2, column=0)

        self.entry_twitch = Entry(self.root)
        self.entry_twitch.grid(row=3, column=0)

        # On met un bouton de validation pour garder les valeurs qu'on a rentré précédement
        boutonvalidation = Button(self.root, text="Envoyé", command=lambda: self.set_key_twitch(username))
        boutonvalidation.grid(row=4, column=0)

        # On met un bouton de diffusion pour twitch
        boutondediffusiontwitch = Button(self.root, text="Diffusion de Twitch On/Off", command=lambda: asyncio.run(self.configure_obs_for_twitch(self.twitch_key_from_database)))
        boutondediffusiontwitch.grid(row=4, column=3)

        label = Label(self.root, text="Clé Youtube :")
        label.grid(row=5, column=0)

        self.entry_youtube = Entry(self.root)
        self.entry_youtube.grid(row=6, column=0)

        # On met un bouton pour envoyer les valeurs remplies du dessus
        boutonvalidation2 = Button(self.root, text="Envoyé", command=lambda: self.set_key_youtube(username))
        boutonvalidation2.grid(row=7, column=0)

        # On met un bouton de diffusion pour twitch
        boutondediffusionyoutube = Button(self.root, text="Diffusion de YouTube On/Off")
        boutondediffusionyoutube.grid(row=7, column=3)

        # On met un boutton fermer, pour pouvoir fermer la page main, et repasser sur la fenetre de connexion
        bouton_fermer = Button(self.root, text="Fermer", command=self.fermer_et_reafficher)
        bouton_fermer.grid(row=13, column=0)

    def lancer_auth_twitch(self):
        # Vérifier si la colonne OAuth est vide dans la base de données
        oauth_token = self.get_oauth_token_from_database()
        if oauth_token is None:
            # Si la colonne OAuth est vide, ouvrir la page d'authentification Twitch
            webbrowser.open(self.oauth_authorization_url)
            messagebox.showinfo("Authentification Twitch", "Veuillez autoriser l'accès à Twitch dans le navigateur Web qui s'est ouvert.")
        else:
            # Si la colonne OAuth n'est pas vide, afficher un message indiquant que l'authentification n'est pas nécessaire
            messagebox.showinfo("Authentification Twitch", "L'authentification Twitch n'est pas nécessaire car le token OAuth est déjà présent dans la base de données.")

    def get_oauth_token_from_database(self):
        # Code pour récupérer la valeur de la colonne OAuth dans la base de données
        # Retourne la valeur de la colonne OAuth
        return "z11zvdhgxvemn2l8kjl1dvczgiyelt"