DisBot.py/disbot.py

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(aliases=["blm"])
async def blame(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:
            
            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)
    
    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