nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix started thingy

Browse source
This commit is contained in:
Nexus 2023-12-05 18:22:01 +00:00
parent 711f666d8c
commit b53121ebb5
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cogs/mcdonalds.py
View file

@ -45,13 +45,15 @@ class McDataBase:
) as cursor:
return await cursor.fetchone()
async def set_break(self, user_id: int, since: float) -> None:
async def set_break(self, user_id: int, since: float, started: float = None) -> None:
if not started:
started = discord.Object(discord.utils.generate_snowflake()).created_at.timestamp()
await self._conn.execute(
"""
INSERT INTO breaks (user_id, since) VALUES (?, ?)
ON CONFLICT(user_id) DO UPDATE SET since = excluded.since
INSERT INTO breaks (user_id, since, started) VALUES (?, ?, ?)
ON CONFLICT(user_id) DO UPDATE SET since = excluded.since, started = excluded.started
""",
(user_id, since)
(user_id, since, started)
)
async def remove_break(self, user_id: int) -> None: