First Commit

utils/bank_functions.py

@@ -0,0 +1,89 @@
+import discord
+from utils.sql_commands import DatabaseManager
+from typing import Dict
+from datetime import datetime
+
+
+db = DatabaseManager("economy")
+
+
+async def create_account(user: discord.Member | discord.User):
+    """Create a new economy account for a user."""
+    db.insert(
+        "INSERT INTO economy (ID, WALLET, BANK) VALUES (%s, %s, %s)", (user.id, 0, 1000)
+    )
+
+
+async def bank_data(user: discord.Member | discord.User) -> Dict[str, int]:
+    """Get the bank data for a user. If the user has no account, create one.
+
+    Args:
+        user: The user to get the bank data for.
+
+    Returns:
+        A dictionary with the keys ``WALLET`` and ``BANK`` containing the user's wallet and bank balances.
+    """
+    balance = db.fetch_one("SELECT * FROM economy WHERE ID = %s", (user.id,))
+    if balance is None:
+        await create_account(user)
+        return await bank_data(user)
+    return balance
+
+
+async def record_transaction(
+    user_id: int, transaction_type: str, amount: float
+) -> None:
+    """Record a transaction in the database.
+
+    Args:
+        user_id: The ID of the user who performed the transaction.
+        transaction_type: The type of transaction performed.
+        amount: The amount of the transaction.
+    """
+    db.insert("transactions", (user_id, transaction_type, amount))
+
+
+async def update_money(
+    user: discord.Member | discord.User, bank: int = 0, wallet: int = 0
+):
+    """Update the bank balance of a user.
+
+    Args:
+        user: The user to update the balance for.
+        bank: The amount to add to the user's bank balance.
+        wallet: The amount to add to the user's wallet balance.
+
+    Returns:
+        A dictionary with the keys ``WALLET`` and ``BANK`` containing the updated balances.
+    """
+    balance = await bank_data(user)
+    if bank or wallet:
+        new_wallet = balance.get("WALLET", 0) + wallet
+        new_bank = balance.get("BANK", 0) + bank
+        db.insert(
+            "INSERT INTO economy (ID, WALLET, BANK) VALUES (%s, %s, %s)",
+            (user.id, new_wallet, new_bank),
+            overwrite=True,
+        )
+
+
+async def reset_bank(user: discord.Member | discord.User) -> None:
+    """Reset the bank account for a user.
+
+    Args:
+        user (discord.Member | discord.User): The user whose bank account is to be reset.
+
+    Returns:
+        None
+    """
+    db.delete("economy", {"ID": user.id})
+    await create_account(user)
+
+
+async def update_daily_timestamp(user: discord.User | discord.Member, timestamp: datetime) -> None:
+    """
+    Updates the DAILY_TIMESTAMP field for the user in the economy table.
+    Stores the timestamp as a float (UNIX time).
+    """
+    db.execute_query("UPDATE economy SET DAILY = %s WHERE ID = %s",(timestamp.timestamp(), user.id),
+    )