add timestamp and author id

bot.py

@@ -3,6 +3,7 @@ from discord import Option, commands
 import redis
 import os
 import random
+import time
 
 # Function to generate auto-incremented IDs
 def generate_id(key):
@@ -26,26 +27,26 @@ r = redis.from_url(redis_url)
 
 r.set('species_identified', 0)
 
-def improve_player_stats_image(ctx):
-    # add 1 to the number of species identified
-    value = r.get(b'species_identified').decode('utf-8')
-    r.set(b'species_identified', int(value) + 1)
+# def improve_player_stats_image(ctx):
+#     # add 1 to the number of species identified
+#     value = r.get(b'species_identified').decode('utf-8')
+#     r.set(b'species_identified', int(value) + 1)
 
-    xp_key = f'{ctx.author.name}:XP'.encode('utf-8')
-    if not r.exists(xp_key):
-        r.set(xp_key, 0)
+#     xp_key = f'{ctx.author.name}:XP'.encode('utf-8')
+#     if not r.exists(xp_key):
+#         r.set(xp_key, 0)
 
-    role_key = f'{ctx.author.name}:role'.encode('utf-8')
-    if not r.exists(role_key):
-        r.set(role_key, 'Naturalist')
+#     role_key = f'{ctx.author.name}:role'.encode('utf-8')
+#     if not r.exists(role_key):
+#         r.set(role_key, 'Naturalist')
 
-    img_oc_key = f'{ctx.author.name}:image'.encode('utf-8')
-    if not r.exists(img_oc_key):
-        r.set(img_oc_key, 0)
+#     img_oc_key = f'{ctx.author.name}:image'.encode('utf-8')
+#     if not r.exists(img_oc_key):
+#         r.set(img_oc_key, 0)
 
-    r.set(xp_key, int(r.get(xp_key)) + 10)
+#     r.set(xp_key, int(r.get(xp_key)) + 10)
 
-    r.set(img_oc_key, int(r.get(img_oc_key)) + 1)
+#     r.set(img_oc_key, int(r.get(img_oc_key)) + 1)
 
 
 bot = discord.Bot()
@@ -137,10 +138,13 @@ def get_conf_score(confidence: str):
     ])
 async def identify_image(ctx, id:int, label:str, confidence: str):
 
+    timestamp = str(time.time())
+    author_id = str(ctx.author_id)
+
     role = "Hobbyist"
     role_names = [role.name for role in ctx.author.roles]
     if "Expert" in role_names:
-        expert = "Expert"
+        role = "Expert"
     
     label_cnt = generate_id(f'cnt:label:image:{id}')
     r.hset(f'label:image:{id}:{label_cnt}'.encode('utf-8'), b'label', label.encode('utf-8'))
@@ -148,6 +152,8 @@ async def identify_image(ctx, id:int, label:str, confidence: str):
     conf_score = get_conf_score(confidence)
     r.hset(f'label:image:{id}:{label_cnt}'.encode('utf-8'), b'confidence', conf_score)
     r.hset(f'label:image:{id}:{label_cnt}'.encode('utf-8'), b'role', role)
+    r.hset(f'label:image:{id}:{label_cnt}'.encode('utf-8'), b'authorid', author_id)
+    r.hset(f'label:image:{id}:{label_cnt}'.encode('utf-8'), b'timestamp', timestamp)
 
     r.hset(f'image:{id}'.encode('utf-8'), b'label', label.encode('utf-8'))
     r.hset(f'image:{id}'.encode('utf-8'), b'author', ctx.author.mention.encode('utf-8'))
@@ -202,10 +208,13 @@ async def sound(ctx, id): # a slash command will be created with the name "ping"
     ])
 async def identify_sound(ctx, id:int, label:str, timestamp:str, confidence: str):
 
+    timestamp = str(time.time())
+    author_id = str(ctx.author_id)
+
     role = "Hobbyist"
     role_names = [role.name for role in ctx.author.roles]
     if "Expert" in role_names:
-        expert = "Expert"
+        role = "Expert"
 
     label_cnt = generate_id(f'cnt:label:sound:{id}')
     r.hset(f'label:sound:{id}:{label_cnt}'.encode('utf-8'), b'label', label.encode('utf-8'))
@@ -214,6 +223,8 @@ async def identify_sound(ctx, id:int, label:str, timestamp:str, confidence: str)
     r.hset(f'label:sound:{id}:{label_cnt}'.encode('utf-8'), b'label_at', timestamp.encode('utf-8'))
     conf_score = get_conf_score(confidence)
     r.hset(f'label:sound:{id}:{label_cnt}'.encode('utf-8'), b'confidence', conf_score)
+    r.hset(f'label:sound:{id}:{label_cnt}'.encode('utf-8'), b'authorid', author_id)
+    r.hset(f'label:sound:{id}:{label_cnt}'.encode('utf-8'), b'timestamp', timestamp)
 
     r.hset(f'sound:{id}'.encode('utf-8'), b'label', label.encode('utf-8'))
     r.hset(f'sound:{id}'.encode('utf-8'), b'label_at', timestamp.encode('utf-8'))
@@ -241,7 +252,11 @@ async def identify_sound(ctx, id:int, label:str, timestamp:str, confidence: str)
     )
 async def next_image(ctx):
     total_images = r.get(b'cnt:image').decode('utf-8')
-    id = random.randint(1, int(total_images))
+    numbers = list(range(1, int(total_images)))
+    exclude_redis = r.lrange('image:accept', 0, -1)
+    exclude = set(int(num.decode('utf-8')) for num in exclude_redis)
+    numbers = [num for num in numbers if num not in exclude]
+    id = random.choice(numbers)
     await image(ctx, id)
 
 @bot.command(
@@ -249,32 +264,41 @@ async def next_image(ctx):
     )
 async def next_sound(ctx):
     total_sound = r.get(b'cnt:sound').decode('utf-8')
-    id = random.randint(1, int(total_sound))
+    numbers = list(range(1, int(total_sound)))
+    exclude_redis = r.lrange('image:accept', 0, -1)
+    exclude = set(int(num.decode('utf-8')) for num in exclude_redis)
+    numbers = [num for num in numbers if num not in exclude]
+    id = random.choice(numbers)
     await sound(ctx, id)
 
 @bot.command(
         description="Check your profile"
     )
 async def profile(ctx):
-    xp_key = f'{ctx.author.name}:XP'.encode('utf-8')
-    if not r.exists(xp_key):
-        r.set(xp_key, 0)
+    # xp_key = f'{ctx.author.name}:XP'.encode('utf-8')
+    # if not r.exists(xp_key):
+    #     r.set(xp_key, 0)
 
-    img_oc_key = f'{ctx.author.name}:image'.encode('utf-8')
-    if not r.exists(img_oc_key):
-        r.set(img_oc_key, 0)
+    # img_oc_key = f'{ctx.author.name}:image'.encode('utf-8')
+    # if not r.exists(img_oc_key):
+    #     r.set(img_oc_key, 0)
 
-    s_oc_key = f'{ctx.author.name}:sound'.encode('utf-8')
-    if not r.exists(s_oc_key):
-        r.set(s_oc_key, 0)
+    # s_oc_key = f'{ctx.author.name}:sound'.encode('utf-8')
+    # if not r.exists(s_oc_key):
+    #     r.set(s_oc_key, 0)
 
      # Create an embed
+    role = "Hobbyist"
+    role_names = [role.name for role in ctx.author.roles]
+    if "Expert" in role_names:
+        role = "Expert"
+
     embed = discord.Embed(color=discord.Colour.blurple())
     embed.set_author(name=f"{ctx.author.name} - Profile", icon_url=ctx.author.display_avatar.url)
-    embed.add_field(name="**PROGRESS**", value=f"**Level**: 1\n**XP**: {r.get(xp_key).decode()}/1000\n", inline=False)
-    embed.add_field(name="**STATS**", value=f"**🌱 Images labeled**: {r.get(img_oc_key).decode()}", inline=False)
+    # embed.add_field(name="**PROGRESS**", value=f"**Level**: 1\n**XP**: {r.get(xp_key).decode()}/1000\n", inline=False)
+    # embed.add_field(name="**STATS**", value=f"**🌱 Images labeled**: {r.get(img_oc_key).decode()}", inline=False)
     embed.set_thumbnail(url=ctx.author.display_avatar.url)
-    # embed.set_footer(text=f"🏆 RANK: {r.get(role_key).decode()}")
+    embed.set_footer(text=f"🏆 Role: {r.get(role).decode()}")
     await ctx.respond(embed=embed)
 
 bot.run('MTA5NzMzMDI2MDI4NDAzNTEyMg.GNFiQP.hZC_2HLTvVAROlKKUmnVQjviT0G4wHeQq23-rs')