152 lines
4.7 KiB
Python
152 lines
4.7 KiB
Python
"""
|
|
Contains the code that deals with money.
|
|
|
|
*Classes*
|
|
---------
|
|
Money
|
|
Deals with money.
|
|
"""
|
|
import discord_slash # Used for typehints
|
|
import discord # Used for typehints
|
|
|
|
|
|
class Money():
|
|
"""
|
|
Deals with money.
|
|
|
|
*Methods*
|
|
---------
|
|
checkBalance(user: str)
|
|
sendBalance(ctx: discord_slash.context.SlashContext)
|
|
addMoney(user: str, amount: int)
|
|
giveMoney(ctx: discord_slash.context.SlashContext, user: discord.User,
|
|
amount: int)
|
|
|
|
*Attributes*
|
|
------------
|
|
bot: Gwendolyn
|
|
The instance of Gwendolyn
|
|
database: pymongo.Client
|
|
The mongo database
|
|
"""
|
|
|
|
def __init__(self, bot):
|
|
"""Initialize the class."""
|
|
self.bot = bot
|
|
self.database = bot.database
|
|
|
|
def checkBalance(self, user: str):
|
|
"""
|
|
Get the account balance of a user.
|
|
|
|
*Parameters*
|
|
------------
|
|
user: str
|
|
The user to get the balance of.
|
|
|
|
*Returns*
|
|
---------
|
|
balance: int
|
|
The balance of the user's account.
|
|
"""
|
|
self.bot.log("checking "+user+"'s account balance")
|
|
|
|
userData = self.database["users"].find_one({"_id": user})
|
|
|
|
if userData is not None:
|
|
return userData["money"]
|
|
else:
|
|
return 0
|
|
|
|
async def sendBalance(self, ctx: discord_slash.context.SlashContext):
|
|
"""
|
|
Get your own account balance.
|
|
|
|
*Parameters*
|
|
------------
|
|
ctx: discord_slash.context.SlashContext
|
|
The context of the command.
|
|
"""
|
|
await self.bot.defer(ctx)
|
|
response = self.checkBalance("#"+str(ctx.author.id))
|
|
user_name = ctx.author.display_name
|
|
if response == 1:
|
|
new_message = f"{user_name} has {response} GwendoBuck"
|
|
else:
|
|
new_message = f"{user_name} has {response} GwendoBucks"
|
|
await ctx.send(new_message)
|
|
|
|
# Adds money to the account of a user
|
|
def addMoney(self, user: str, amount: int):
|
|
"""
|
|
Add money to a user account.
|
|
|
|
*Parameters*
|
|
------------
|
|
user: str
|
|
The id of the user to give money.
|
|
amount: int
|
|
The amount to add to the user's account.
|
|
"""
|
|
self.bot.log("adding "+str(amount)+" to "+user+"'s account")
|
|
|
|
userData = self.database["users"].find_one({"_id": user})
|
|
|
|
if userData is not None:
|
|
updater = {"$inc": {"money": amount}}
|
|
self.database["users"].update_one({"_id": user}, updater)
|
|
else:
|
|
newUser = {
|
|
"_id": user,
|
|
"user name": self.bot.database_funcs.get_name(user),
|
|
"money": amount
|
|
}
|
|
self.database["users"].insert_one(newUser)
|
|
|
|
# Transfers money from one user to another
|
|
async def giveMoney(self, ctx: discord_slash.context.SlashContext,
|
|
user: discord.User, amount: int):
|
|
"""
|
|
Give someone else money from your account.
|
|
|
|
*Parameters*
|
|
------------
|
|
ctx: discord_slash.context.SlashContext
|
|
The context of the command.
|
|
user: discord.User
|
|
The user to give money.
|
|
amount: int
|
|
The amount to transfer.
|
|
"""
|
|
await self.bot.defer(ctx)
|
|
username = user.display_name
|
|
if self.bot.database_funcs.get_id(username) is None:
|
|
async for member in ctx.guild.fetch_members(limit=None):
|
|
if member.display_name.lower() == username.lower():
|
|
username = member.display_name
|
|
user_id = f"#{member.id}"
|
|
newUser = {
|
|
"_id": user_id,
|
|
"user name": username,
|
|
"money": 0
|
|
}
|
|
self.bot.database["users"].insert_one(newUser)
|
|
|
|
userid = f"#{ctx.author.id}"
|
|
userData = self.database["users"].find_one({"_id": userid})
|
|
targetUser = self.bot.database_funcs.get_id(username)
|
|
|
|
if amount <= 0:
|
|
self.bot.log("They tried to steal")
|
|
await ctx.send("Yeah, no. You can't do that")
|
|
elif targetUser is None:
|
|
self.bot.log("They weren't in the system")
|
|
await ctx.send("The target doesn't exist")
|
|
elif userData is None or userData["money"] < amount:
|
|
self.bot.log("They didn't have enough GwendoBucks")
|
|
await ctx.send("You don't have that many GwendoBuck")
|
|
else:
|
|
self.addMoney(f"#{ctx.author.id}", -1 * amount)
|
|
self.addMoney(targetUser, amount)
|
|
await ctx.send(f"Transferred {amount} GwendoBucks to {username}")
|