FEAT: Added Plex Module
FIX: Fixed Spotify Module Fix: Fixed Sync Issues DOC: Added .env references for new APIs
This commit is contained in:
85
modules/plex_module.py
Normal file
85
modules/plex_module.py
Normal file
@ -0,0 +1,85 @@
|
||||
import discord
|
||||
from discord import app_commands
|
||||
from plexapi.server import PlexServer
|
||||
import config
|
||||
|
||||
|
||||
class PlexModule:
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.plex = PlexServer(config.PLEX_URL, config.PLEX_TOKEN)
|
||||
self.add_commands()
|
||||
|
||||
def add_commands(self):
|
||||
@app_commands.command(
|
||||
name="list_libraries", description="List all Plex libraries"
|
||||
)
|
||||
async def list_libraries(interaction: discord.Interaction):
|
||||
await interaction.response.defer()
|
||||
try:
|
||||
libraries = self.plex.library.sections()
|
||||
library_names = [lib.title for lib in libraries]
|
||||
await interaction.followup.send(
|
||||
f"Plex Libraries: {', '.join(library_names)}"
|
||||
)
|
||||
except Exception as e:
|
||||
await interaction.followup.send(f"An error occurred: {e}")
|
||||
|
||||
@app_commands.command(
|
||||
name="search_library",
|
||||
description="Search for a movie in a library"
|
||||
)
|
||||
async def search_library(
|
||||
interaction: discord.Interaction, library: str, query: str
|
||||
):
|
||||
await interaction.response.defer()
|
||||
try:
|
||||
lib = self.plex.library.section(library)
|
||||
results = lib.search(query)
|
||||
if not results:
|
||||
await interaction.followup.send(
|
||||
f"No results found for '{query}' in '{library}'"
|
||||
)
|
||||
return
|
||||
|
||||
result_titles = [result.title for result in results]
|
||||
await interaction.followup.send(
|
||||
f"Search results in {library}: {', '.join(result_titles)}"
|
||||
)
|
||||
except Exception as e:
|
||||
await interaction.followup.send(f"An error occurred: {e}")
|
||||
|
||||
@app_commands.command(
|
||||
name="play_movie", description="Play a movie on a specified Plex "
|
||||
"client"
|
||||
)
|
||||
async def play_movie(
|
||||
interaction: discord.Interaction, client_name: str, movie_name: str
|
||||
):
|
||||
await interaction.response.defer()
|
||||
try:
|
||||
client = next(
|
||||
(c for c in self.plex.clients() if c.title == client_name),
|
||||
None
|
||||
)
|
||||
if not client:
|
||||
await interaction.followup.send(
|
||||
f"No client found with the name '{client_name}'"
|
||||
)
|
||||
return
|
||||
|
||||
movie = self.plex.library.section('Movies').get(movie_name)
|
||||
client.playMedia(movie)
|
||||
await interaction.followup.send(
|
||||
f"Playing '{movie_name}' on '{client_name}'"
|
||||
)
|
||||
except Exception as e:
|
||||
await interaction.followup.send(f"An error occurred: {e}")
|
||||
|
||||
self.bot.tree.add_command(list_libraries)
|
||||
self.bot.tree.add_command(search_library)
|
||||
self.bot.tree.add_command(play_movie)
|
||||
|
||||
|
||||
async def setup(bot):
|
||||
PlexModule(bot)
|
@ -30,10 +30,12 @@ class SpotifyModule:
|
||||
try:
|
||||
current = self.sp.currently_playing()
|
||||
if current is None or current["item"] is None:
|
||||
await interaction.followup.send("No song is currently playing")
|
||||
await interaction.followup.send(
|
||||
"No song is currently playing"
|
||||
)
|
||||
else:
|
||||
track = current["item"]
|
||||
artist = ", ".join([artist["name"] for artist in track["artists"]])
|
||||
artist = ", ".join([a["name"] for a in track["artists"]])
|
||||
await interaction.followup.send(
|
||||
f"Currently playing: {track['name']} by {artist} "
|
||||
f"({track['album']['name']})"
|
||||
@ -57,13 +59,16 @@ class SpotifyModule:
|
||||
|
||||
devices = self.sp.devices()
|
||||
if not devices["devices"]:
|
||||
await interaction.followup.send("No active devices found. Please open Spotify on a device.")
|
||||
await interaction.followup.send(
|
||||
"No active devices found. Please open Spotify on a "
|
||||
"device."
|
||||
)
|
||||
return
|
||||
|
||||
self.sp.start_playback(uris=[uri])
|
||||
await interaction.followup.send(
|
||||
f"Now playing: {track['name']} by "
|
||||
f"{', '.join([artist['name'] for artist in track['artists']])} "
|
||||
f"{', '.join([a['name'] for a in track['artists']])} "
|
||||
f"({track['album']['name']})"
|
||||
)
|
||||
except Exception as e:
|
||||
@ -109,7 +114,9 @@ class SpotifyModule:
|
||||
await interaction.response.defer()
|
||||
try:
|
||||
self.sp.previous_track()
|
||||
await interaction.followup.send("Returned to the previous track.")
|
||||
await interaction.followup.send(
|
||||
"Returned to the previous track."
|
||||
)
|
||||
except Exception as e:
|
||||
await interaction.followup.send(f"An error occurred: {e}")
|
||||
|
||||
|
Reference in New Issue
Block a user