bacon/discord_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Finished audio list generator

Browse Source
This commit is contained in:
beacon
2022-09-01 15:15:30 +03:00
parent 5b134e6446
commit 3c0851102d
7 changed files with 139 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cogs/audio.py
View File

@@ -21,7 +21,6 @@ class Audio(commands.Cog, name='Audio'):
# todo: complete check activity
@commands.Cog.listener()
async def on_voice_state_update(self, member, before, after):
print(type(member.roles))
if before.channel is None and not member.bot:
if any('Escape from Tarkov' in str(user.activity) for user in after.channel.members):
logger.info('Skip playing by Game')
@@ -46,12 +45,12 @@ class Audio(commands.Cog, name='Audio'):
if audio_db is None: audio_db = []
logger.info(f'Play audio from DB')
full_audio = def_audio_db + audio_db
await play_audio(random.choice(full_audio), self.bot, after.channel)
await play_audio(f'audio/{random.choice(full_audio)}', self.bot, after.channel)
elif len(member.roles) == 1 or _role is None:
logger.info(f'Skip playing by role')
elif any(str(role.id) in _role for role in member.roles):
logger.info(f'Play audio from list by role')
await play_audio(random.choice(def_audio_ls), self.bot, after.channel)
await play_audio(f'audio/{random.choice(def_audio_ls)}', self.bot, after.channel)
else:
logger.info(f'Skip playing by any else')

1
cogs/info.py
View File

@@ -15,6 +15,7 @@ class Bot_info(commands.Cog, name='Bot Info'):
@commands.Cog.listener() # this is a decorator for events/listeners
async def on_ready(self):
logger.info(f'Cog {__name__.split(".")[1]} is ready!.')
logger.info(f'{[names.name for names in self.bot.users]}')
@commands.slash_command(name="info_bot",
description='Shows general info about bot') # this is for making a command

33
cogs/test.py
View File

@@ -4,7 +4,7 @@ import disnake
from disnake import Option, OptionType, OptionChoice
from disnake.ext import commands
from lib import list_files
from lib import ListGenerator
from lib import logger
from lib import play_audio
@@ -18,38 +18,49 @@ class Testing(commands.Cog, name='Testing'):
logger.info(f'Cog {__name__.split(".")[1]} is ready!.')
@commands.slash_command(name="play_audio",
description="Make possible playing audio by command",
options=[
Option(name="audio",
type=OptionType.string,
required=True
# choices=OptionChoice(list_to_play)
)
])
async def playaudio(self, inter: disnake.ApplicationCommandInteraction,
audio: str
):
if inter.author.voice is not None:
await inter.response.send_message(f'Played {audio}')
await inter.response.send_message(f'Played {audio}', ephemeral=True)
await play_audio(audio, self.bot, inter.author.voice.channel)
else:
await inter.response.send_message('You`re not in voice')
await inter.response.send_message('You`re not in voice', ephemeral=True)
@playaudio.autocomplete('audio')
async def list_to_play(self, inter: disnake.ApplicationCommandInteraction, current: str) -> List[OptionChoice]:
def_list = await list_files()
_def_iter = ListGenerator('audio')
_def_dict: dict = {}
for f in _def_iter:
_def_dict[f.name] = f'{f.path}/{f.name}'
_user_dict: dict = {}
try:
user_list = await list_files(str(inter.author.id))
_user_iter = ListGenerator(f'audio/{inter.author.id}')
for f in _user_iter:
_user_dict[f.name] = f'{f.path}/{f.name}'
# user_dict = []
# for _track in user_list:
except IndexError:
user_list = []
pass
_list = def_list + user_list
_dict = {}
_dict.update(_def_dict)
_dict.update(_user_dict)
return [
OptionChoice(name=choice, value=choice)
for choice in def_list if current.lower() in choice.lower()
]
OptionChoice(name=choice, value=f'{_dict[choice]}')
for choice in _dict if current.lower() in choice.lower()
]
def setup(bot): # an extension must have a setup function

72
lib/DB_Worker.py
View File

@@ -1,6 +1,78 @@
import sqlite3
class User:
def __init__(self, userid: str = None):
self.userid = userid
self.list: list = self._lister(self.userid)
self._current_index = 0
def __str__(self) -> str:
return self.name
def __repr__(self) -> str:
return (
f'<Audio file name={self.name} path={self.userid} type={self.type}>'
)
@property
def name(self) -> str:
return self.list[self._current_index]
@property
def type(self) -> str:
guess = mimetypes.guess_type(f'{self.path}/{self.name}')[0].split('/')[0]
return guess
@classmethod
def _lister(cls, path) -> list:
_dict: list = []
for _files in os.walk(path):
_dict.extend(_files)
break
return _dict[2]
def __iter__(self):
return _ListGenerationIter(self)
class _FileAttrs:
def __init__(self, name, path, type):
self.name = name
self.path = path
self.type = type
def __str__(self):
return 'Audio Dict Generator'
def __repr__(self):
return f'<File attrs name={self.name} path={self.path} type={self.path}>'
class _ListGenerationIter:
def __init__(self, list_class):
self._list = list_class.list
self._name = list_class.name
self._type = list_class.type
self._path = list_class.path
self._size = len(self._list)
self._current_index = 0
def __iter__(self):
return self
def __next__(self):
if self._current_index < self._size:
_name = self._list[self._current_index]
_path = self._path
_type = self._type
self._current_index += 1
memb = _FileAttrs(_name, _path, _type)
return memb
raise StopIteration
async def prepare_db(guild: int):
try:
connect = sqlite3.connect('user.db')

89
lib/ListGenerator.py
View File

@@ -2,76 +2,66 @@ import mimetypes
import os
class FileAttrs:
def __iter__(self, name, path, type):
self.name = name
self.path = path
self.type = type
@self.name.setter
def name(self, value):
self.name = value
@self.type.setter
def type(self, value):
self.type = value
@self.path.setter
def path(self, value):
self.path = value
class ListGenerator:
def __init__(self, path=None):
self._path = path
self.list: list = self._lister(self._path)
def __init__(self, path: str = None):
self.path = path
self.list: list = self._lister(self.path)
self._current_index = 0
def __str__(self) -> str:
return self.name
if self.path:
return self.name
else:
return 'Audio iter generator'
def __repr__(self) -> str:
return (
f'<Audio file name={self.name} path={self.path} type={self.type} size={self._size}>'
)
if self.path:
return (
f'<Audio file name={self.name} path={self.path} type={self.type}>'
)
else:
return (
f'<Audio file name={None} path={None} type={None}>'
)
@property
def name(self) -> str:
return self.list[self._current_index]
@property
def path(self) -> str:
return str(self._path)
@property
def type(self) -> str:
guess = mimetypes.guess_type(f'{self._path}/{self.name}')[0].split('/')[0]
guess = mimetypes.guess_type(f'{self.path}/{self.name}')[0].split('/')[0]
return guess
@classmethod
def _lister(cls, path) -> list:
_dict: list = []
for _files in os.walk(path):
_dict.extend(_files)
break
return _dict[2]
try:
for _files in os.walk(path):
_dict.extend(_files)
break
return _dict[2]
except TypeError:
pass
def __iter__(self):
return ListGenerationIter(self)
@name.setter
def name(self, value):
self.name = value
@type.setter
def type(self, value):
self.type = value
@path.setter
def path(self, value):
self.path = value
return _ListGenerationIter(self)
class ListGenerationIter:
class _FileAttrs:
def __init__(self, name, path, type):
self.name = name
self.path = path
self.type = type
def __str__(self):
return self.name
def __repr__(self):
return f'<File attrs name={self.name} path={self.path} type={self.path}>'
class _ListGenerationIter:
def __init__(self, list_class):
self._list = list_class.list
self._name = list_class.name
@@ -86,11 +76,10 @@ class ListGenerationIter:
def __next__(self):
if self._current_index < self._size:
_name = self._list[self._current_index]
_path = self._path
_type = self._type
self._current_index += 1
memb = FileAttrs(_name, _path, _type)
memb = _FileAttrs(_name, _path, _type)
return memb
raise StopIteration

4
lib/Logger.py
View File

@@ -1,7 +1,7 @@
import logging
class CustomFormatter(logging.Formatter):
class _CustomFormatter(logging.Formatter):
grey = "\x1b[38;20m"
yellow = "\x1b[33;20m"
@@ -29,5 +29,5 @@ logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
ch.setFormatter(CustomFormatter())
ch.setFormatter(_CustomFormatter())
logger.addHandler(ch)

2
lib/Player.py
View File

@@ -10,7 +10,7 @@ async def play_audio(audio, bot, vc):
await sleep(1)
vp = await vc.connect()
if not vp.is_playing():
vp.play(FFmpegPCMAudio(f'audio/{audio}'), after=None)
vp.play(FFmpegPCMAudio(f'{audio}'), after=None)
while vp.is_playing():
await sleep(0.5)
await sleep(1)