mirror of
https://github.com/jarulsamy/Plex-Bot.git
synced 2024-08-19 15:01:55 +02:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import command
|
|
from fuzzywuzzy import fuzz
|
|
from plexapi.server import PlexServer
|
|
|
|
|
|
class General(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@command()
|
|
async def stop(self, ctx):
|
|
await ctx.send(f"Stopping upon the request of {ctx.author.mention}")
|
|
await self.bot.close()
|
|
|
|
|
|
class Plex(commands.Cog):
|
|
def __init__(self, bot, base_url, plex_token, lib_name) -> None:
|
|
self.bot = bot
|
|
self.base_url = base_url
|
|
self.plex_token = plex_token
|
|
self.library_name = lib_name
|
|
|
|
self.pms = PlexServer(self.base_url, self.plex_token)
|
|
self.music = self.pms.library.section(self.library_name)
|
|
|
|
def _search_tracks(self, title):
|
|
tracks = self.music.searchTracks()
|
|
score = [[None], -1]
|
|
for i in tracks:
|
|
# scores[i].append(fuzz.ratio(title.lower(), i.lower))
|
|
s = fuzz.ratio(title.lower(), i.title.lower())
|
|
if s > score[1]:
|
|
score[0] = [i]
|
|
score[1] = s
|
|
elif s == score[1]:
|
|
score[0].append(i)
|
|
|
|
return score
|
|
|
|
@command()
|
|
async def hello(self, ctx, *, member: discord.member = None):
|
|
member = member or ctx.author
|
|
await ctx.send(f"Hello {member}")
|
|
|
|
@command()
|
|
async def play(self, ctx, *args):
|
|
title = " ".join(args)
|
|
track = self._search_tracks(title)
|
|
if track[0][0]:
|
|
await ctx.send(track[0][0].title)
|
|
else:
|
|
await ctx.send("Song not found!")
|