131 lines
6.2 KiB
Python
131 lines
6.2 KiB
Python
import discord
|
|
from datetime import datetime
|
|
from core.bot import bot, server_repo, channel_repo, channel_app_repo, home_debt_repo
|
|
from models import DiscordServer, DiscordChannel, DiscordChannelApp
|
|
|
|
@bot.tree.command(name='set-home-debt', description='Set channel to home debt')
|
|
async def set_home_debt(interaction: discord.Interaction, channel: discord.TextChannel):
|
|
try:
|
|
# Get or create server
|
|
server = await server_repo.get(str(interaction.guild_id))
|
|
if not server:
|
|
server = DiscordServer(
|
|
id=str(interaction.guild_id),
|
|
name=interaction.guild.name,
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now()
|
|
)
|
|
server = await server_repo.create(server)
|
|
|
|
# Get or create channel
|
|
discord_channel = await channel_repo.get(str(channel.id))
|
|
if not discord_channel:
|
|
discord_channel = DiscordChannel(
|
|
id=str(channel.id),
|
|
server_id=str(interaction.guild_id),
|
|
name=channel.name,
|
|
type='text',
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now()
|
|
)
|
|
discord_channel = await channel_repo.create(discord_channel)
|
|
|
|
# Create or update home debt app
|
|
apps = await channel_app_repo.get_by_channel(str(channel.id))
|
|
home_debt_app = next((app for app in apps if app.app_type == 'home_debt'), None)
|
|
|
|
if home_debt_app:
|
|
home_debt_app.config = {'enabled': True}
|
|
await channel_app_repo.update(home_debt_app)
|
|
else:
|
|
home_debt_app = DiscordChannelApp(
|
|
id=f"{channel.id}_home_debt",
|
|
channel_id=str(channel.id),
|
|
app_type='home_debt',
|
|
config={'enabled': True},
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now()
|
|
)
|
|
await channel_app_repo.create(home_debt_app)
|
|
|
|
await interaction.response.send_message(f"Home debt app has been set up for {channel.mention}")
|
|
except Exception as e:
|
|
await interaction.response.send_message(f"Error setting up home debt app: {str(e)}")
|
|
|
|
@bot.tree.command(name="add_member", description="Thêm thành viên mới vào nhà")
|
|
async def add_member(interaction: discord.Interaction, member: discord.Member):
|
|
"""Thêm thành viên mới vào nhà"""
|
|
try:
|
|
# Kiểm tra xem thành viên đã tồn tại chưa
|
|
existing_member = await home_debt_repo.get_member(member.id)
|
|
if existing_member:
|
|
await interaction.response.send_message(f"Thành viên {member.name} đã tồn tại trong hệ thống!", ephemeral=True)
|
|
return
|
|
|
|
# Thêm thành viên mới
|
|
await home_debt_repo.add_member(member.id, member.name)
|
|
await interaction.response.send_message(f"Đã thêm thành viên {member.name} vào hệ thống!", ephemeral=True)
|
|
except Exception as e:
|
|
await interaction.response.send_message(f"Có lỗi xảy ra: {str(e)}", ephemeral=True)
|
|
|
|
@bot.tree.command(name="add_expense", description="Thêm khoản chi tiêu mới")
|
|
async def add_expense(interaction: discord.Interaction, amount: float, description: str):
|
|
"""Thêm khoản chi tiêu mới"""
|
|
try:
|
|
# Kiểm tra xem người dùng có phải là thành viên không
|
|
member = await home_debt_repo.get_member(interaction.user.id)
|
|
if not member:
|
|
await interaction.response.send_message("Bạn chưa được thêm vào hệ thống! Vui lòng sử dụng lệnh /add_member trước.", ephemeral=True)
|
|
return
|
|
|
|
# Thêm khoản chi tiêu
|
|
expense = await home_debt_repo.add_expense(amount, description, interaction.user.id)
|
|
await interaction.response.send_message(f"Đã thêm khoản chi tiêu: {description} - {amount}đ", ephemeral=True)
|
|
except Exception as e:
|
|
await interaction.response.send_message(f"Có lỗi xảy ra: {str(e)}", ephemeral=True)
|
|
|
|
@bot.tree.command(name="check_balance", description="Kiểm tra số dư của bạn")
|
|
async def check_balance(interaction: discord.Interaction):
|
|
"""Kiểm tra số dư của bạn"""
|
|
try:
|
|
# Kiểm tra xem người dùng có phải là thành viên không
|
|
member = await home_debt_repo.get_member(interaction.user.id)
|
|
if not member:
|
|
await interaction.response.send_message("Bạn chưa được thêm vào hệ thống! Vui lòng sử dụng lệnh /add_member trước.", ephemeral=True)
|
|
return
|
|
|
|
# Lấy số dư
|
|
balance = await home_debt_repo.get_balance(interaction.user.id)
|
|
await interaction.response.send_message(f"Số dư của bạn: {balance}đ", ephemeral=True)
|
|
except Exception as e:
|
|
await interaction.response.send_message(f"Có lỗi xảy ra: {str(e)}", ephemeral=True)
|
|
|
|
@bot.tree.command(name="list_expenses", description="Xem danh sách chi tiêu")
|
|
async def list_expenses(interaction: discord.Interaction):
|
|
"""Xem danh sách chi tiêu"""
|
|
try:
|
|
# Kiểm tra xem người dùng có phải là thành viên không
|
|
member = await home_debt_repo.get_member(interaction.user.id)
|
|
if not member:
|
|
await interaction.response.send_message("Bạn chưa được thêm vào hệ thống! Vui lòng sử dụng lệnh /add_member trước.", ephemeral=True)
|
|
return
|
|
|
|
# Lấy danh sách chi tiêu
|
|
expenses = await home_debt_repo.get_expenses()
|
|
if not expenses:
|
|
await interaction.response.send_message("Chưa có khoản chi tiêu nào!", ephemeral=True)
|
|
return
|
|
|
|
# Tạo embed message
|
|
embed = discord.Embed(title="Danh sách chi tiêu", color=discord.Color.blue())
|
|
for expense in expenses:
|
|
member = await home_debt_repo.get_member(expense.member_id)
|
|
embed.add_field(
|
|
name=f"{expense.description} - {expense.amount}đ",
|
|
value=f"Người chi: {member.name}",
|
|
inline=False
|
|
)
|
|
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
except Exception as e:
|
|
await interaction.response.send_message(f"Có lỗi xảy ra: {str(e)}", ephemeral=True) |