From b53121ebb5e685571220a13fbf5933c192babe29 Mon Sep 17 00:00:00 2001 From: nex Date: Tue, 5 Dec 2023 18:22:01 +0000 Subject: [PATCH] Fix started thingy --- cogs/mcdonalds.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cogs/mcdonalds.py b/cogs/mcdonalds.py index 031ca3c..2dcc2a3 100644 --- a/cogs/mcdonalds.py +++ b/cogs/mcdonalds.py @@ -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: