106 lines
4.7 KiB
Python
106 lines
4.7 KiB
Python
import logging
|
|
import tempfile
|
|
from os import path, makedirs, rename, remove
|
|
|
|
from disnake.ext import commands
|
|
|
|
from lib.DB import read_db, check_exist_audio, add_audio
|
|
from lib.Player import play_audio
|
|
|
|
|
|
class Audio(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.Cog.listener()
|
|
async def on_ready(self):
|
|
logging.info(f'Cog {__name__.split(".")[1]} is ready!.')
|
|
|
|
@commands.Cog.listener()
|
|
async def on_voice_state_update(self, member, before, after):
|
|
from lib.Comands import read_json
|
|
_role = await read_json(member.guild.id, 'tigger_role')
|
|
# Read audio from DB
|
|
audio_db = await read_db(member.guild.id, member.id, 'usertracks')
|
|
def_audio_db = await read_db(member.guild.id, member.id, 'defaulttracks')
|
|
print(f'Trigger role: {_role}')
|
|
print(f' Memder roles {member.roles}')
|
|
print(role.id in _role for role in member.roles)
|
|
|
|
print(f'Audio DB: {audio_db}')
|
|
print(f'Def audio DB: {def_audio_db}')
|
|
print(f' Memder is bot {member.bot}')
|
|
print(f' Before channel {before.channel}')
|
|
print('------------------------------------------')
|
|
|
|
if audio_db is not None:
|
|
audio_db = audio_db.split(', ') # Need to fix creating list
|
|
for i in range(len(audio_db)):
|
|
audio_db[i] = f'{member.id}/{audio_db[i]}'
|
|
if def_audio_db is not None:
|
|
def_audio_db = def_audio_db.split(', ')
|
|
from lib.Comands import list_files
|
|
def_audio_ls = await list_files()
|
|
if before.channel is None and not member.bot:
|
|
if def_audio_db or audio_db is not None:
|
|
if def_audio_db is None: def_audio_db = []
|
|
if audio_db is None: audio_db = []
|
|
print(audio_db, def_audio_db)
|
|
logging.info(f'Play audio from DB')
|
|
full_audio = def_audio_db + audio_db
|
|
|
|
if not self.bot.voice_clients:
|
|
await play_audio(full_audio, self.bot, after.channel)
|
|
elif (role.id in _role for role in member.roles):
|
|
logging.info(f'Play audio from list by role')
|
|
if not self.bot.voice_clients:
|
|
await play_audio(def_audio_ls, self.bot, after.channel)
|
|
else:
|
|
logging.info(f'Skip playing')
|
|
|
|
@commands.command(name="upload_audio")
|
|
async def upload_audio(self, ctx, user=None):
|
|
user = user or ctx.author
|
|
print(tempfile.tempdir)
|
|
if ctx.author.guild_permissions.administrator or user is ctx.author:
|
|
if ctx.message.attachments:
|
|
from os import error
|
|
if not path.isdir(f'audio/{user.id}'):
|
|
try:
|
|
makedirs(f'audio/{user.id}')
|
|
except error as _error:
|
|
pass
|
|
for at in ctx.message.attachments:
|
|
import mimetypes
|
|
|
|
await at.save(f'{tempfile.tempdir}/{tempfile.mkdtemp(dir=f"tmp/{user.id}")}/{at.filename}')
|
|
guess = mimetypes.guess_type(f'{tempfile.tempdir}/{user.id}/{at.filename}')
|
|
if guess[0].split('/')[0] == 'audio':
|
|
from pymediainfo import MediaInfo
|
|
file = f'{tempfile.tempdir}/{user.id}/{at.filename}'
|
|
duration = round(MediaInfo.parse(file).tracks[0].duration / 1000)
|
|
if duration > 20:
|
|
await ctx.reply(f'Audio duration is {duration}, but max is 15')
|
|
remove(f'{tempfile.tempdir}/{user.id}/{at.filename}')
|
|
else:
|
|
a = await read_db(ctx.guild.id, user.id, 'usertracks')
|
|
if a:
|
|
audiolist = a + ", " + f'{at.filename}'
|
|
else:
|
|
audiolist = f'{at.filename}'
|
|
|
|
await check_exist_audio(ctx, ctx.guild.id, user.id, 'usertracks', at.filename)
|
|
await add_audio(ctx.guild.id, user.id, audiolist)
|
|
rename(f'{tempfile.tempdir}/{user.id}/{at.filename}', f'audio/{user.id}/{at.filename}')
|
|
elif guess[0].split('/')[0] != 'audio':
|
|
await ctx.reply(f'It not audio {at.filename}\n it`s {guess[0]}')
|
|
remove(f'{tempfile.tempdir}/{user.id}/{at.filename}')
|
|
else:
|
|
await ctx.reply("Has no Attachment")
|
|
else:
|
|
await ctx.reply(f'You`re not admin. You can add audio only for your own account')
|
|
|
|
|
|
def setup(bot): # an extension must have a setup function
|
|
bot.add_cog(Audio(bot)) # adding a cog
|