71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
from discord.ext import commands, tasks
|
|
from discord import app_commands
|
|
import sqlite3
|
|
import discord
|
|
|
|
from global_variables import BOT_COLOR
|
|
|
|
|
|
class Stats(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
def cog_load(self):
|
|
connection = sqlite3.connect('count.db')
|
|
cursor = connection.cursor()
|
|
cursor.execute("CREATE TABLE IF NOT EXISTS count (command_name, count, PRIMARY KEY (command_name))")
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
self.dump_count.start()
|
|
|
|
@tasks.loop(seconds=30)
|
|
async def dump_count(self):
|
|
connection = sqlite3.connect('count.db')
|
|
cursor = connection.cursor()
|
|
|
|
for command_name, count in self.bot.temp_command_count.items():
|
|
try:
|
|
cursor.execute("INSERT INTO count (command_name, count) VALUES (?, ?)", (command_name, count))
|
|
except sqlite3.IntegrityError:
|
|
cursor.execute("UPDATE count SET count = count + ? WHERE command_name = ?", (count, command_name))
|
|
|
|
connection.commit()
|
|
connection.close()
|
|
self.bot.temp_command_count = {}
|
|
|
|
|
|
@commands.Cog.listener()
|
|
async def on_app_command_completion(self, interaction, command):
|
|
try:
|
|
self.bot.temp_command_count[interaction.command.name] += 1
|
|
except KeyError:
|
|
self.bot.temp_command_count[interaction.command.name] = 1
|
|
|
|
|
|
@commands.command()
|
|
@commands.dm_only()
|
|
@commands.is_owner()
|
|
async def stats(self, ctx: commands.Context):
|
|
connection = sqlite3.connect('count.db')
|
|
cursor = connection.cursor()
|
|
|
|
embed = discord.Embed(
|
|
title="Command Statistics",
|
|
color=BOT_COLOR
|
|
)
|
|
|
|
total = 0
|
|
data = cursor.execute("SELECT * FROM count").fetchall()
|
|
for entry in data:
|
|
embed.add_field(name=entry[0], value=f"` {entry[1]} `", inline=True)
|
|
total += entry[1]
|
|
|
|
embed.add_field(name="TOTAL", value=f"` {total} `", inline=False)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Stats(bot))
|