DisBot.py/disbot.py
2024-10-28 00:14:09 +01:00

197 lines
4.8 KiB
Python

import discord
from discord.ext import commands
#from private.config import token
import json
import os
os.chdir("/home/godot/Documents/scripts/discord-bot.py")
#os.chdir("/opt/disbot")
intents = discord.Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix='$', intents=intents)
@bot.event
# Start succes indicator
async def on_ready():
print(f'We have logged in as {bot.user}')
@bot.command()
# Ahoj function
async def ahoj(ctx):
await ctx.send("Zdravím miláčkové!")
@bot.command(aliases = ["bal"])
async def balance(ctx, member:discord.Member = "neni"):
# test if acount is opend and makes on if not
if member == "neni":
member = ctx.author
if member != ctx.author:
role = await check_role(ctx, ctx.author)
if role == False:
await ctx.send("Větší snahy ze strany tvé nastati budou než já podlehnu.")
return
await open_account(ctx.author)
await open_account(member)
users = await get_bank_data()
wallet_amt = users[str(member.id)]["wallet"]
leader_board = {}
total = []
for user in users:
name = int(user)
leader_board[users[user]["wallet"]] = name
total.append([users[user]["wallet"]])
total = sorted(total, reverse=True)
poradi = total.index([users[str(member.id)]["wallet"]]) +1
em = discord.Embed(title = f"{member.name}",color = discord.Color.blue())
em.add_field(name = "Bodíčků máš", value = wallet_amt, inline = False)
em.add_field(name = "V poradi jsi:", value = poradi, inline = False)
em.set_thumbnail(url = member.avatar)
await ctx.send("Co bych pro tebe neudělal!")
await ctx.author.send(embed = em)
@bot.command()
async def shame(ctx):
role = await check_role(ctx, ctx.author)
if role == False:
await ctx.send("Já nevěřit tvým šprímům, ty více ukázati musíš.")
return
users = await get_bank_data()
loser_dict = {}
em = discord.Embed(title = "List of shame", color = discord.Color.red())
for user in users:
if users[user]["wallet"] < 0:
#creates dictionary of needed info
debt = users[user]["wallet"]
if int(users[user]["wallet"]) in loser_dict:
loser_dict[debt].append(user)
else:
loser_dict[debt] = []
loser_dict[debt].append(user)
#sortes
loser_dict = dict(sorted(loser_dict.items()))
# formats string to desired format
for loser in loser_dict:
i = 0
while i < len(loser_dict[loser]):
loser_dict[loser][i] = f'<@{loser_dict[loser][i]}>'
i += 1
em.add_field(name = loser, value = ", ".join(loser_dict[loser]), inline = False)
await ctx.author.send(embed = em)
@bot.command()
async def payout(ctx,member:discord.Member,amount = None):
await open_account(member)
role = await check_role(ctx, ctx.author)
if amount == None:
await ctx.send("Částku zapoměl napsat jsi.")
return
if role == False:
await ctx.send("Já nevěřit tvým šprímům, ty více ukázati musíš.")
if role:
await update_bank(member,amount)
await ctx.send(f"{member} dostal vyplaceno {amount} bodíčků!")
@bot.command()
async def cut(ctx,member:discord.Member,amount = None):
await open_account(member)
role = await check_role(ctx, ctx.author)
if amount == None:
await ctx.send("Částku zapoměl napsat jsi.")
return
if role == False:
await ctx.send("Já nevěřit tvým šprímům, ty více ukázati musíš.")
if role:
await update_bank(member,int(amount)*-1)
await ctx.send(f"{member} dostal zabaveno {amount} bodíčků!")
async def open_account(user):
# opens json file to found if user id is present
users = await get_bank_data()
# tests presence
if str(user.id) in users:
return False
else:
users[str(user.id)] = {}
users[str(user.id)]["wallet"] = 100
# saves a new data in json
with open("mainbank.json", "w") as f:
json.dump(users,f)
return True
# opens json stored data
async def get_bank_data():
with open("mainbank.json", "r") as f:
users = json.load(f)
return users
async def update_bank(user,change = 0):
users = await get_bank_data()
users[str(user.id)]["wallet"] += int(change)
with open("mainbank.json", "w") as f:
json.dump(users,f)
bal = [users[str(user.id)]["wallet"]]
return bal
async def check_role(ctx, user:discord.Member):
role = discord.utils.get(ctx.guild.roles,name = "Učitelé")
if role in user.roles:
return True
else:
return False