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

up db test

Browse Source
This commit is contained in:
Slava
2022-06-27 21:50:26 +03:00
parent 611b722f9a
commit 30e9f0e10e
4 changed files with 268 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@@ -15,10 +15,8 @@ logging.basicConfig(stream=sys.stdout, filemode='w', level='INFO',
format='%(asctime)s - %(levelname)s - %(threadName)s - %(message)s')
intents = discord.Intents.all()
intents.typing = True
intents.presences = True
intents = discord.Intents.default()
intents.members = True
bot = commands.Bot(command_prefix='$', guild_subscriptions=True, intents=intents)
f = []
for filenames in walk('audio'):
@@ -29,7 +27,7 @@ f = f[2]
@bot.event
async def on_voice_state_update(member, before, after):
channel = bot.get_channel(783729824896122930)
# channel = bot.get_channel(783729824896122930)
_role = 929729495370461205
_memb = 375664768087752714
_bot_id = 946819004314570852
@@ -37,7 +35,7 @@ async def on_voice_state_update(member, before, after):
if before.channel is None and role in member.roles:
track = random.randint(0, len(f) - 1)
audio_source = FFmpegPCMAudio(f'audio/{f[track]}')
logging.error(f'{track}\t\t\t{f[track]}')
logging.error(f'{track}\t\t{f[track]}')
if not bot.voice_clients:
await sleep(1)
_channel = after.channel
@@ -66,8 +64,11 @@ async def on_voice_state_update(member, before, after):
@bot.event
async def on_member_join(member):
if member.bot == 0:
role = discord.utils.get(member.guild.roles, id=734358428939452486)
print(role)
else:
role = discord.utils.get(member.guild.roles, id=734358434945826858)
logging.info(f"Adding to {member} role {role}")
await member.add_roles(role)

364
test.py
View File

@@ -1,17 +1,18 @@
import json
import os
import logging
import sqlite3
import sys
import threading
import logging
import discord
import sqlite3
from os import walk, path, makedirs, remove, rename
from typing import List, Union, Any
from os import walk
import discord
from discord.ext import commands
from dislash import InteractionClient, Option, OptionType, SelectMenu, SelectOption
from dislash import InteractionClient, Option, OptionType
class DB:
@staticmethod
def _prepare_db(guild):
try:
connect = sqlite3.connect('user.db')
@@ -24,6 +25,7 @@ class DB:
except sqlite3.Error as error:
logging.error("failed to connecct db", error)
@staticmethod
def _work_with_db(db_func, data_turple):
try:
connect = sqlite3.connect('user.db')
@@ -35,18 +37,21 @@ class DB:
except sqlite3.Error as error:
logging.error("failed to connecct db", error)
def _fill_bd(name, userid, isbot, nick, guild):
@staticmethod
def _fill_bd(name: str, userid: int, isbot: bool, nick: str, guild: int):
sqlite_insert_with_param = (f"""INSERT OR IGNORE INTO "{guild}"
(username, userid, nick, isbot)
VALUES (?, ?, ?, ?)""")
data_tuple = (name, userid, nick, isbot)
DB._work_with_db(sqlite_insert_with_param, data_tuple)
@staticmethod
def _add_audio(guildid, user, audio):
sql_update_query = f"""UPDATE "{guildid}" set track = ? where userid = ?"""
data_tuple = (audio, user)
DB._work_with_db(sql_update_query, data_tuple)
@staticmethod
def _read_db(guildid, user):
try:
sql_con = sqlite3.connect("user.db")
@@ -54,9 +59,10 @@ class DB:
sql_read = f"""SELECT * FROM "{guildid}" where userid = {user}"""
cursor.execute(sql_read)
record = cursor.fetchone()
if record[4] is None:
return "None"
else:
# if record[4] is None:
# return "None"
# else:
# return record[4]
return record[4]
except sqlite3.Error as error:
@@ -67,38 +73,76 @@ threading.current_thread().name = "main"
logging.basicConfig(stream=sys.stdout, filemode='w', level='INFO',
format='%(asctime)s - %(levelname)s - %(threadName)s - %(message)s')
with open("config.json", 'r') as f:
prefixes = json.load(f)
default_prefix = "$"
DEFAULT_PREFIX = "$" # The prefix you want everyone to have if you don't define your own
if not path.isfile('prefix.json'):
with open('prefix.json', 'w+') as f:
f.write("")
f.close()
def prefix(bot, message):
id = message.guild.id
return prefixes.get(id, default_prefix)
def determine_prefix(bot, msg):
guild = msg.guild
base = DEFAULT_PREFIX
try:
with open('prefix.json', 'r', encoding='utf-8') as fp: # Open the JSON
jsonObject = json.load(fp) # Load the custom prefixes
except:
jsonObject = {}
if guild: # If the guild exists
try:
guild_conf = jsonObject[f"{guild.id}"]
try:
prefix = guild_conf["prefix"] # Get the prefix
base = prefix
except:
pass
except KeyError:
pass
return base
intents = discord.Intents.default()
intents.members = True
bot = commands.Bot(command_prefix=prefix, guild_subscriptions=True, intents=intents)
bot = commands.Bot(command_prefix=determine_prefix, guild_subscriptions=True, intents=intents)
inter_client = InteractionClient(bot)
class Arg:
class Commands:
async def setprefix(ctx, prefix: str):
with open('prefix.json', 'r', encoding='utf-8') as fp:
try:
jsonObject = json.load(fp)
except:
jsonObject = {}
try:
jsonObject[f"{ctx.guild.id}"]["prefix"] = prefix
except KeyError:
guild = jsonObject[f"{ctx.guild.id}"] = {'prefix': ''}
new = {"prefix": prefix}
guild.update(new)
jsonObject[str(ctx.guild.id)] = guild
await ctx.send(f"Prefix set to: `{prefix}`")
with open("prefix.json", "w") as fl:
json.dump(jsonObject, fl)
fl.close()
def list_files(folder):
f = []
for filenames in walk('audio'):
for filenames in walk(folder):
f.extend(filenames)
break
f = f[2]
dict = {}
keys = range(len(f))
for i in keys:
files = {}
for x in f:
dict[x] = x
files[x] = x
class Audio:
def add(user,):
pass
return files
@bot.event
@@ -113,13 +157,6 @@ async def on_ready():
@bot.event
async def on_guild_join(guild):
with open('config.json', 'r') as f:
prefixes = json.load(f)
prefixes[str(guild.id)] = '$'
with open('config.json', 'r') as f:
json.dump(prefixes, f, indent=4)
for g in guild.members:
DB._fill_bd(g.name, g.id, g.bot, g.nick, guild.id)
@@ -128,30 +165,100 @@ async def on_guild_join(guild):
async def on_member_join(member):
DB._fill_bd(member.name, member.id, member.bot, member.nick, member.guild.id)
try:
with open('prefix.json', 'r', encoding='utf-8') as fp: # Open the JSON
jsonObject = json.load(fp) # Load the custom prefixes
@inter_client.slash_command(
name="add-audio",
description="Add audio track to user",
options=[
Option("audio", "Specify audio name", OptionType.STRING, required=True),
Option("user", "Specify any user", OptionType.USER)
]
)
async def add_audio(ctx, audio, user=None):
user = user or ctx.author
a = DB._read_db(ctx.guild.id, user.id)
if a == "None":
audiolist = audio
except:
jsonObject = {}
if member.guild.id: # If the guild exists
try:
guild_conf = jsonObject[f"{member.guild.id}"]
try:
bot_role = guild_conf["bot_role"] # Get bot role
guest_role = guild_conf["guest_role"] # Get guest role
except:
pass
except KeyError:
pass
if bot_role and guest_role:
if member.bot == 0:
role = discord.utils.get(member.guild.roles, id=guest_role)
else:
audiolist = DB._read_db(ctx.guild.id, user.id) + ", " + audio
role = discord.utils.get(member.guild.roles, id=bot_role)
logging.info(f"Adding to {member} role {role}")
await member.add_roles(role)
@bot.event
async def on_member_update(before: discord.Member, after: discord.Member):
sql_update_query = f"""UPDATE "{after.guild.id}" set nick = ? where userid = ?"""
data_tuple = (after.nick, before.id)
DB._work_with_db(sql_update_query, data_tuple)
@bot.command(name="upload_audio")
async def upload_audio(ctx, user=None):
user = user or ctx.author
if ctx.author.guild_permissions.administrator or user is ctx.author:
if ctx.message.attachments:
from os import error
if not path.isdir(f'tmp/{user.id}'):
try:
makedirs(f'tmp/{user.id}')
except error as error:
logging.info(f"Failed to create dir", error)
if not path.isdir(f'audio/{user.id}'):
try:
makedirs(f'audio/{user.id}')
except error as error:
logging.info(f"Failed to create dir", error)
for at in ctx.message.attachments:
import mimetypes
await at.save(f'tmp/{user.id}/{at.filename}')
guess = mimetypes.guess_type(f'tmp/{user.id}/{at.filename}')
if guess[0]:
from pymediainfo import MediaInfo
file = f'tmp/{user.id}/{at.filename}'
duration = round(MediaInfo.parse(file).tracks[0].duration / 1000)
if duration > 15:
await ctx.reply(f'Audio duration is {duration}, but max is 15')
else:
a = DB._read_db(ctx.guild.id, user.id)
if a is None:
audiolist = f'{user.id}/{at.filename}'
else:
audiolist = DB._read_db(ctx.guild.id, user.id) + ", " + f'{user.id}/{at.filename}'
DB._add_audio(ctx.guild.id, user.id, audiolist)
emb = discord.Embed(
title=f"Sucseed added {audio} to {user}",
color=discord.Color.blue()
rename(f'tmp/{user.id}/{at.filename}', f'audio/{user.id}/{at.filename}')
else:
await ctx.reply(f'Failed to find MIME type of {at.filename}')
remove(f'tmp/{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')
@bot.command(name="set_prefix")
@commands.has_permissions(administrator=True)
async def command_setprefix(ctx, prefix: str):
await Commands.setprefix(ctx, prefix)
@inter_client.slash_command(
name="set_prefix",
description="Settin up bot prefix",
options=[
Option("prefix", "Specify prefix", OptionType.STRING, required=True),
]
)
emb.set_thumbnail(url=user.avatar_url)
await ctx.reply(embed=emb)
async def slash_setprefix(ctx, prefix: str):
await Commands.setprefix(ctx, prefix)
@inter_client.slash_command(
@@ -170,12 +277,11 @@ async def info(ctx, user=None):
else:
roles = "Not added any role"
if audio == "None":
if audio is None:
audios = "Not selected audio"
else:
audios = "• " + "\n• ".join(sorted(audio.split(", ")))
emb = discord.Embed(
title=f"General information",
description=f"General information on server about {user}",
@@ -193,91 +299,91 @@ async def info(ctx, user=None):
await ctx.reply(embed=emb)
@bot.command()
async def select_audio(ctx, user=None):
msg = await ctx.send(
"Select audio for user!",
components=[
SelectMenu(
custom_id="test",
placeholder="List of audio?",
max_values=2,
@inter_client.slash_command(
name="set_trigger_role",
description="Setting up role to trigger bot",
options=[
SelectOption("Option 1", "value 1"),
SelectOption("Option 2", "value 2"),
SelectOption("Option 3", "value 3")
Option("role", "Specify role", OptionType.ROLE, required=True),
]
)
]
)
# Wait for someone to click on it
inter = await msg.wait_for_dropdown()
# Send what you received
labels = [option.label for option in inter.select_menu.selected_options]
await inter.reply(f"Options: {', '.join(labels)}")
@inter_client.slash_command(description="Select audio from groups to user")
async def groups(inter):
pass
@groups.sub_command_group()
async def select_user_audio(inter, user=None):
pass
@bot.command(name="upload_audio")
async def upload_audio(ctx, user=None):
user = user or ctx.author
if ctx.author.guild_permissions.administrator or user is ctx.author:
if ctx.message.attachments:
if not os.path.isdir(f'tmp/{user.id}'):
@commands.has_permissions(administrator=True)
async def set_trigger_role(ctx, role):
with open('prefix.json', 'r', encoding='utf-8') as fp:
try:
os.makedirs(f'tmp/{user.id}')
except os.error as error:
logging.info(f"Failed to create dir", error)
for a in ctx.message.attachments:
import mimetypes
jsonObject = json.load(fp)
except:
jsonObject = {}
try:
jsonObject[f"{ctx.guild.id}"]["tigger_role"] = role.id
except KeyError:
trigger_role = jsonObject[f"{ctx.guild.id}"] = {'tigger_role': ''}
new = {"tigger_role": role.id}
trigger_role.update(new)
jsonObject[str(ctx.guild.id)] = trigger_role
await a.save(f'tmp/{user.id}/{a.filename}')
guess = mimetypes.guess_type(f'tmp/{user.id}/{a.filename}')
if guess[0]:
from pymediainfo import MediaInfo
file = f'tmp/{user.id}/{a.filename}'
duration = round(MediaInfo.parse(file).tracks[0].duration / 1000)
if duration > 15:
await ctx.reply(f'Audio duration is {duration}, but max is 10')
else:
await ctx.reply(f'Failed to find MIME type of {a.filename}')
os.remove(f'tmp/{user.id}/{a.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')
await ctx.send(f"Role to trigger set to : `{role.name}`")
with open("prefix.json", "w") as fl:
json.dump(jsonObject, fl)
fl.close()
@bot.event
async def on_member_update(before: discord.Member, after: discord.Member):
sql_update_query = f"""UPDATE "{after.guild.id}" set nick = ? where userid = ?"""
data_tuple = (after.nick, before.id)
DB._work_with_db(sql_update_query, data_tuple)
@inter_client.slash_command(
name="set_bot_role",
description="Set Default bot role",
options=[
Option("role", "Specify role", OptionType.ROLE, required=True),
]
)
@commands.has_permissions(administrator=True)
async def set_bot_role(ctx, role):
with open('prefix.json', 'r', encoding='utf-8') as fp:
try:
jsonObject = json.load(fp)
except:
jsonObject = {}
try:
jsonObject[f"{ctx.guild.id}"]["bot_role"] = role.id
except KeyError:
bot_role = jsonObject[f"{ctx.guild.id}"] = {'bot_role': ''}
new = {"bot_role": role.id}
bot_role.update(new)
jsonObject[str(ctx.guild.id)] = bot_role
await ctx.send(f"Setted up bot role to: `{role.name}`")
with open("prefix.json", "w") as fl:
json.dump(jsonObject, fl)
fl.close()
@bot.command(pass_context=True)
@commands.has_permissions(administrator=True) #ensure that only administrators can use this command
async def changeprefix(ctx, prefix): #command: bl!changeprefix ...
with open('config.json', 'r') as f:
prefixes = json.load(f)
@inter_client.slash_command(
name="set_guest_role",
description="Set Default bot role",
options=[
Option("role", "Specify role", OptionType.ROLE, required=True),
]
)
@commands.has_permissions(administrator=True)
async def set_guest_role(ctx, role):
with open('prefix.json', 'r', encoding='utf-8') as fp:
try:
jsonObject = json.load(fp)
except:
jsonObject = {}
try:
jsonObject[f"{ctx.guild.id}"]["guest_role"] = role.id
except KeyError:
bot_role = jsonObject[f"{ctx.guild.id}"] = {'guest_role': ''}
new = {"guest_role": role.id}
bot_role.update(new)
jsonObject[str(ctx.guild.id)] = bot_role
prefixes[str(ctx.guild.id)] = prefix
await ctx.send(f"Setted up bot role to: `{role.name}`")
with open('config.json', 'w') as f: #writes the new prefix into the .json
json.dump(prefixes, f, indent=4)
await ctx.send(f'Prefix changed to: {prefix}') #confirms the prefix it's been changed to
#next step completely optional: changes bot nickname to also have prefix in the nickname
name=f'{prefix}BotBot'
with open("prefix.json", "w") as fl:
json.dump(jsonObject, fl)
fl.close()
bot.run('OTQ3OTUzOTAxNzgzNjIxNjYy.GTXbMv.KrztaTO7-ivsPEAVjsyikSQ-GP-ANwULmDraig')

42
test2.py
View File

@@ -1,33 +1,13 @@
import discord
import dislash
from discord.ext import commands
from dislash.slash_commands import check
from os import walk
bot = commands.Bot(command_prefix="!")
inter_client = dislash.InteractionClient(bot)
f = []
for filenames in walk('audio'):
f.extend(filenames)
break
f = f[2]
files = {}
keys = range(len(f))
for x in f:
files[x] = x
class Greetings(commands.Cog):
def __init__(self, bot):
self.bot = bot
self._last_member = None
@commands.Cog.listener()
async def on_member_join(self, member):
channel = member.guild.system_channel
if channel is not None:
await channel.send('Welcome {0.mention}.'.format(member))
@commands.command()
async def hello(self, ctx, *, member: discord.Member = None):
"""Says hello"""
member = member or ctx.author
if self._last_member is None or self._last_member.id != member.id:
await ctx.send('Hello {0.name}~'.format(member))
else:
await ctx.send('Hello {0.name}... This feels familiar.'.format(member))
self._last_member = member
bot.add_cog(Greetings(bot))
bot.run("OTQ3OTUzOTAxNzgzNjIxNjYy.GTXbMv.KrztaTO7-ivsPEAVjsyikSQ-GP-ANwULmDraig")
print(files)

BIN
user.db
View File

Binary file not shown.