Untitled

import discord
import random
from discord.ext import commands
from discord import app_commands
from pymongo import MongoClient
from discord.ext import commands

# MongoDB Connection Setup
client = MongoClient("mongodb+srv://williamjmaurice:ygDY0QI2Om4Y7YDZ@economy.iaynb.mongodb.net/?retryWrites=true&w=majority&appName=Economy")  # Replace with your MongoDB connection URI
db = client['Economy']  # The database name
collection = db['Metal']  # The collection name

# Bot Setup
intents = discord.Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix="!", intents=intents)

@bot.event
async def on_ready():
    print(f'Logged in as {bot.user}')

@bot.command()
async def mine(ctx):
    num = random.randint(1,4)
    await ctx.send(f"You have mined {num} ore")

    """
    Saves an integer to MongoDB with the user's ID.
    """
    # Create a dictionary with user information and the integer value
    user_data = {
        'user_id': ctx.author.id,
        'username': ctx.author.name,
        'ore': num,
    }

    # Insert the data into MongoDB
    collection.update_one(
        {'user_id': ctx.author.id},  # Find document by user ID
        {'$set': user_data},          # Update or insert the new data
        upsert=True                   # If no document exists, create one
    )    

@bot.command()
async def mine_bal(ctx):
    """
    Retrieves the integer saved for a specific user from MongoDB.
    """
    # Find the user's data in the collection
    user_data = collection.find_one({'user_id': ctx.author.id})

    if user_data:
        saved_integer = user_data.get('saved_integer')
        await ctx.send(f"Your metal balance is: {saved_integer}")
    else:
        await ctx.send("No integer has been saved for you.")

# Run the bot with your token
bot.run("TOKEN")