Nobody2503
be89cc3acd
- Removed the old npc_memory.db file. - Updated time.txt with a new timestamp. - Refactored transaction recording in bank_functions.py to use parameterized queries. - Enhanced DatabaseManager in sql_commands.py to support singleton pattern and improved table creation logic. - Added methods for sanitizing SQL identifiers and parsing insert columns for upsert operations. - Improved error handling and connection management in execute_query, fetch_one, fetch_all, and fetch_as_dataframe methods. - Introduced a new bootstrap_database.py script for initializing the database schema. - Updated app.py to use the new initialize_database function for database management.
93 lines
2.8 KiB
Python
Executable File
93 lines
2.8 KiB
Python
Executable File
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.execute_query(
|
|
"INSERT INTO transactions (USERID, TYPE, AMOUNT) VALUES (%s, %s, %s)",
|
|
(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),
|
|
)
|