From b612e18955693c20e1009c324301a3f3bbaa1492 Mon Sep 17 00:00:00 2001 From: strNophix Date: Sat, 6 Aug 2022 22:06:11 +0200 Subject: [PATCH] Added delta for record deduplication --- lurker/bot.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/lurker/bot.py b/lurker/bot.py index 633f00a..88a1956 100644 --- a/lurker/bot.py +++ b/lurker/bot.py @@ -1,5 +1,7 @@ import sys import typing +from datetime import datetime +from datetime import timedelta import aiohttp import discord @@ -70,7 +72,7 @@ class RecordBatchStore(typing.Generic[T]): class Bot(discord.Client): - last_insert: dict[int, typing.Any] = {} + last_insert: dict[int, datetime] = {} track_store: RecordBatchStore[SpotifyRecord] = RecordBatchStore(submit_records, 1) async def on_ready(self): @@ -83,7 +85,8 @@ class Bot(discord.Client): return spotify: discord.Spotify - if self.last_insert.get(user_id) == spotify.created_at: + last_insert = self.last_insert.get(user_id) + if last_insert and last_insert - spotify.created_at > timedelta(seconds=15): return self.last_insert[user_id] = spotify.created_at