blob: 67e01394088e0d6168300ac9e99504fce4b574dc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from discord.ext import commands, tasks
import aiosqlite
import sqlite3
class Count(commands.Cog):
def __init__(self, bot):
self.bot = bot
async def cog_load(self):
self.dump_count.start()
@tasks.loop(seconds=5)
async def dump_count(self):
try:
cur = await aiosqlite.connect("./code/count/count.db")
count = await cur.execute("SELECT count FROM count")
count = await count.fetchone()
if count is None:
await cur.execute("INSERT INTO count (count) VALUES (?)", (self.bot.count_hold,))
else:
await cur.execute("UPDATE count SET count = count + ?", (self.bot.count_hold,))
await cur.commit()
await cur.close()
self.bot.count_hold = 0
except sqlite3.OperationalError:
try:
await cur.commit()
await cur.close()
except:
pass
@commands.Cog.listener()
async def on_app_command_completion(self, interaction, command):
self.bot.count_hold += 1
async def setup(bot):
await bot.add_cog(Count(bot))
|
